v2_advanced_query_test.go

package test

import (
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/tidwall/gjson"
)

func TestAPIV2AdvancedQuery(t *testing.T) {
	startup(t)
	t.Parallel()

	query := func(body string, headers *http.Header) (*http.Response, *gjson.Result) {
		t.Helper()
		return JsonRequest(t, "/PenguinStats/api/v2/result/advanced", body, headers, 10000)
	}

	t.Run("valid body", func(t *testing.T) {
		t.Run("with non-zero endTime", func(t *testing.T) {
			h, j := query(`{"queries":[{"stageId":"main_01-07","itemIds":[],"server":"CN","isPersonal":false,"sourceCategory":"all","start":1556668800000,"end":1562630400000}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})

		t.Run("with null endTime (until now)", func(t *testing.T) {
			h, j := query(`{"queries":[{"stageId":"main_01-07","itemIds":[],"server":"CN","isPersonal":false,"sourceCategory":"all","start":1556668800000,"end":null}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})

		t.Run("with undefined endTime (until now)", func(t *testing.T) {
			h, j := query(`{"queries":[{"stageId":"main_01-07","itemIds":[],"server":"CN","isPersonal":false,"sourceCategory":"all","start":1556668800000}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})

		t.Run("previous 500 from sentry 1", func(t *testing.T) {
			h, j := query(`{"queries":[{"end":1578268800000,"isPersonal":false,"itemIds":["ap_supply_lt_010"],"server":"CN","sourceCategory":"automated","stageId":"main_01-07","start":1577145600000},{"end":1590624000000,"isPersonal":false,"itemIds":["ap_supply_lt_010"],"server":"CN","sourceCategory":"automated","stageId":"main_01-07","start":1589500800000},{"end":1582243200000,"isPersonal":false,"itemIds":["ap_supply_lt_010"],"server":"CN","sourceCategory":"automated","stageId":"main_01-07","start":1581033600000}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})

		t.Run("previous 500 from sentry 2", func(t *testing.T) {
			h, j := query(`{"queries":[{"end":1582243200000,"isPersonal":false,"itemIds":["ap_supply_lt_010"],"server":"CN","sourceCategory":"automated","stageId":"main_01-07","start":1581033600000}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})
	})

	t.Run("invalid body", func(t *testing.T) {
		t.Run("with zero endTime", func(t *testing.T) {
			h, j := query(`{"queries":[{"stageId":"main_01-07","itemIds":[],"server":"CN","isPersonal":false,"sourceCategory":"all","start":1556668800000,"end":0}]}`, nil)
			assert.Equal(t, http.StatusOK, h.StatusCode)
			assert.NotEmpty(t, len(j.Get("advanced_results").String()))
		})
	})
}