add rating methods

2023-03-14 00:30:45 +03:00
parent d1565b9d00
commit a763342d11
3 changed files with 313 additions and 0 deletions
--- a/ozon/rating_test.go
+++ b/ozon/rating_test.go
@@ -0,0 +1,125 @@
+package ozon
+
+import (
+	"net/http"
+	"testing"
+
+	core "github.com/diphantxm/ozon-api-client"
+)
+
+func TestGetCurrentRatingInfo(t *testing.T) {
+	tests := []struct {
+		statusCode int
+		headers    map[string]string
+		response   string
+	}{
+		{
+			http.StatusOK,
+			map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
+			`{
+				"groups": [
+				  {
+					"group_name": "my-group",
+					"items": [
+					  {
+						"change": {
+						  "direction": "DIRECTION_RISE",
+						  "meaning": "MEANING_GOOD"
+						},
+						"current_value": 15.2,
+						"name": "my-rating-name",
+						"past_value": 15.1,
+						"rating": "my-rating",
+						"rating_direction": "HIGHER_IS_BETTER",
+						"status": "OK",
+						"value_type": "INDEX"
+					  }
+					]
+				  }
+				],
+				"penalty_score_exceeded": true,
+				"premium": true
+			}`,
+		},
+	}
+
+	for _, test := range tests {
+		c := NewMockClient(core.NewMockHttpHandler(test.statusCode, test.response, test.headers))
+
+		resp, err := c.GetCurrentSellerRatingInfo()
+		if err != nil {
+			t.Error(err)
+		}
+
+		if resp.StatusCode != test.statusCode {
+			t.Errorf("got wrong status code: got: %d, expected: %d", resp.StatusCode, test.statusCode)
+		}
+	}
+}
+
+func TestGetRatingInfoForPeriod(t *testing.T) {
+	tests := []struct {
+		statusCode int
+		headers    map[string]string
+		params     *GetSellerRatingInfoForPeriodParams
+		response   string
+	}{
+		{
+			http.StatusOK,
+			map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
+			&GetSellerRatingInfoForPeriodParams{
+				DateFrom:          core.TimeFromString(t, "2006-01-02T15:04:05Z", "2019-08-24T14:15:22Z"),
+				DateTo:            core.TimeFromString(t, "2006-01-02T15:04:05Z", "2019-08-24T14:15:22Z"),
+				Ratings:           []string{"rating_order_accuracy", "rating_shipment_delay"},
+				WithPremiumScores: true,
+			},
+			`{
+				"premium_scores": [
+				  {
+					"rating": "string",
+					"scores": [
+					  {
+						"date": "2019-08-24T14:15:22Z",
+						"rating_value": 5,
+						"value": 5
+					  }
+					]
+				  }
+				],
+				"ratings": [
+				  {
+					"danger_threshold": 1,
+					"premium_threshold": 2,
+					"rating": "my-rating-system",
+					"values": [
+					  {
+						"date_from": "2019-08-24T14:15:22Z",
+						"date_to": "2019-08-24T14:15:22Z",
+						"status": {
+						  "danger": true,
+						  "premium": true,
+						  "warning": true
+						},
+						"value": 12
+					  }
+					],
+					"warning_threshold": 3
+				  }
+				]
+			}`,
+		},
+	}
+
+	for _, test := range tests {
+		c := NewMockClient(core.NewMockHttpHandler(test.statusCode, test.response, test.headers))
+
+		resp, err := c.GetSellerRatingInfoForPeriod(test.params)
+		if err != nil {
+			t.Error(err)
+		}
+
+		if resp.StatusCode != test.statusCode {
+			t.Errorf("got wrong status code: got: %d, expected: %d", resp.StatusCode, test.statusCode)
+		}
+	}
+}