additional test cases

uktrade · Dec 2, 2024 · 94b6d62 · 94b6d62
1 parent 7815274
commit 94b6d62
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/unit_tests/caseworker/advice/test_rules.py b/unit_tests/caseworker/advice/test_rules.py
@@ -225,3 +225,17 @@ def test_can_user_make_desnz_edit_invalid_advice_and_user(mock_gov_user, data_fa
     data_assigned_case
     request = get_mock_request(mock_gov_user["user"], data_fake_queue)
     assert not rules.test_rule("can_user_make_desnz_edit", request, data_assigned_case)
+
+
+def test_can_user_make_desnz_edit_request_missing_attributes(mock_gov_user, data_fake_queue, data_standard_case):
+    case = Case(data_standard_case["case"])
+    request = None
+
+    assert not advice_rules.can_user_make_desnz_edit(request, case)
+
+
+def test_can_user_make_desnz_edit_user_not_allocated(mock_gov_user, data_fake_queue, data_standard_case):
+    case = Case(data_standard_case["case"])
+    request = get_mock_request(mock_gov_user["user"], data_fake_queue)
+
+    assert not rules.test_rule("can_user_make_desnz_edit", request, case)