doc: add comments

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()