(self)
| 57 | self.assertEqual(len(retrieved), 2, "Should have 2 traces.") |
| 58 | |
| 59 | def test_update(self): |
| 60 | saved = TraceDBTest._create_save_trace( |
| 61 | trace_tag="test_trace", action_executions=[], rules=[], trigger_instances=[] |
| 62 | ) |
| 63 | retrieved = Trace.query(trace_tag=saved.trace_tag) |
| 64 | self.assertEqual(len(retrieved), 1, "Should have 1 trace.") |
| 65 | self.assertEqual(retrieved[0].id, saved.id, "Incorrect trace retrieved.") |
| 66 | |
| 67 | no_action_executions = 4 |
| 68 | no_rules = 4 |
| 69 | no_trigger_instances = 5 |
| 70 | saved = TraceDBTest._create_save_trace( |
| 71 | trace_tag="test_trace", |
| 72 | id_=retrieved[0].id, |
| 73 | action_executions=[ |
| 74 | str(bson.ObjectId()) for _ in range(no_action_executions) |
| 75 | ], |
| 76 | rules=[str(bson.ObjectId()) for _ in range(no_rules)], |
| 77 | trigger_instances=[ |
| 78 | str(bson.ObjectId()) for _ in range(no_trigger_instances) |
| 79 | ], |
| 80 | ) |
| 81 | retrieved = Trace.query(trace_tag=saved.trace_tag) |
| 82 | |
| 83 | self.assertEqual(len(retrieved), 1, "Should have 1 trace.") |
| 84 | self.assertEqual(retrieved[0].id, saved.id, "Incorrect trace retrieved.") |
| 85 | # validate update |
| 86 | self.assertEqual( |
| 87 | len(retrieved[0].action_executions), |
| 88 | no_action_executions, |
| 89 | "Failed to update action_executions.", |
| 90 | ) |
| 91 | self.assertEqual(len(retrieved[0].rules), no_rules, "Failed to update rules.") |
| 92 | self.assertEqual( |
| 93 | len(retrieved[0].trigger_instances), |
| 94 | no_trigger_instances, |
| 95 | "Failed to update trigger_instances.", |
| 96 | ) |
| 97 | |
| 98 | def test_update_via_list_push(self): |
| 99 | no_action_executions = 4 |
nothing calls this directly
no test coverage detected