Function executeTwoConcurrentRequests

main_test.go:1196–1249

executeTwoConcurrentRequests concurrently executes 2 requests for the same query. Results are asserted according to the specified input parameters.

(t *testing.T, query string, firstStatusCode, secondStatusCode int,
	firstBody, secondBody string)

// executeTwoConcurrentRequests concurrently executes 2 requests for the same query.
// Results are asserted according to the specified input parameters.
func executeTwoConcurrentRequests(t *testing.T, query string, firstStatusCode, secondStatusCode int,
	firstBody, secondBody string) {
	u := fmt.Sprintf("http://127.0.0.1:9090?query=%s&user=concurrent_user", url.QueryEscape(query))

	var wg sync.WaitGroup
	wg.Add(2)
	var resp1 string
	var resp2 string
	errs := make(chan error, 0)
	defer close(errs)
	errors := make([]error, 0)
	go func() {
		for err := range errs {
			errors = append(errors, err)
		}
	}()
	go func() {
		defer wg.Done()
		req, err := http.NewRequest("GET", u, nil)
		checkErr(t, err)
		resp, err := httpRequest(t, req, firstStatusCode)
		if err != nil {
			errs <- err
			return
		}
		resp1 = bbToString(t, resp.Body)
	}()

	go func() {
		defer wg.Done()
		time.Sleep(20 * time.Millisecond)
		req, err := http.NewRequest("GET", u, nil)
		checkErr(t, err)
		resp, err := httpRequest(t, req, secondStatusCode)
		if err != nil {
			errs <- err
			return
		}
		resp2 = bbToString(t, resp.Body)
	}()
	wg.Wait()

	if len(errors) != 0 {
		t.Fatalf("concurrent test scenario failed due to: %v", errors)
	}

	if !strings.Contains(resp1, firstBody) {
		t.Fatalf("concurrent test scenario: unexpected resp body: %s, expected : %s", resp1, firstBody)
	}

	if !strings.Contains(resp2, secondBody) {
		t.Fatalf("concurrent test scenario: unexpected resp body: %s, expected : %s", resp2, secondBody)
	}
}

var bytesWithInvalidUTFPairs = []byte{239, 191, 189, 1, 32, 50, 239, 191}