diff --git a/hcloud/exp/mockutils/mockutils.go b/hcloud/exp/mockutils/mockutils.go index 8c22fd8e..486bfc36 100644 --- a/hcloud/exp/mockutils/mockutils.go +++ b/hcloud/exp/mockutils/mockutils.go @@ -43,10 +43,14 @@ func Handler(t *testing.T, requests []Request) http.HandlerFunc { } expected := requests[index] - require.Equal(t, - expected.Method+" "+expected.Path, - r.Method+" "+r.RequestURI, - ) + + expectedCall := expected.Method + foundCall := r.Method + if expected.Path != "" { + expectedCall += " " + expected.Path + foundCall += " " + r.RequestURI + } + require.Equal(t, expectedCall, foundCall) if expected.Want != nil { expected.Want(t, r) diff --git a/hcloud/exp/mockutils/mockutils_test.go b/hcloud/exp/mockutils/mockutils_test.go index 70f79058..63fe0a39 100644 --- a/hcloud/exp/mockutils/mockutils_test.go +++ b/hcloud/exp/mockutils/mockutils_test.go @@ -1,7 +1,9 @@ package mockutils import ( + "fmt" "io" + "math/rand" "net/http" "net/http/httptest" "strings" @@ -31,6 +33,13 @@ func TestHandler(t *testing.T) { Method: "GET", Path: "/", Status: 503, }, + { + Method: "GET", + Want: func(t *testing.T, r *http.Request) { + require.True(t, strings.HasPrefix(r.RequestURI, "/random?key=")) + }, + Status: 200, + }, })) defer server.Close() @@ -54,6 +63,13 @@ func TestHandler(t *testing.T) { assert.Equal(t, 503, resp.StatusCode) assert.Equal(t, "", resp.Header.Get("Content-Type")) assert.Equal(t, "", readBody(t, resp)) + + // Request 4 + resp, err = http.Get(fmt.Sprintf("%s/random?key=%d", server.URL, rand.Int63())) + require.NoError(t, err) + assert.Equal(t, 200, resp.StatusCode) + assert.Equal(t, "", resp.Header.Get("Content-Type")) + assert.Equal(t, "", readBody(t, resp)) } func readBody(t *testing.T, resp *http.Response) string {