1 files changed, 24 insertions, 0 deletions
diff --git a/internal/proxy/certdownload_test.go b/internal/proxy/certdownload_test.go
new file mode 100644
index 0000000..aae5f48
--- /dev/null
+++ b/internal/proxy/certdownload_test.go
@@ -0,0 +1,24 @@
+package proxy
+
+import "testing"
+
+func TestIsCertDownloadHost(t *testing.T) {
+	tests := []struct {
+		host string
+		want bool
+	}{
+		{"mitmux.cert", true},
+		{"mitmux.cert:80", true},
+		{"MITMUX.CERT", true},
+		{"Mitmux.Cert:8080", true},
+		{"example.com", false},
+		{"notmitmux.cert", false},
+		{"mitmux.cert.evil.com", false},
+		{"", false},
+	}
+	for _, tt := range tests {
+		if got := isCertDownloadHost(tt.host); got != tt.want {
+			t.Errorf("isCertDownloadHost(%q) = %v, want %v", tt.host, got, tt.want)
+		}
+	}
+}