Function TestDNSPerf

perfops/run_test.go:221–275

(t *testing.T)

}

func TestDNSPerf(t *testing.T) {
	reqTestCases := map[string]struct {
		dnsPerfReq DNSPerfRequest
		tr         *recordingTransport
		expReqBody string
	}{
		"Required only": {DNSPerfRequest{Target: "example.com"}, &recordingTransport{}, `{"target":"example.com"}`},
		"With server":   {DNSPerfRequest{Target: "example.com", DNSServer: "127.0.0.1"}, &recordingTransport{}, `{"target":"example.com","dnsServer":"127.0.0.1"}`},
		"With location": {DNSPerfRequest{Target: "example.com", Location: "Asia"}, &recordingTransport{}, `{"target":"example.com","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.DNSPerf(ctx, &tc.dnsPerfReq)
			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
		dnsServer string
		limit     int
		testID    string
		tr        *respondingTransport
		err       error
	}{
		"Invalid target":     {"meep", "127.0.0.1", 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "135"}`)}, &argError{"target"}},
		"Invalid DNS server": {"example.com", "127.0", 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "135"}`)}, &argError{"dns server"}},
		"Invalid limit":      {"example.com", "127.0.0.1", freeMaxNodeCap + 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"id": "135"}`)}, &argError{"limit"}},
		"HTTP error":         {"example.com", "127.0.0.1", 1, "", &respondingTransport{resp: dummyResp(400, "POST", `{"Error": "an error"}`)}, errors.New(`400: {"Error": "an error"}`)},
		"Failed":             {"example.com", "127.0.0.1", 1, "", &respondingTransport{resp: dummyResp(201, "POST", `{"Error": "an error"}`)}, errors.New("an error")},
		"Created":            {"example.com", "127.0.0.1", 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.DNSPerf(ctx, &DNSPerfRequest{Target: tc.target, DNSServer: tc.dnsServer, 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)
			}
		})
	}
}

func TestDNSPerfOutput(t *testing.T) {
	testCases := map[string]struct {