Fix test issues

ansible · Mar 16, 2021 · 6b19557 · 6b19557
1 parent e9efcc5
commit 6b19557
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/awx/main/tests/unit/models/test_ha.py b/awx/main/tests/unit/models/test_ha.py
@@ -3,13 +3,12 @@
 from unittest.mock import Mock
 
 from awx.main.models import (
-    Job,
     InstanceGroup,
 )
 
 
 def T(impact):
-    j = mock.Mock(Job())
+    j = mock.Mock(spec_set=['task_impact'])
     j.task_impact = impact
     return j
 

diff --git a/awx/main/tests/unit/scheduler/test_kubernetes.py b/awx/main/tests/unit/scheduler/test_kubernetes.py
@@ -1,5 +1,4 @@
 import pytest
-from unittest import mock
 from django.conf import settings
 
 from awx.main.models import (
@@ -14,7 +13,7 @@
 
 @pytest.fixture
 def container_group():
-    instance_group = mock.Mock(InstanceGroup(name='container-group'))
+    instance_group = InstanceGroup(name='container-group', id=1)
 
     return instance_group