1 files changed, 51 insertions, 0 deletions
diff --git a/internal/findlinks/findlinks_test.go b/internal/findlinks/findlinks_test.go
new file mode 100644
index 0000000..79fb241
--- /dev/null
+++ b/internal/findlinks/findlinks_test.go
@@ -0,0 +1,51 @@
+package findlinks
+
+import (
+	"os"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func findLinksFile(filename string) ([]Link, error) {
+	f, err := os.Open(filename)
+	if err != nil {
+		panic("can't open test file")
+	}
+	defer f.Close()
+
+	return FindLinks(f)
+}
+
+func TestFindlinks(t *testing.T) {
+	type test struct {
+		filename string
+		links    []Link
+	}
+
+	tests := []test{
+		{
+			"html/ex1.html",
+			[]Link{
+				{
+					Href: "/other-page",
+					Text: "A link to another page",
+				},
+			},
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.filename, func(t *testing.T) {
+			links, err := findLinksFile(test.filename)
+			if err != nil {
+				t.Error(err)
+			}
+
+			if !cmp.Equal(links, test.links) {
+				t.Errorf("got %v, want %v", links, test.links)
+			}
+		})
+	}
+
+}