add test for new helper method

mitre · Sep 25, 2023 · d2d5566 · d2d5566
1 parent 8092725
commit d2d5566
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/objects/test_operation.py b/tests/objects/test_operation.py
@@ -587,3 +587,12 @@ async def test_check_reason_skipped_other(self, custom_agent, test_ability, mock
         assert reason['reason_id'] == Operation.Reason.OTHER.value
         assert reason['ability_id'] == test_ability.ability_id
         assert reason['ability_name'] == test_ability.name
+
+    async def test_add_ignored_link(self, make_test_link, operation_agent):
+        test_agent = operation_agent
+        test_link = make_test_link(9876, test_agent.paw, Link().states['DISCARD'])
+        op = Operation(name='test', agents=[test_agent], state='running')
+        op.add_ignored_link(test_link)
+        assert op.ignored_links
+        assert test_link in op.ignored_links
+        assert len(op.ignored_links) == 1