Test file checker's Check method

8 years ago · a88bac502a
3 changed files with 53 additions and 0 deletions
--- a/file_checker_test.go
+++ b/file_checker_test.go
@ -9,6 +9,44 @@ import (
 	"golang.org/x/net/html"
 )

+func TestFileCheckerCheck(t *testing.T) {
+	c := newFileChecker(0, "", newSemaphore(1024))
+
+	for _, f := range []string{"README.md", "test/foo.md", "test/foo.html"} {
+		rs, err := c.Check(f)
+
+		assert.NotEqual(t, 0, len(rs))
+		assert.Equal(t, nil, err)
+
+		for _, r := range rs {
+			assert.Equal(t, nil, r.err)
+		}
+	}
+
+	for _, f := range []string{"READYOU.md", "test"} {
+		rs, err := c.Check(f)
+
+		assert.Equal(t, ([]urlResult)(nil), rs)
+		assert.NotEqual(t, nil, err)
+	}
+
+	for _, f := range []string{"test/bad.md", "test/bad.html"} {
+		rs, err := c.Check(f)
+
+		assert.Equal(t, nil, err)
+
+		ok := true
+
+		for _, r := range rs {
+			if r.err != nil {
+				ok = false
+			}
+		}
+
+		assert.False(t, ok)
+	}
+}
+
 func TestFileCheckerExtractURLs(t *testing.T) {
 	c := newFileChecker(0, "", newSemaphore(42))

--- a/test/bad.html
+++ b/test/bad.html
@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html>
+
+<head>
+	<title>My title</title>
+</head>
+
+<body>
+	<div>
+		<a href="http://malicious-harm-harm-url.com">Google</a>
+	</div>
+</body>
+
+</html>
--- a/test/bad.md
+++ b/test/bad.md
@ -0,0 +1 @@
+[Ha-ha!](http://laksjdflkajsdlkfjaslkfdj)