Test url_result.go

url_result_test.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"errors"
+	"regexp"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestURLResultString(t *testing.T) {
+	for _, r := range []urlResult{
+		{"https://google.com", nil},
+		{"https://yahoo.com", errors.New("error")},
+	} {
+		p := "OK"
+
+		if r.err != nil {
+			p = "ERROR"
+		}
+
+		b, err := regexp.MatchString(p, r.String())
+
+		assert.Equal(t, nil, err)
+		assert.True(t, b)
+
+		b, err = regexp.MatchString(r.url, r.String())
+
+		assert.Equal(t, nil, err)
+		assert.True(t, b)
+	}
+}