|
1 | 1 | import json |
| 2 | + |
2 | 3 | import pytest |
3 | 4 | from aioresponses import aioresponses |
4 | 5 |
|
|
20 | 21 | MOCK_FEATURE_RESPONSE = {"version": 1, "features": []} |
21 | 22 | PROJECT_URL = f"{FULL_FEATURE_URL}?project={PROJECT_NAME}" |
22 | 23 |
|
| 24 | + |
23 | 25 | @pytest.mark.asyncio |
24 | 26 | async def test_get_feature_toggles_success(): |
25 | 27 | with aioresponses() as m: |
26 | | - m.get(FULL_FEATURE_URL, status=200, payload=MOCK_FEATURE_RESPONSE, headers={"etag": ETAG_VALUE}) |
| 28 | + m.get( |
| 29 | + FULL_FEATURE_URL, |
| 30 | + status=200, |
| 31 | + payload=MOCK_FEATURE_RESPONSE, |
| 32 | + headers={"etag": ETAG_VALUE}, |
| 33 | + ) |
27 | 34 |
|
28 | 35 | body, etag = await get_feature_toggles_async( |
29 | | - URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, |
30 | | - REQUEST_TIMEOUT, REQUEST_RETRIES |
| 36 | + URL, |
| 37 | + APP_NAME, |
| 38 | + INSTANCE_ID, |
| 39 | + CUSTOM_HEADERS, |
| 40 | + CUSTOM_OPTIONS, |
| 41 | + REQUEST_TIMEOUT, |
| 42 | + REQUEST_RETRIES, |
31 | 43 | ) |
32 | 44 |
|
33 | 45 | assert json.loads(body)["version"] == 1 |
34 | 46 | assert etag == ETAG_VALUE |
35 | 47 |
|
| 48 | + |
36 | 49 | @pytest.mark.asyncio |
37 | 50 | async def test_get_feature_toggles_project_and_etag_present(): |
38 | 51 | with aioresponses() as m: |
39 | 52 | m.get(PROJECT_URL, status=304, headers={"etag": ETAG_VALUE}) |
40 | 53 |
|
41 | 54 | body, etag = await get_feature_toggles_async( |
42 | | - URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, |
43 | | - REQUEST_TIMEOUT, REQUEST_RETRIES, project=PROJECT_NAME, cached_etag=ETAG_VALUE |
| 55 | + URL, |
| 56 | + APP_NAME, |
| 57 | + INSTANCE_ID, |
| 58 | + CUSTOM_HEADERS, |
| 59 | + CUSTOM_OPTIONS, |
| 60 | + REQUEST_TIMEOUT, |
| 61 | + REQUEST_RETRIES, |
| 62 | + project=PROJECT_NAME, |
| 63 | + cached_etag=ETAG_VALUE, |
44 | 64 | ) |
45 | 65 |
|
46 | 66 | assert body is None |
47 | 67 | assert etag == ETAG_VALUE |
48 | 68 |
|
| 69 | + |
49 | 70 | @pytest.mark.asyncio |
50 | 71 | async def test_get_feature_toggles_retries_then_success(): |
51 | 72 | with aioresponses() as m: |
52 | 73 | m.get(PROJECT_URL, status=500) # first attempt |
53 | | - m.get(PROJECT_URL, status=200, payload=MOCK_FEATURE_RESPONSE, headers={"etag": ETAG_VALUE}) |
| 74 | + m.get( |
| 75 | + PROJECT_URL, |
| 76 | + status=200, |
| 77 | + payload=MOCK_FEATURE_RESPONSE, |
| 78 | + headers={"etag": ETAG_VALUE}, |
| 79 | + ) |
54 | 80 |
|
55 | 81 | body, etag = await get_feature_toggles_async( |
56 | | - URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, |
57 | | - REQUEST_TIMEOUT, request_retries=1, project=PROJECT_NAME, cached_etag=ETAG_VALUE |
| 82 | + URL, |
| 83 | + APP_NAME, |
| 84 | + INSTANCE_ID, |
| 85 | + CUSTOM_HEADERS, |
| 86 | + CUSTOM_OPTIONS, |
| 87 | + REQUEST_TIMEOUT, |
| 88 | + request_retries=1, |
| 89 | + project=PROJECT_NAME, |
| 90 | + cached_etag=ETAG_VALUE, |
58 | 91 | ) |
59 | 92 |
|
60 | 93 | assert json.loads(body)["version"] == 1 |
61 | 94 | assert etag == ETAG_VALUE |
62 | 95 |
|
| 96 | + |
63 | 97 | @pytest.mark.asyncio |
64 | 98 | async def test_get_feature_toggles_failure_after_retries(): |
65 | 99 | with aioresponses() as m: |
66 | 100 | m.get(PROJECT_URL, status=500) |
67 | 101 | m.get(PROJECT_URL, status=500) |
68 | 102 | body, etag = await get_feature_toggles_async( |
69 | | - URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, |
70 | | - REQUEST_TIMEOUT, request_retries=1, project=PROJECT_NAME |
| 103 | + URL, |
| 104 | + APP_NAME, |
| 105 | + INSTANCE_ID, |
| 106 | + CUSTOM_HEADERS, |
| 107 | + CUSTOM_OPTIONS, |
| 108 | + REQUEST_TIMEOUT, |
| 109 | + request_retries=1, |
| 110 | + project=PROJECT_NAME, |
71 | 111 | ) |
72 | 112 | assert body is None |
73 | 113 | assert etag == "" |
0 commit comments