add unit test to check panic in Dial and DialContext functions

1 files changed, 44 insertions, 0 deletions

diff --git a/dialer_test.go b/dialer_test.go
new file mode 100644
index 0000000..aa00a60
--- /dev/null
+++ b/dialer_test.go
@@ -0,0 +1,44 @@
+package obfsvpn
+
+import (
+	"context"
+	"testing"
+)
+
+const cert = "8nuAbPJwFrKc/29KcCfL5LBuEWxQrjBASYXdUbwcm9d9pKseGK4r2Tg47e23+t6WghxGGw"
+
+func TestDial(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Error("Dial paniced")
+		}
+	}()
+
+	dialer, err := NewDialerFromCert(cert)
+	if err != nil {
+		t.Errorf("Error creating client from cert: %v", err)
+	}
+
+	_, err = dialer.Dial("tcp", "127.0.0.1:4430")
+	if err == nil {
+		t.Errorf("No Error in d.Dial when trying invalid address: %v", err)
+	}
+}
+
+func TestDialContext(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Error("DialContext paniced")
+		}
+	}()
+
+	dialer, err := NewDialerFromCert(cert)
+	if err != nil {
+		t.Errorf("Error creating client from cert: %v", err)
+	}
+
+	_, err = dialer.DialContext(context.Background(), "tcp", "127.0.0.1:4430")
+	if err == nil {
+		t.Errorf("No Error in d.Dial when trying invalid address: %v", err)
+	}
+}