172 lines
3.8 KiB
Go
172 lines
3.8 KiB
Go
package ozon
|
||
|
||
import (
|
||
"context"
|
||
"net/http"
|
||
"testing"
|
||
|
||
core "github.com/diphantxm/ozon-api-client"
|
||
)
|
||
|
||
func TestGetListOfWarehouses(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
tests := []struct {
|
||
statusCode int
|
||
headers map[string]string
|
||
response string
|
||
}{
|
||
// Test Ok
|
||
{
|
||
http.StatusOK,
|
||
map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
|
||
`{
|
||
"result": [
|
||
{
|
||
"warehouse_id": 15588127982000,
|
||
"name": "Proffi (Панорама Групп)",
|
||
"is_rfbs": false
|
||
},
|
||
{
|
||
"warehouse_id": 22142605386000,
|
||
"name": "Склад на производственной",
|
||
"is_rfbs": true
|
||
},
|
||
{
|
||
"warehouse_id": 22208673494000,
|
||
"name": "Тест 37349",
|
||
"is_rfbs": true
|
||
},
|
||
{
|
||
"warehouse_id": 22240462819000,
|
||
"name": "Тест12",
|
||
"is_rfbs": true
|
||
}
|
||
]
|
||
}`,
|
||
},
|
||
// Test No Client-Id or Api-Key
|
||
{
|
||
http.StatusUnauthorized,
|
||
map[string]string{},
|
||
`{
|
||
"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.Warehouses().GetListOfWarehouses(ctx)
|
||
if err != nil {
|
||
t.Error(err)
|
||
continue
|
||
}
|
||
|
||
compareJsonResponse(t, test.response, &GetListOfWarehousesResponse{})
|
||
|
||
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) > 0 {
|
||
if resp.Result[0].WarehouseId == 0 {
|
||
t.Errorf("Warehouse id cannot be 0")
|
||
}
|
||
if resp.Result[0].Name == "" {
|
||
t.Errorf("Name cannot be empty")
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestGetListOfDeliveryMethods(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
tests := []struct {
|
||
statusCode int
|
||
headers map[string]string
|
||
params *GetListOfDeliveryMethodsParams
|
||
response string
|
||
}{
|
||
// Test Ok
|
||
{
|
||
http.StatusOK,
|
||
map[string]string{"Client-Id": "my-client-id", "Api-Key": "my-api-key"},
|
||
&GetListOfDeliveryMethodsParams{
|
||
Filter: GetListOfDeliveryMethodsFilter{
|
||
WarehouseId: 15588127982000,
|
||
},
|
||
Limit: 100,
|
||
Offset: 0,
|
||
},
|
||
`{
|
||
"result": [
|
||
{
|
||
"id": 15588127982000,
|
||
"company_id": 1,
|
||
"name": "Ozon Логистика курьеру, Есипово",
|
||
"status": "ACTIVE",
|
||
"cutoff": "13:00",
|
||
"provider_id": 24,
|
||
"template_id": 0,
|
||
"warehouse_id": 15588127982000,
|
||
"created_at": "2019-04-04T15:22:31.048202Z",
|
||
"updated_at": "2021-08-15T10:21:44.854209Z"
|
||
}
|
||
],
|
||
"has_next": false
|
||
}`,
|
||
},
|
||
// Test No Client-Id or Api-Key
|
||
{
|
||
http.StatusUnauthorized,
|
||
map[string]string{},
|
||
&GetListOfDeliveryMethodsParams{},
|
||
`{
|
||
"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.Warehouses().GetListOfDeliveryMethods(ctx, test.params)
|
||
if err != nil {
|
||
t.Error(err)
|
||
continue
|
||
}
|
||
|
||
compareJsonResponse(t, test.response, &GetListOfDeliveryMethodsResponse{})
|
||
|
||
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) > 0 {
|
||
if resp.Result[0].Id == 0 {
|
||
t.Errorf("Id cannot be 0")
|
||
}
|
||
if resp.Result[0].Name == "" {
|
||
t.Errorf("Name cannot be empty")
|
||
}
|
||
if resp.Result[0].Status == "" {
|
||
t.Errorf("Status cannot be empty")
|
||
}
|
||
if resp.Result[0].WarehouseId == 0 {
|
||
t.Errorf("Warehouse id cannot be 0")
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|