add test for slice

Signed-off-by: yanxuean <yan.xuean@zte.com.cn>
rikatz · Nov 16, 2017 · 562cacc · 562cacc
1 parent fe4da33
commit 562cacc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/slice_test.go b/slice_test.go
@@ -89,3 +89,20 @@ func TestShuffleStrings(t *testing.T) {
 		}
 	}
 }
+
+func TestContainsString(t *testing.T) {
+	src := []string{"aa", "bb", "cc"}
+	if !ContainsString(src, "bb", nil) {
+		t.Errorf("ContainsString didn't find the string as expected")
+	}
+
+	modifier := func(s string) string {
+		if s == "cc" {
+			return "ee"
+		}
+		return s
+	}
+	if !ContainsString(src, "ee", modifier) {
+		t.Errorf("ContainsString didn't find the string by modifier")
+	}
+}