From beadfd1db3d2398b9b1e66d60779a7b2649af044 Mon Sep 17 00:00:00 2001
From: rafiramadhana <rf.ramadhana@gmail.com>
Date: Wed, 15 Nov 2023 22:18:06 +0700
Subject: Add multiple starting URLs

Add multiple starting URLs by adding Module.URLs.
---
 modules/starturl/starturl_test.go | 83 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)

(limited to 'modules/starturl/starturl_test.go')

diff --git a/modules/starturl/starturl_test.go b/modules/starturl/starturl_test.go
index 78efa6a..54f899a 100644
--- a/modules/starturl/starturl_test.go
+++ b/modules/starturl/starturl_test.go
@@ -38,3 +38,86 @@ func TestStartURL(t *testing.T) {
 	require.Equal(t, "http://www.example.com/foo/bar", url)
 	require.Equal(t, 0, depth)
 }
+
+func TestStartURL_MultipleStartingURLs(t *testing.T) {
+	testCases := []struct {
+		name          string
+		startURLModFn func() *starturl.Module
+		urls          []string
+	}{
+		{
+			name: ".URL and .URLs",
+			startURLModFn: func() *starturl.Module {
+				return &starturl.Module{
+					URL: "http://www.example.com/foo",
+					URLs: []string{
+						"http://www.example.com/bar",
+						"http://www.example.com/baz",
+					},
+				}
+			},
+			urls: []string{
+				"http://www.example.com/foo",
+				"http://www.example.com/bar",
+				"http://www.example.com/baz",
+			},
+		},
+		{
+			name: "only .URL",
+			startURLModFn: func() *starturl.Module {
+				return &starturl.Module{
+					URL: "http://www.example.com/foo",
+				}
+			},
+			urls: []string{
+				"http://www.example.com/foo",
+			},
+		},
+		{
+			name: "only .URLs",
+			startURLModFn: func() *starturl.Module {
+				return &starturl.Module{
+					URLs: []string{
+						"http://www.example.com/bar",
+						"http://www.example.com/baz",
+					},
+				}
+			},
+			urls: []string{
+				"http://www.example.com/bar",
+				"http://www.example.com/baz",
+			},
+		},
+		{
+			name: "empty",
+			startURLModFn: func() *starturl.Module {
+				return &starturl.Module{}
+			},
+			urls: []string{},
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			urls := []string{}
+
+			mods := []flyscrape.Module{
+				tc.startURLModFn(),
+				hook.Module{
+					AdaptTransportFn: func(rt http.RoundTripper) http.RoundTripper {
+						return flyscrape.MockTransport(http.StatusOK, "")
+					},
+					BuildRequestFn: func(r *flyscrape.Request) {
+						urls = append(urls, r.URL)
+					},
+				},
+			}
+
+			scraper := flyscrape.NewScraper()
+			scraper.Modules = mods
+			scraper.Run()
+
+			require.ElementsMatch(t, tc.urls, urls)
+		})
+	}
+}
-- 
cgit v1.2.3