Feature: Add tests for CreateListener

2 files changed, 19 insertions, 1 deletions

diff --git a/util/network.go b/util/network.go
index 1ceb95b0..4c393c37 100644
--- a/util/network.go
+++ b/util/network.go
@@ -10,7 +10,11 @@ import (
 func CreateListener(address string) (net.Listener, error) {
 	dsn := strings.Split(address, "://")
 	if len(dsn) != 2 {
-		return nil, errors.New("invalid socket DSN (tcp://:6001, unix://file.sock)")
+		return nil, errors.New("Invalid DSN (tcp://:6001, unix://file.sock)")
+	}
+
+	if dsn[0] != "unix" && dsn[0] != "tcp" {
+		return nil, errors.New("Invalid Protocol (tcp://:6001, unix://file.sock)")
 	}
 
 	if dsn[0] == "unix" {
diff --git a/util/network_test.go b/util/network_test.go
new file mode 100644
index 00000000..bdc7e0b7
--- /dev/null
+++ b/util/network_test.go
@@ -0,0 +1,14 @@
+package util
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestCreateListener(t *testing.T) {
+	_, err := CreateListener("unexpected dsn");
+	assert.Error(t, err, "Invalid DSN (tcp://:6001, unix://file.sock)")
+
+	_, err = CreateListener("aaa://192.168.0.1");
+	assert.Error(t, err, "Invalid Protocol (tcp://:6001, unix://file.sock)")
+}