Add pagination tests for vouchers

rdzak · Mar 19, 2020 · 8da1e30 · 8da1e30
1 parent ee246cf
commit 8da1e30
Showing 1 changed file with 174 additions and 2 deletions.
diff --git a/tests/api/pagination/test_discount.py b/tests/api/pagination/test_discount.py
@@ -4,8 +4,8 @@
 from django.utils import timezone
 from freezegun import freeze_time
 
-from saleor.discount import DiscountValueType
-from saleor.discount.models import Sale
+from saleor.discount import DiscountValueType, VoucherType
+from saleor.discount.models import Sale, Voucher
 from tests.api.utils import get_graphql_content
 
 
@@ -141,3 +141,175 @@ def test_sales_pagination_with_filtering(
     assert sales_order[0] == sales_nodes[0]["node"]["name"]
     assert sales_order[1] == sales_nodes[1]["node"]["name"]
     assert len(sales_nodes) == page_size
+
+
+@pytest.fixture
+@freeze_time("2020-03-18 12:00:00")
+def vouchers_for_pagination(db):
+    now = timezone.now()
+    return Voucher.objects.bulk_create(
+        [
+            Voucher(
+                code="Code1",
+                name="Voucher1",
+                start_date=now + timezone.timedelta(hours=4),
+                end_date=now + timezone.timedelta(hours=14),
+                discount_value_type=DiscountValueType.PERCENTAGE,
+                discount_value=Decimal("1"),
+                min_spent_amount=Decimal("10"),
+                usage_limit=10,
+                type=VoucherType.SPECIFIC_PRODUCT,
+            ),
+            Voucher(
+                code="Code2",
+                name="Voucher2",
+                end_date=now + timezone.timedelta(hours=1),
+                discount_value=Decimal("7"),
+                min_spent_amount=Decimal("10"),
+                usage_limit=1000,
+                used=10,
+                type=VoucherType.ENTIRE_ORDER,
+            ),
+            Voucher(
+                code="Code3",
+                name="Voucher3",
+                end_date=now + timezone.timedelta(hours=2),
+                discount_value_type=DiscountValueType.PERCENTAGE,
+                discount_value=Decimal("5"),
+                min_spent_amount=Decimal("120"),
+                usage_limit=100,
+                used=35,
+                type=VoucherType.ENTIRE_ORDER,
+            ),
+            Voucher(
+                code="Code4",
+                name="Voucher4",
+                end_date=now + timezone.timedelta(hours=1),
+                discount_value=Decimal("5"),
+                min_spent_amount=Decimal("50"),
+                usage_limit=100,
+                type=VoucherType.SPECIFIC_PRODUCT,
+            ),
+            Voucher(
+                code="Code15",
+                name="Voucher15",
+                start_date=now + timezone.timedelta(hours=1),
+                end_date=now + timezone.timedelta(hours=5),
+                discount_value_type=DiscountValueType.PERCENTAGE,
+                discount_value=Decimal("2"),
+                min_spent_amount=Decimal("100"),
+                usage_limit=10,
+            ),
+        ]
+    )
+
+
+QUERY_VOUCHERS_PAGINATION = """
+    query (
+        $first: Int, $last: Int, $after: String, $before: String,
+        $sortBy: VoucherSortingInput, $filter: VoucherFilterInput
+    ){
+        vouchers(
+            first: $first, last: $last, after: $after, before: $before,
+            sortBy: $sortBy, filter: $filter
+        ) {
+            edges {
+                node {
+                    name
+                }
+            }
+            pageInfo{
+                startCursor
+                endCursor
+                hasNextPage
+                hasPreviousPage
+            }
+        }
+    }
+"""
+
+
+@pytest.mark.parametrize(
+    "sort_by, vouchers_order",
+    [
+        ({"field": "CODE", "direction": "ASC"}, ["Voucher1", "Voucher15", "Voucher2"]),
+        ({"field": "CODE", "direction": "DESC"}, ["Voucher4", "Voucher3", "Voucher2"]),
+        (
+            {"field": "START_DATE", "direction": "ASC"},
+            ["Voucher2", "Voucher3", "Voucher4"],
+        ),
+        (
+            {"field": "END_DATE", "direction": "ASC"},
+            ["Voucher2", "Voucher4", "Voucher3"],
+        ),
+        ({"field": "VALUE", "direction": "ASC"}, ["Voucher1", "Voucher15", "Voucher3"]),
+        ({"field": "TYPE", "direction": "ASC"}, ["Voucher2", "Voucher3", "Voucher15"]),
+        (
+            {"field": "USAGE_LIMIT", "direction": "ASC"},
+            ["Voucher1", "Voucher15", "Voucher3"],
+        ),
+        (
+            {"field": "MINIMUM_SPENT_AMOUNT", "direction": "ASC"},
+            ["Voucher1", "Voucher2", "Voucher4"],
+        ),
+    ],
+)
+def test_vouchers_pagination_with_sorting(
+    sort_by,
+    vouchers_order,
+    staff_api_client,
+    permission_manage_discounts,
+    vouchers_for_pagination,
+):
+    page_size = 3
+
+    variables = {"first": page_size, "after": None, "sortBy": sort_by}
+    response = staff_api_client.post_graphql(
+        QUERY_VOUCHERS_PAGINATION,
+        variables,
+        permissions=[permission_manage_discounts],
+        check_no_permissions=False,
+    )
+    content = get_graphql_content(response)
+    vouchers_nodes = content["data"]["vouchers"]["edges"]
+    assert vouchers_order[0] == vouchers_nodes[0]["node"]["name"]
+    assert vouchers_order[1] == vouchers_nodes[1]["node"]["name"]
+    assert vouchers_order[2] == vouchers_nodes[2]["node"]["name"]
+    assert len(vouchers_nodes) == page_size
+
+
+@pytest.mark.parametrize(
+    "filter_by, vouchers_order",
+    [
+        ({"status": "SCHEDULED"}, ["Voucher1", "Voucher15"]),
+        ({"status": "ACTIVE"}, ["Voucher2", "Voucher3"]),
+        ({"timesUsed": {"gte": 1}}, ["Voucher2", "Voucher3"]),
+        ({"timesUsed": {"lte": 1}}, ["Voucher1", "Voucher4"]),
+        ({"discountType": "FIXED"}, ["Voucher2", "Voucher4"]),
+        ({"discountType": "PERCENTAGE"}, ["Voucher1", "Voucher3"]),
+        ({"started": {"gte": "2020-03-18T13:00:00+00:00"}}, ["Voucher1", "Voucher15"]),
+        ({"started": {"lte": "2020-03-18T13:00:00+00:00"}}, ["Voucher2", "Voucher3"]),
+    ],
+)
+@freeze_time("2020-03-18 12:15:00")
+def test_vouchers_pagination_with_filtering(
+    filter_by,
+    vouchers_order,
+    staff_api_client,
+    permission_manage_discounts,
+    vouchers_for_pagination,
+):
+    page_size = 2
+
+    variables = {"first": page_size, "after": None, "filter": filter_by}
+    response = staff_api_client.post_graphql(
+        QUERY_VOUCHERS_PAGINATION,
+        variables,
+        permissions=[permission_manage_discounts],
+        check_no_permissions=False,
+    )
+    content = get_graphql_content(response)
+    vouchers_nodes = content["data"]["vouchers"]["edges"]
+    assert vouchers_order[0] == vouchers_nodes[0]["node"]["name"]
+    assert vouchers_order[1] == vouchers_nodes[1]["node"]["name"]
+    assert len(vouchers_nodes) == page_size