From 4fdc39884f6d083e4266459b550a538c3229cae4 Mon Sep 17 00:00:00 2001
From: Evgeny Kuznetsov <evgeny@kuznetsov.md>
Date: Thu, 23 Mar 2023 12:30:08 +0300
Subject: feat: add target check

Part of https://www.w3.org/TR/webmention/#request-verification
---
 main_test.go | 42 ++++++++++++++++++++++++++++--------------
 1 file changed, 28 insertions(+), 14 deletions(-)

(limited to 'main_test.go')

diff --git a/main_test.go b/main_test.go
index d1b864d..0510af8 100644
--- a/main_test.go
+++ b/main_test.go
@@ -9,22 +9,36 @@ import (
 )
 
 func TestSyncRejection(t *testing.T) {
-	server := httptest.NewServer(endpoint{"my.site"})
+	server := httptest.NewServer(endpoint{"my.site/part"})
 	defer server.Close()
 
-	client := http.DefaultClient
-	r, err := client.PostForm(server.URL, url.Values{
-		"source": []string{"https||:example.org/somewhere"},
-		"target": []string{"my.site/target"},
-	})
-	if err != nil {
-		t.Fatal(err)
+	tests := []struct {
+		name   string
+		source string
+		target string
+		expect string
+	}{
+		{"invalid source", "https||:example.org/somewhere", "my.site/part/target", errSrcInvalid},
+		{"target no accepted", "https://example.org/somewhere", "wrong.site/tgt", errTgtNotAccepted},
 	}
-	if r.StatusCode != 400 {
-		t.Fatalf("want 400, got %v", r.Status)
-	}
-	bb, _ := io.ReadAll(r.Body)
-	if string(bb) != errSrcInvalid {
-		t.Fatalf("want %s, got %s", errSrcInvalid, string(bb))
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			client := http.DefaultClient
+			r, err := client.PostForm(server.URL, url.Values{
+				"source": []string{tc.source},
+				"target": []string{tc.target},
+			})
+			if err != nil {
+				t.Fatal(err)
+			}
+			if r.StatusCode != 400 {
+				t.Fatalf("want 400, got %v", r.Status)
+			}
+			bb, _ := io.ReadAll(r.Body)
+			if string(bb) != tc.expect {
+				t.Fatalf("want %s, got %s", tc.expect, string(bb))
+			}
+		})
 	}
 }
-- 
cgit v1.2.3