Function TestCurl

perfops/run_test.go:422–479

(t *testing.T)

}

func TestCurl(t *testing.T) {
	reqTestCases := map[string]struct {
		curlReq    CurlRequest
		tr         *recordingTransport
		expReqBody string
	}{
		"Required only": {CurlRequest{Target: "example.com"}, &recordingTransport{}, `{"target":"example.com","head":false}`},
		"With head":     {CurlRequest{Target: "example.com", Head: true}, &recordingTransport{}, `{"target":"example.com","head":true}`},
		"With insecure": {CurlRequest{Target: "example.com", Insecure: true}, &recordingTransport{}, `{"target":"example.com","head":false,"insecure":true}`},
		"With HTTP2":    {CurlRequest{Target: "example.com", HTTP2: true}, &recordingTransport{}, `{"target":"example.com","head":false,"http2":true}`},
		"With location": {CurlRequest{Target: "example.com", Location: "Asia"}, &recordingTransport{}, `{"target":"example.com","head":false,"location":"Asia"}`},
	}
	ctx := context.Background()
	for name, tc := range reqTestCases {
		t.Run(name, func(t *testing.T) {
			c, err := newTestClient(tc.tr)
			if err != nil {
				t.Fatalf("unexpected error %v", err)
			}
			c.Run.Curl(ctx, &tc.curlReq)
			if got := reqBody(tc.tr.req); tc.expReqBody != "" && tc.expReqBody != got {
				t.Fatalf("expected %v; got %v", tc.expReqBody, got)
			}
		})
	}

	testCases := map[string]struct {
		target   string
		head     bool
		insecure bool
		http2    bool
		limit    int
		testID   string
		tr       *respondingTransport
		err      error
	}{
		"Invalid target": {"meep", true, true, false, 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "135"}`)}, &argError{"target"}},
		"Invalid limit":  {"example.com", true, true, false, freeMaxNodeCap + 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "135"}`)}, &argError{"limit"}},
		"HTTP error":     {"example.com", true, true, false, 1, "", &respondingTransport{resp: dummyResp(400, "POST", `{"Error": "an error"}`)}, errors.New(`400: {"Error": "an error"}`)},
		"Failed":         {"example.com", true, true, false, 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"Error": "an error"}`)}, errors.New("an error")},
		"Created":        {"example.com", true, true, false, 1, "0123456789abcdefghij", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "0123456789abcdefghij"}`)}, nil},
	}
	for name, tc := range testCases {
		t.Run(name, func(t *testing.T) {
			c, err := newTestClient(tc.tr)
			if err != nil {
				t.Fatalf("unexpected error %v", err)
			}
			got, err := c.Run.Curl(ctx, &CurlRequest{Target: tc.target, Head: tc.head, Insecure: tc.insecure, HTTP2: tc.http2, Limit: tc.limit})
			if !cmpError(err, tc.err) {
				t.Fatalf("expected %v; got %v", tc.err, err)
			}
			if string(got) != tc.testID {
				t.Fatalf("expected %v; got %v", tc.testID, got)
			}
		})
	}
}