archive, keep: host -> regexp ignore list

1 files changed, 22 insertions, 2 deletions

diff --git a/archive_test.go b/archive_test.go
index 096b628..7e7d4ff 100644
--- a/archive_test.go
+++ b/archive_test.go
@@ -9,7 +9,7 @@ func TestIsArchived(t *testing.T) {
 
 	url := "http://example.com/"
 	archived, status := isArchived(url)
-	if archived != true || status != 200 {
+	if !archived || status != 200 {
 		t.Errorf("Received %t, %d; want %t, %d", archived, status, true, 200)
 	}
 }
@@ -18,11 +18,31 @@ func TestIsNotArchived(t *testing.T) {
 
 	url := "http://invalidurl.local/"
 	archived, _ := isArchived(url)
-	if archived == true {
+	if archived {
 		t.Errorf("Received %t; want %t", archived, false)
 	}
 }
 
+func TestIsIgnored(t *testing.T) {
+
+	ignoreRegex := []string{`^https?://([^/]*\.)?example\.[^/]+/`}
+	url := "https://example.com/path"
+	ignored := isIgnored(ignoreRegex, url)
+	if !ignored {
+		t.Errorf("Received %t; want %t", ignored, true)
+	}
+}
+
+func TestIsNotIgnored(t *testing.T) {
+
+	ignoreRegex := []string{`^https?://([^/]*\.)?example\.[^/]+/`}
+	url := "https://google.com/path"
+	ignored := isIgnored(ignoreRegex, url)
+	if ignored {
+		t.Errorf("Received %t; want %t", ignored, false)
+	}
+}
+
 func TestArchive200(t *testing.T) {
 
 	url := "http://example.com/"