1 files changed, 143 insertions, 0 deletions
diff --git a/plugins/rpc/config_test.go b/plugins/rpc/config_test.go
new file mode 100755
index 00000000..67532bc8
--- /dev/null
+++ b/plugins/rpc/config_test.go
@@ -0,0 +1,143 @@
+package rpc
+
+import (
+	"runtime"
+	"testing"
+
+	j "github.com/json-iterator/go"
+	"github.com/stretchr/testify/assert"
+)
+
+var json = j.ConfigCompatibleWithStandardLibrary
+
+type testCfg struct{ cfg string }
+
+func (cfg *testCfg) Unmarshal(out interface{}) error {
+	return json.Unmarshal([]byte(cfg.cfg), out)
+}
+
+func TestConfig_Listener(t *testing.T) {
+	cfg := &Config{Listen: "tcp://:18001"}
+
+	ln, err := cfg.Listener()
+	assert.NoError(t, err)
+	assert.NotNil(t, ln)
+	defer func() {
+		err := ln.Close()
+		if err != nil {
+			t.Errorf("error closing the listener: error %v", err)
+		}
+	}()
+
+	assert.Equal(t, "tcp", ln.Addr().Network())
+	if runtime.GOOS == "windows" {
+		assert.Equal(t, "[::]:18001", ln.Addr().String())
+	} else {
+		assert.Equal(t, "0.0.0.0:18001", ln.Addr().String())
+	}
+}
+
+func TestConfig_ListenerUnix(t *testing.T) {
+	cfg := &Config{Listen: "unix://file.sock"}
+
+	ln, err := cfg.Listener()
+	assert.NoError(t, err)
+	assert.NotNil(t, ln)
+	defer func() {
+		err := ln.Close()
+		if err != nil {
+			t.Errorf("error closing the listener: error %v", err)
+		}
+	}()
+
+	assert.Equal(t, "unix", ln.Addr().Network())
+	assert.Equal(t, "file.sock", ln.Addr().String())
+}
+
+func Test_Config_Error(t *testing.T) {
+	cfg := &Config{Listen: "uni:unix.sock"}
+	ln, err := cfg.Listener()
+	assert.Nil(t, ln)
+	assert.Error(t, err)
+	assert.Equal(t, "invalid DSN (tcp://:6001, unix://file.sock)", err.Error())
+}
+
+func Test_Config_ErrorMethod(t *testing.T) {
+	cfg := &Config{Listen: "xinu://unix.sock"}
+
+	ln, err := cfg.Listener()
+	assert.Nil(t, ln)
+	assert.Error(t, err)
+}
+
+func TestConfig_Dialer(t *testing.T) {
+	cfg := &Config{Listen: "tcp://:18001"}
+
+	ln, _ := cfg.Listener()
+	defer func() {
+		err := ln.Close()
+		if err != nil {
+			t.Errorf("error closing the listener: error %v", err)
+		}
+	}()
+
+	conn, err := cfg.Dialer()
+	assert.NoError(t, err)
+	assert.NotNil(t, conn)
+	defer func() {
+		err := conn.Close()
+		if err != nil {
+			t.Errorf("error closing the connection: error %v", err)
+		}
+	}()
+
+	assert.Equal(t, "tcp", conn.RemoteAddr().Network())
+	assert.Equal(t, "127.0.0.1:18001", conn.RemoteAddr().String())
+}
+
+func TestConfig_DialerUnix(t *testing.T) {
+	cfg := &Config{Listen: "unix://file.sock"}
+
+	ln, _ := cfg.Listener()
+	defer func() {
+		err := ln.Close()
+		if err != nil {
+			t.Errorf("error closing the listener: error %v", err)
+		}
+	}()
+
+	conn, err := cfg.Dialer()
+	assert.NoError(t, err)
+	assert.NotNil(t, conn)
+	defer func() {
+		err := conn.Close()
+		if err != nil {
+			t.Errorf("error closing the connection: error %v", err)
+		}
+	}()
+
+	assert.Equal(t, "unix", conn.RemoteAddr().Network())
+	assert.Equal(t, "file.sock", conn.RemoteAddr().String())
+}
+
+func Test_Config_DialerError(t *testing.T) {
+	cfg := &Config{Listen: "uni:unix.sock"}
+	ln, err := cfg.Dialer()
+	assert.Nil(t, ln)
+	assert.Error(t, err)
+	assert.Equal(t, "invalid socket DSN (tcp://:6001, unix://file.sock)", err.Error())
+}
+
+func Test_Config_DialerErrorMethod(t *testing.T) {
+	cfg := &Config{Listen: "xinu://unix.sock"}
+
+	ln, err := cfg.Dialer()
+	assert.Nil(t, ln)
+	assert.Error(t, err)
+}
+
+func Test_Config_Defaults(t *testing.T) {
+	c := &Config{}
+	c.InitDefaults()
+	assert.Equal(t, "tcp://127.0.0.1:6001", c.Listen)
+}