diff options
Diffstat (limited to 'main_test.go')
| -rw-r--r-- | main_test.go | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/main_test.go b/main_test.go index 29a8341..e231051 100644 --- a/main_test.go +++ b/main_test.go @@ -43,7 +43,11 @@ https://example.com/2`, for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - w.Write([]byte(tt.content)) + _, err := w.Write([]byte(tt.content)) + if err != nil { + t.Errorf("Couldn't write to content %v", err) + } + })) defer ts.Close() @@ -111,7 +115,11 @@ local-zone: "domain2.com" refuse for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - w.Write([]byte(tt.content)) + _, err := w.Write([]byte(tt.content)) + if err != nil { + t.Errorf("Couldn't write to content %v", err) + } + })) defer ts.Close() |