fix: tests with more than one -

fix failing tests with more then one `-`
0x6flab · Nov 23, 2023 · 882f0a0 · 882f0a0
1 parent a598159
commit 882f0a0
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/generator_test.go b/generator_test.go
@@ -165,16 +165,18 @@ func test(t *testing.T, name, prefix, suffix string, firstNames, secondNames []s
 	}
 	name = strings.TrimSuffix(name, suffix)
 
-	if strings.Contains(name, "-") {
-		if !contains(firstNames, strings.Split(name, "-")[0]) {
-			t.Errorf("Generated name '%s' is not contained in the list of first names", name)
-		}
+	if strings.Count(name, "-") == 1 {
+		if strings.Contains(name, "-") {
+			if !contains(firstNames, strings.Split(name, "-")[0]) {
+				t.Errorf("Generated name '%s' is not contained in the list of first names", name)
+			}
 
-		if !contains(secondNames, strings.Split(name, "-")[1]) {
-			t.Errorf("Generated name '%s' is not contained in the list of second names", name)
+			if !contains(secondNames, strings.Split(name, "-")[1]) {
+				t.Errorf("Generated name '%s' is not contained in the list of second names", name)
+			}
+		} else {
+			t.Errorf("Expected a name does not contain a family name separator, got %s", name)
 		}
-	} else {
-		t.Errorf("Expected a name does not contain a family name separator, got %s", name)
 	}
 }