66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import pytest
|
|
from unittest import mock
|
|
|
|
from awx.main.models.label import Label
|
|
from awx.main.models.unified_jobs import UnifiedJobTemplate, UnifiedJob
|
|
|
|
|
|
mock_query_set = mock.MagicMock()
|
|
|
|
mock_objects = mock.MagicMock(filter=mock.MagicMock(return_value=mock_query_set))
|
|
|
|
|
|
@mock.patch('awx.main.models.label.Label.objects', mock_objects)
|
|
class TestLabelFilterMocked:
|
|
def test_get_orphaned_labels(self, mocker):
|
|
ret = Label.get_orphaned_labels()
|
|
|
|
assert mock_query_set == ret
|
|
Label.objects.filter.assert_called_with(organization=None, unifiedjobtemplate_labels__isnull=True)
|
|
|
|
def test_is_detached(self, mocker):
|
|
mock_query_set.count.return_value = 1
|
|
|
|
label = Label(id=37)
|
|
ret = label.is_detached()
|
|
|
|
assert ret is True
|
|
Label.objects.filter.assert_called_with(id=37, unifiedjob_labels__isnull=True, unifiedjobtemplate_labels__isnull=True)
|
|
mock_query_set.count.assert_called_with()
|
|
|
|
def test_is_detached_not(self, mocker):
|
|
mock_query_set.count.return_value = 0
|
|
|
|
label = Label(id=37)
|
|
ret = label.is_detached()
|
|
|
|
assert ret is False
|
|
Label.objects.filter.assert_called_with(id=37, unifiedjob_labels__isnull=True, unifiedjobtemplate_labels__isnull=True)
|
|
mock_query_set.count.assert_called_with()
|
|
|
|
@pytest.mark.parametrize("jt_count,j_count,expected", [
|
|
(1, 0, True),
|
|
(0, 1, True),
|
|
(1, 1, False),
|
|
])
|
|
def test_is_candidate_for_detach(self, mocker, jt_count, j_count, expected):
|
|
mock_job_qs = mocker.MagicMock()
|
|
mock_job_qs.count = mocker.MagicMock(return_value=j_count)
|
|
mocker.patch.object(UnifiedJob, 'objects', mocker.MagicMock(
|
|
filter=mocker.MagicMock(return_value=mock_job_qs)))
|
|
|
|
mock_jt_qs = mocker.MagicMock()
|
|
mock_jt_qs.count = mocker.MagicMock(return_value=jt_count)
|
|
mocker.patch.object(UnifiedJobTemplate, 'objects', mocker.MagicMock(
|
|
filter=mocker.MagicMock(return_value=mock_jt_qs)))
|
|
|
|
label = Label(id=37)
|
|
ret = label.is_candidate_for_detach()
|
|
|
|
UnifiedJob.objects.filter.assert_called_with(labels__in=[label.id])
|
|
UnifiedJobTemplate.objects.filter.assert_called_with(labels__in=[label.id])
|
|
mock_job_qs.count.assert_called_with()
|
|
mock_jt_qs.count.assert_called_with()
|
|
|
|
assert ret is expected
|