From 6b19557e8f4301a2ba3e6bd1528722cdc12e07cb Mon Sep 17 00:00:00 2001
From: Yanis Guenane <yguenane@gmail.com>
Date: Mon, 15 Mar 2021 20:51:05 +0100
Subject: [PATCH] Fix test issues

---
 awx/main/tests/unit/models/test_ha.py            | 3 +--
 awx/main/tests/unit/scheduler/test_kubernetes.py | 3 +--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/awx/main/tests/unit/models/test_ha.py b/awx/main/tests/unit/models/test_ha.py
index 2534acfd15f1..0483ebfe278b 100644
--- 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
index 4121d0133bd7..44e9e4281612 100644
--- 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