diff --git a/cmd/web_test.go b/cmd/web_test.go
index 3852487..79972a9 100644
--- a/cmd/web_test.go
+++ b/cmd/web_test.go
@@ -2,7 +2,7 @@ package cmd
import (
"testing"
- "strings"
+
"github.com/tnypxl/rollup/internal/config"
)
@@ -67,9 +67,9 @@ func TestSanitizeFilename(t *testing.T) {
func TestGetFilenameFromContent(t *testing.T) {
tests := []struct {
- content string
- url string
- expected string
+ content string
+ url string
+ expected string
expectErr bool
}{
{"
Test Page", "http://example.com", "Test_Page.rollup.md", false},
@@ -102,53 +102,3 @@ func TestGetFilenameFromContent(t *testing.T) {
func mockExtractAndConvertContent(urlStr string) (string, error) {
return "Mocked content for " + urlStr, nil
}
-
-func mockExtractLinks() ([]string, error) {
- return []string{"http://example.com/link1", "http://example.com/link2"}, nil
-}
-
-func TestScrapeURL(t *testing.T) {
- // Store the original functions
- originalExtractAndConvertContent := testExtractAndConvertContent
- originalExtractLinks := testExtractLinks
-
- // Define mock functions
- testExtractAndConvertContent = func(urlStr string) (string, error) {
- return "Mocked content for " + urlStr, nil
- }
- testExtractLinks = func(urlStr string) ([]string, error) {
- return []string{"http://example.com/link1", "http://example.com/link2"}, nil
- }
-
- // Defer the restoration of original functions
- defer func() {
- testExtractAndConvertContent = originalExtractAndConvertContent
- testExtractLinks = originalExtractLinks
- }()
-
- tests := []struct {
- url string
- depth int
- expectedCalls int
- }{
- {"http://example.com", 0, 1},
- {"http://example.com", 1, 3},
- {"http://example.com", 2, 3}, // Same as depth 1 because our mock only returns 2 links
- }
-
- for _, test := range tests {
- visited := make(map[string]bool)
- content, err := scrapeURL(test.url, test.depth, visited)
- if err != nil {
- t.Errorf("scrapeURL(%q, %d) returned error: %v", test.url, test.depth, err)
- continue
- }
- if len(visited) != test.expectedCalls {
- t.Errorf("scrapeURL(%q, %d) made %d calls, expected %d", test.url, test.depth, len(visited), test.expectedCalls)
- }
- expectedContent := "Mocked content for " + test.url
- if !strings.Contains(content, expectedContent) {
- t.Errorf("scrapeURL(%q, %d) content doesn't contain %q", test.url, test.depth, expectedContent)
- }
- }
-}