From 81f233c226f632fe83423812a6d9c02ccfb41161 Mon Sep 17 00:00:00 2001
From: Qian Wang <cbeuw.andy@gmail.com>
Date: Wed, 7 Aug 2019 00:19:30 +0100
Subject: [PATCH] Update obfs_test

---
 internal/multiplex/obfs_test.go | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/internal/multiplex/obfs_test.go b/internal/multiplex/obfs_test.go
index 36f1c45..c685d5c 100644
--- a/internal/multiplex/obfs_test.go
+++ b/internal/multiplex/obfs_test.go
@@ -39,24 +39,38 @@ func TestGenerateObfs(t *testing.T) {
 		obfuscator, err := GenerateObfs(0x00, sessionKey)
 		if err != nil {
 			t.Errorf("failed to generate obfuscator %v", err)
+		} else {
+			run(obfuscator, t)
 		}
-		run(obfuscator, t)
 	})
 	t.Run("aes-gcm", func(t *testing.T) {
 		obfuscator, err := GenerateObfs(0x01, sessionKey)
 		if err != nil {
 			t.Errorf("failed to generate obfuscator %v", err)
+		} else {
+			run(obfuscator, t)
 		}
-		run(obfuscator, t)
 	})
 	t.Run("chacha20-poly1305", func(t *testing.T) {
 		obfuscator, err := GenerateObfs(0x02, sessionKey)
 		if err != nil {
 			t.Errorf("failed to generate obfuscator %v", err)
+		} else {
+			run(obfuscator, t)
+		}
+	})
+	t.Run("unknown encryption method", func(t *testing.T) {
+		_, err := GenerateObfs(0xff, sessionKey)
+		if err == nil {
+			t.Errorf("unknown encryption mehtod error expected")
+		}
+	})
+	t.Run("bad key length", func(t *testing.T) {
+		_, err := GenerateObfs(0xff, sessionKey[:31])
+		if err == nil {
+			t.Errorf("bad key length error expected")
 		}
-		run(obfuscator, t)
 	})
-
 }
 
 func BenchmarkObfs(b *testing.B) {