@@ -139,8 +139,8 @@ def test_returns_all_alliances_when_all_ids_exist(
139139 """
140140
141141 mock_existing_alliances = [
142- Alliance (alliance_id = 1 , name = "Alliance 1" ),
143- Alliance (alliance_id = 2 , name = "Alliance 2" ),
142+ Alliance (pk = 1 , name = "Alliance 1" ),
143+ Alliance (pk = 2 , name = "Alliance 2" ),
144144 ]
145145 mock_qs = MagicMock ()
146146 mock_qs .__iter__ .return_value = iter (mock_existing_alliances )
@@ -154,7 +154,7 @@ def test_returns_all_alliances_when_all_ids_exist(
154154 self .assertEqual (len (result ), 2 )
155155 self .assertEqual (result [1 ].name , "Alliance 1" )
156156 self .assertEqual (result [2 ].name , "Alliance 2" )
157- mock_filter .assert_called_once_with (alliance_id__in = {1 , 2 })
157+ mock_filter .assert_called_once_with (pk__in = {1 , 2 })
158158 mock_bulk_create .assert_not_called ()
159159 mock_get_alliance .assert_not_called ()
160160
@@ -178,7 +178,7 @@ def test_creates_new_alliances_when_some_ids_do_not_exist(
178178 """
179179
180180 mock_existing_alliances = [
181- Alliance (alliance_id = 1 , name = "Alliance 1" ),
181+ Alliance (pk = 1 , name = "Alliance 1" ),
182182 ]
183183 mock_qs = MagicMock ()
184184 mock_qs .__iter__ .return_value = iter (mock_existing_alliances )
@@ -192,8 +192,8 @@ def test_creates_new_alliances_when_some_ids_do_not_exist(
192192 mock_get_alliance .side_effect = [a2 , a3 ]
193193
194194 mock_bulk_create .return_value = [
195- Alliance (alliance_id = 2 , name = "Alliance 2" ),
196- Alliance (alliance_id = 3 , name = "Alliance 3" ),
195+ Alliance (pk = 2 , name = "Alliance 2" ),
196+ Alliance (pk = 3 , name = "Alliance 3" ),
197197 ]
198198
199199 result = Alliance .bulk_get_or_create_from_esi (
@@ -204,7 +204,7 @@ def test_creates_new_alliances_when_some_ids_do_not_exist(
204204 self .assertEqual (result [1 ].name , "Alliance 1" )
205205 self .assertEqual (result [2 ].name , "Alliance 2" )
206206 self .assertEqual (result [3 ].name , "Alliance 3" )
207- mock_filter .assert_called_once_with (alliance_id__in = {1 , 2 , 3 })
207+ mock_filter .assert_called_once_with (pk__in = {1 , 2 , 3 })
208208 mock_get_alliance .assert_any_call (alliance_id = 2 , force_refresh = False )
209209 mock_get_alliance .assert_any_call (alliance_id = 3 , force_refresh = False )
210210 mock_bulk_create .assert_called_once ()
@@ -239,7 +239,7 @@ def test_skips_creation_when_esi_returns_no_data(
239239 )
240240
241241 self .assertEqual (len (result ), 0 )
242- mock_filter .assert_called_once_with (alliance_id__in = {1 })
242+ mock_filter .assert_called_once_with (pk__in = {1 })
243243 mock_get_alliance .assert_called_once_with (alliance_id = 1 , force_refresh = False )
244244 mock_bulk_create .assert_not_called ()
245245
0 commit comments