@@ -177,6 +177,7 @@ def job_mock():
177
177
def test_get_container_request (
178
178
activation_with_instance : models .Activation ,
179
179
settings : SettingsWrapper ,
180
+ container_engine_mock : MagicMock ,
180
181
):
181
182
"""Test build_cmdline."""
182
183
override_settings = {
@@ -185,7 +186,10 @@ def test_get_container_request(
185
186
"RULEBOOK_LIVENESS_CHECK_SECONDS" : 73 ,
186
187
}
187
188
apply_settings (settings , ** override_settings )
188
- activation_manager = ActivationManager (activation_with_instance )
189
+ activation_manager = ActivationManager (
190
+ activation_with_instance ,
191
+ container_engine = container_engine_mock ,
192
+ )
189
193
request = activation_manager ._get_container_request ()
190
194
assert isinstance (request , ContainerRequest )
191
195
cmdline = request .cmdline
@@ -200,28 +204,40 @@ def test_get_container_request(
200
204
201
205
202
206
@pytest .mark .django_db
203
- def test_get_container_request_no_instance (basic_activation ):
207
+ def test_get_container_request_no_instance (
208
+ basic_activation , container_engine_mock
209
+ ):
204
210
"""Test build_cmdline when no instance exists."""
205
- activation_manager = ActivationManager (basic_activation )
211
+ activation_manager = ActivationManager (
212
+ basic_activation , container_engine_mock
213
+ )
206
214
with pytest .raises (exceptions .ActivationManagerError ):
207
215
activation_manager ._get_container_request ()
208
216
209
217
210
218
@pytest .mark .django_db
211
- def test_start_deleted_activation (activation_with_instance ):
219
+ def test_start_deleted_activation (
220
+ activation_with_instance , container_engine_mock
221
+ ):
212
222
"""Test start verb when activation is deleted."""
213
- activation_manager = ActivationManager (activation_with_instance )
223
+ activation_manager = ActivationManager (
224
+ activation_with_instance , container_engine_mock
225
+ )
214
226
activation_with_instance .delete ()
215
227
with pytest .raises (exceptions .ActivationStartError ) as exc :
216
228
activation_manager .start ()
217
229
assert "does not exist" in str (exc .value )
218
230
219
231
220
232
@pytest .mark .django_db
221
- def test_start_disabled_activation (activation_with_instance , eda_caplog ):
233
+ def test_start_disabled_activation (
234
+ activation_with_instance , eda_caplog , container_engine_mock
235
+ ):
222
236
"""Test start verb when activation is deleted."""
223
237
err_msg = "is disabled. Can not be started."
224
- activation_manager = ActivationManager (activation_with_instance )
238
+ activation_manager = ActivationManager (
239
+ activation_with_instance , container_engine_mock
240
+ )
225
241
activation_with_instance .is_enabled = False
226
242
activation_with_instance .save (update_fields = ["is_enabled" ])
227
243
with pytest .raises (exceptions .ActivationStartError ) as exc :
@@ -232,12 +248,17 @@ def test_start_disabled_activation(activation_with_instance, eda_caplog):
232
248
233
249
@pytest .mark .django_db
234
250
def test_start_no_awx_token (
235
- basic_activation , rulebook_with_job_template , preseed_credential_types
251
+ basic_activation ,
252
+ rulebook_with_job_template ,
253
+ container_engine_mock ,
254
+ preseed_credential_types ,
236
255
):
237
256
"""Test start verb when no AWX token is present."""
238
257
basic_activation .rulebook = rulebook_with_job_template
239
258
basic_activation .save (update_fields = ["rulebook" ])
240
- activation_manager = ActivationManager (basic_activation )
259
+ activation_manager = ActivationManager (
260
+ basic_activation , container_engine_mock
261
+ )
241
262
basic_activation .user .awxtoken_set .all ().delete ()
242
263
with pytest .raises (exceptions .ActivationStartError ) as exc :
243
264
activation_manager .start ()
@@ -249,9 +270,13 @@ def test_start_no_awx_token(
249
270
250
271
251
272
@pytest .mark .django_db
252
- def test_start_no_decision_environment (basic_activation ):
273
+ def test_start_no_decision_environment (
274
+ basic_activation , container_engine_mock
275
+ ):
253
276
"""Test start verb when no decision environment is present."""
254
- activation_manager = ActivationManager (basic_activation )
277
+ activation_manager = ActivationManager (
278
+ basic_activation , container_engine_mock
279
+ )
255
280
basic_activation .decision_environment .delete ()
256
281
with pytest .raises (exceptions .ActivationStartError ) as exc :
257
282
activation_manager .start ()
@@ -655,11 +680,14 @@ def test_start_max_running_activations(
655
680
settings : SettingsWrapper ,
656
681
eda_caplog : LogCaptureFixture ,
657
682
job_mock : MagicMock ,
683
+ container_engine_mock : MagicMock ,
658
684
preseed_credential_types ,
659
685
):
660
686
"""Test start verb when max running activations is reached."""
661
687
apply_settings (settings , MAX_RUNNING_ACTIVATIONS = 1 )
662
- activation_manager = ActivationManager (basic_activation )
688
+ activation_manager = ActivationManager (
689
+ basic_activation , container_engine_mock
690
+ )
663
691
664
692
with pytest .raises (exceptions .MaxRunningProcessesError ), mock .patch (
665
693
"aap_eda.services.activation.activation_manager.get_current_job" ,
0 commit comments