Skip to content

Commit 811b1db

Browse files
Add more robust tests for graphql_offers_for_countries_query
1 parent 8c835e3 commit 811b1db

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

test/simplejustwatchapi/test_graphql.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from pytest import mark
2+
13
from simplejustwatchapi.graphql import (
24
graphql_details_query,
35
graphql_episodes_query,
@@ -278,8 +280,8 @@ def test_graphql_episodes_query():
278280
assert expected_query == query
279281

280282

281-
def test_graphql_offers_for_countries_query():
282-
country_codes = {"gb", "Us", "fR", "CA"}
283+
@mark.parametrize("country_codes", [{"gb", "Us", "fR", "CA"}, {"us"}, set()])
284+
def test_graphql_offers_for_countries_query(country_codes):
283285
offer_requests = [
284286
GRAPHQL_COUNTRY_OFFERS_ENTRY.format(country_code=country_code.upper())
285287
for country_code in country_codes

0 commit comments

Comments
 (0)