ozon-api-client/ozon/analytics_test.go

package ozon

import (
	"context"
	"net/http"
	"testing"
	"time"

	core "github.com/diphantxm/ozon-api-client"
)

func TestGetAnalyticsData(t *testing.T) {
	t.Parallel()

	tests := []struct {
		statusCode int
		headers    map[string]string
		params     *GetAnalyticsDataParams
		response   string
	}{
		// Test ok
		{
			http.StatusOK,
			map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
			&GetAnalyticsDataParams{
				DateFrom:  core.NewTimeFormat(time.Now().Add(time.Duration(30)*24*time.Hour), core.ShortDateLayout),
				DateTo:    core.NewTimeFormat(time.Now(), core.ShortDateLayout),
				Dimension: []GetAnalyticsDataDimension{SKUDimension, DayDimension},
				Metrics:   []GetAnalyticsDataFilterMetric{HistViewPDP},
				Sort: []GetAnalyticsDataSort{
					{
						Key:   HistViewPDP,
						Order: Descending,
					},
				},
				Limit:  1000,
				Offset: 0,
			},
			`{
				"result": {
				  "data": [],
				  "totals": [
					0
				  ]
				},
				"timestamp": "2021-11-25 15:19:21"
			}`,
		},
		{
			// Test No Client-Id or Api-Key
			http.StatusUnauthorized,
			map[string]string{},
			&GetAnalyticsDataParams{},
			`{
				"code": 16,
				"message": "Client-Id and Api-Key headers are required"
			}`,
		},
	}

	for _, test := range tests {
		c := NewMockClient(core.NewMockHttpHandler(test.statusCode, test.response, test.headers))

		ctx, _ := context.WithTimeout(context.Background(), testTimeout)
		resp, err := c.Analytics().GetAnalyticsData(ctx, test.params)
		if err != nil {
			t.Error(err)
			continue
		}

		compareJsonResponse(t, test.response, &GetAnalyticsDataResponse{})

		if resp.StatusCode != test.statusCode {
			t.Errorf("got wrong status code: got: %d, expected: %d", resp.StatusCode, test.statusCode)
		}
	}
}

func TestGetStocksOnWarehouses(t *testing.T) {
	t.Parallel()

	tests := []struct {
		statusCode int
		headers    map[string]string
		params     *GetStocksOnWarehousesParams
		response   string
	}{
		// Test Ok
		{
			http.StatusOK,
			map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
			&GetStocksOnWarehousesParams{
				Limit:         1000,
				Offset:        0,
				WarehouseType: "ALL",
			},
			`{
				"result": {
				  "rows": [
					{
					  "free_to_sell_amount": 15,
					  "item_code": "my-code",
					  "item_name": "my-name",
					  "promised_amount": 12,
					  "reserved_amount": 11,
					  "sku": 12345,
					  "warehouse_name": "my-warehouse"
					}
				  ]
				}
			}`,
		},
		// Test No Client-Id or Api-Key
		{
			http.StatusUnauthorized,
			map[string]string{},
			&GetStocksOnWarehousesParams{},
			`{
				"code": 16,
				"message": "Client-Id and Api-Key headers are required"
			}`,
		},
	}

	for _, test := range tests {
		c := NewMockClient(core.NewMockHttpHandler(test.statusCode, test.response, test.headers))

		ctx, _ := context.WithTimeout(context.Background(), testTimeout)
		resp, err := c.Analytics().GetStocksOnWarehouses(ctx, test.params)
		if err != nil {
			t.Error(err)
			continue
		}

		compareJsonResponse(t, test.response, &GetStocksOnWarehousesResponse{})

		if resp.StatusCode != test.statusCode {
			t.Errorf("got wrong status code: got: %d, expected: %d", resp.StatusCode, test.statusCode)
		}

		if resp.StatusCode == http.StatusOK {
			if len(resp.Result.Rows) > int(test.params.Limit) {
				t.Errorf("Length of rows is bigger than limit")
			}
		}
	}
}

func TestGetProductTurnover(t *testing.T) {
	t.Parallel()

	tests := []struct {
		statusCode int
		headers    map[string]string
		params     *GetProductTurnoverParams
		response   string
	}{
		// Test Ok
		{
			http.StatusOK,
			map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
			&GetProductTurnoverParams{
				Limit: 1,
				SKU:   []string{"string"},
			},
			`{
				"items": [
				  {
					"ads": 0,
					"current_stock": 0,
					"idc": 0,
					"idc_grade": "GRADES_NONE"
				  }
				]
			}`,
		},
		// Test No Client-Id or Api-Key
		{
			http.StatusUnauthorized,
			map[string]string{},
			&GetProductTurnoverParams{},
			`{
				"code": 16,
				"message": "Client-Id and Api-Key headers are required"
			}`,
		},
	}

	for _, test := range tests {
		c := NewMockClient(core.NewMockHttpHandler(test.statusCode, test.response, test.headers))

		ctx, _ := context.WithTimeout(context.Background(), testTimeout)
		resp, err := c.Analytics().GetProductTurnover(ctx, test.params)
		if err != nil {
			t.Error(err)
			continue
		}

		compareJsonResponse(t, test.response, &GetProductTurnoverResponse{})

		if resp.StatusCode != test.statusCode {
			t.Errorf("got wrong status code: got: %d, expected: %d", resp.StatusCode, test.statusCode)
		}

		if resp.StatusCode == http.StatusOK {
			if len(resp.Items) > int(test.params.Limit) {
				t.Errorf("Length of items is bigger than limit")
			}
		}
	}
}