Add test for multiple files

pabateman · Nov 6, 2024 · 0209303 · 0209303
1 parent b16cdef
commit 0209303
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/internal/lookuper/parsers_test.go b/internal/lookuper/parsers_test.go
@@ -25,6 +25,28 @@ func TestParserBasic(t *testing.T) {
 	require.True(t, reflect.DeepEqual(input.UnparsedNames, expected.UnparsedNames))
 }
 
+func TestParserMultipleFiles(t *testing.T) {
+	input := newDomainNames()
+	err := input.parseFile(path.Join(testDataPath, "lists/1.lst"))
+	require.NoError(t, err)
+	err = input.parseFile(path.Join(testDataPath, "lists/2.lst"))
+	require.NoError(t, err)
+
+	expected := newDomainNames()
+	expected.ParsedNames = []string{
+		"cloudflare.com",
+		"google.com",
+		"hashicorp.com",
+		"linked.in",
+		"releases.hashicorp.com",
+		"rpm.releases.hashicorp.com",
+		"terraform.io",
+	}
+
+	require.True(t, reflect.DeepEqual(input.ParsedNames, expected.ParsedNames))
+	require.True(t, reflect.DeepEqual(input.UnparsedNames, expected.UnparsedNames))
+}
+
 func TestParserEmpty(t *testing.T) {
 	input := newDomainNames()
 	err := input.parseFile(path.Join(testDataPath, "lists/empty.lst"))