Include 2 letter query param key in unit test

golang-migrate · Dec 8, 2021 · 169d0fc · 169d0fc
1 parent 4f82413
commit 169d0fc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/util_test.go b/util_test.go
@@ -21,12 +21,15 @@ func TestSuint(t *testing.T) {
 }
 
 func TestFilterCustomQuery(t *testing.T) {
-	n, err := nurl.Parse("foo://host?a=b&x-custom=foo&c=d")
+	n, err := nurl.Parse("foo://host?a=b&x-custom=foo&c=d&ok=y")
 	if err != nil {
 		t.Fatal(err)
 	}
 	nx := FilterCustomQuery(n).Query()
 	if nx.Get("x-custom") != "" {
 		t.Fatalf("didn't expect x-custom")
 	}
+	if nx.Get("ok") != "y" {
+		t.Fatalf("expected ok=y, got %v", nx.Get("ok"))
+	}
 }