35 lines
810 B
Go
35 lines
810 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_responseCodeForSource(t *testing.T) {
|
||
|
|
for _, code := range []int{200, 404} {
|
||
|
|
t.Run(strconv.Itoa(code), func(t *testing.T) {
|
||
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(code)
|
||
|
|
}))
|
||
|
|
defer ts.Close()
|
||
|
|
got, err := responseCodeForSource(ts.URL)
|
||
|
|
if err != nil || got != code {
|
||
|
|
t.Errorf("Wrong response code: Got %d, but expected %d", got, code)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
t.Run("Error", func(t *testing.T) {
|
||
|
|
ts := httptest.NewUnstartedServer(nil)
|
||
|
|
defer ts.Close()
|
||
|
|
got, err := responseCodeForSource(ts.URL)
|
||
|
|
if err == nil {
|
||
|
|
t.Errorf("Error is nil")
|
||
|
|
}
|
||
|
|
if got != 0 {
|
||
|
|
t.Errorf("Wrong response code: Got %d, but expected %d", got, 0)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|