From 187b73fb990ee871d8f09b01de003c5801839bef Mon Sep 17 00:00:00 2001
From: yangxin <245051644@qq.com>
Date: Thu, 24 Nov 2022 11:16:13 +0800
Subject: [PATCH] Add test of rexec.

Signed-off-by: yangxin <245051644@qq.com>
---
 qtfs/rexec/Makefile       |  3 ++-
 qtfs/rexec/common_test.go | 52 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+), 1 deletion(-)
 create mode 100644 qtfs/rexec/common_test.go

diff --git a/qtfs/rexec/Makefile b/qtfs/rexec/Makefile
index 8f95c42..86d4f78 100644
--- a/qtfs/rexec/Makefile
+++ b/qtfs/rexec/Makefile
@@ -5,6 +5,7 @@ rexec : client.go common.go
 
 rexec_server : server.go common.go
 	go build -o rexec_server server.go common.go
-
+test:
+	go test -v ./common_test.go ./common.go
 clean:
 	rm -rf rexec rexec_server
diff --git a/qtfs/rexec/common_test.go b/qtfs/rexec/common_test.go
new file mode 100644
index 0000000..858557e
--- /dev/null
+++ b/qtfs/rexec/common_test.go
@@ -0,0 +1,52 @@
+package main
+import(
+	"os"
+	"testing"
+)
+
+func TestParseNetAddrTcp(t *testing.T) {
+	var na NetAddr
+	var err error
+
+	os.Setenv("CMD_NET_ADDR", "tcp://127.0.0.1:1234")
+	na, err = parseNetAddr()
+	if err != nil {
+		t.Fatalf("parseNetAddr tcp://127.0.0.1:1234 failed with error: %s\n", err)
+	}
+	
+	if na.Proto != "tcp" {
+		t.Fatalf("parseNetAddr tcp://127.0.0.1:1234 failed: expect proto to be tcp, get :%s\n", na.Proto) 
+	}
+
+	if na.Addr != "127.0.0.1:1234" {
+		t.Fatalf("parseNetAddr tcp://127.0.0.1:1234 failed: get addr %s\n", na.Addr)
+	}
+}
+
+func TestParseNetAddrUdp(t *testing.T) {
+	var err error
+	os.Setenv("CMD_NET_ADDR", "udp://127.0.0.1:1234")
+	_, err = parseNetAddr()
+	if err == nil {
+		t.Fatalf("parseNetAddr udp://127.0.0.1:1234 should fail\n")
+	}
+}
+
+func TestParseNetAddrUnix(t *testing.T) {
+	var na NetAddr
+	var err error
+
+	os.Setenv("CMD_NET_ADDR", "unix:///tmp/test.sock")
+	na, err = parseNetAddr()
+	if err != nil {
+		t.Fatalf("parseNetAddr unix:///tmp/test.sock failed with error: %s\n", err)
+	}
+	
+	if na.Proto != "unix" {
+		t.Fatalf("parseNetAddr unix:///tmp/test.sock failed: expect proto to be unix, get :%s\n", na.Proto) 
+	}
+
+	if na.Addr != "/tmp/test.sock" {
+		t.Fatalf("parseNetAddr unix:///tmp/test.sock failed: get addr %s\n", na.Addr)
+	}
+}
\ No newline at end of file
-- 
Gitee