1+ from unittest import mock
2+
13from checks .normandy .remotesettings_recipes import NORMANDY_URL , run
24
35
46NORMANDY_SERVER = "http://n"
57REMOTESETTINGS_SERVER = "http://rs/v1"
68REMOTESETTINGS_BASELINE_URL = (
7- REMOTESETTINGS_SERVER + "/buckets/main/collections/normandy-recipes/records"
9+ REMOTESETTINGS_SERVER
10+ + "/buckets/main/collections/normandy-recipes/changeset?_expected=42"
811)
912REMOTESETTINGS_CAPABILITIES_URL = (
1013 REMOTESETTINGS_SERVER
11- + "/buckets/main/collections/normandy-recipes-capabilities/records "
14+ + "/buckets/main/collections/normandy-recipes-capabilities/changeset?_expected=42 "
1215)
1316
1417NORMANDY_RECIPE = {
@@ -67,13 +70,16 @@ async def test_positive(mock_aioresponses):
6770 )
6871 mock_aioresponses .get (
6972 REMOTESETTINGS_CAPABILITIES_URL ,
70- payload = {"data " : [REMOTESETTINGS_RECIPE , REMOTESETTINGS_RECIPE_WITH_CAPS ]},
73+ payload = {"changes " : [REMOTESETTINGS_RECIPE , REMOTESETTINGS_RECIPE_WITH_CAPS ]},
7174 )
7275 mock_aioresponses .get (
73- REMOTESETTINGS_BASELINE_URL , payload = {"data " : [REMOTESETTINGS_RECIPE ]}
76+ REMOTESETTINGS_BASELINE_URL , payload = {"changes " : [REMOTESETTINGS_RECIPE ]}
7477 )
7578
76- status , data = await run (NORMANDY_SERVER , REMOTESETTINGS_SERVER )
79+ with mock .patch (
80+ "checks.normandy.remotesettings_recipes.random.randint" , return_value = 42
81+ ):
82+ status , data = await run (NORMANDY_SERVER , REMOTESETTINGS_SERVER )
7783
7884 assert status is True
7985 assert data == {
@@ -94,13 +100,17 @@ async def test_negative(mock_aioresponses):
94100 )
95101 mock_aioresponses .get (
96102 REMOTESETTINGS_CAPABILITIES_URL ,
97- payload = {"data " : [REMOTESETTINGS_RECIPE_WITH_CAPS ]},
103+ payload = {"changes " : [REMOTESETTINGS_RECIPE_WITH_CAPS ]},
98104 )
99105 mock_aioresponses .get (
100106 REMOTESETTINGS_BASELINE_URL ,
101- payload = {"data " : [RECIPE_42 ]},
107+ payload = {"changes " : [RECIPE_42 ]},
102108 )
103- status , data = await run (NORMANDY_SERVER , REMOTESETTINGS_SERVER )
109+
110+ with mock .patch (
111+ "checks.normandy.remotesettings_recipes.random.randint" , return_value = 42
112+ ):
113+ status , data = await run (NORMANDY_SERVER , REMOTESETTINGS_SERVER )
104114
105115 assert status is False
106116 assert data == {
0 commit comments