Remove duplicated type methods and old Django logic

awx/main/tests/unit/utils/test_common.py

@@ -22,7 +22,11 @@ from awx.main.models import (
     InventoryUpdate,
     ProjectUpdate,
     SystemJob,
-    WorkflowJob
+    WorkflowJob,
+    Inventory,
+    JobTemplate,
+    UnifiedJobTemplate,
+    UnifiedJob
 )
 
 
@@ -92,19 +96,40 @@ def test_set_environ():
     assert key not in os.environ
 
 
-# Cases relied on for scheduler dependent jobs list
-@pytest.mark.parametrize('model,name', [
+TEST_MODELS = [
     (Job, 'job'),
     (AdHocCommand, 'ad_hoc_command'),
     (InventoryUpdate, 'inventory_update'),
     (ProjectUpdate, 'project_update'),
     (SystemJob, 'system_job'),
-    (WorkflowJob, 'workflow_job')
-])
+    (WorkflowJob, 'workflow_job'),
+    (UnifiedJob, 'unified_job'),
+    (Inventory, 'inventory'),
+    (JobTemplate, 'job_template'),
+    (UnifiedJobTemplate, 'unified_job_template')
+]
+
+
+# Cases relied on for scheduler dependent jobs list
+@pytest.mark.parametrize('model,name', TEST_MODELS)
 def test_get_type_for_model(model, name):
     assert common.get_type_for_model(model) == name
 
 
+@pytest.mark.django_db
+def test_get_model_for_invalid_type():
+    with pytest.raises(LookupError):
+        common.get_model_for_type('foobar')
+
+
+@pytest.mark.django_db
+@pytest.mark.parametrize("model_type,model_class", [
+    (name, cls) for cls, name in TEST_MODELS
+])
+def test_get_model_for_valid_type(model_type, model_class):
+    assert common.get_model_for_type(model_type) == model_class
+
+
 @pytest.fixture
 def memoized_function(mocker, mock_cache):
     with mock.patch('awx.main.utils.common.get_memoize_cache', return_value=mock_cache):