(self)
| 35 | self.assertEqual(retrieved.id, saved.id, "Incorrect trace retrieved.") |
| 36 | |
| 37 | def test_query(self): |
| 38 | saved = TraceDBTest._create_save_trace( |
| 39 | trace_tag="test_trace", |
| 40 | action_executions=[str(bson.ObjectId()) for _ in range(4)], |
| 41 | rules=[str(bson.ObjectId()) for _ in range(4)], |
| 42 | trigger_instances=[str(bson.ObjectId()) for _ in range(5)], |
| 43 | ) |
| 44 | retrieved = Trace.query(trace_tag=saved.trace_tag) |
| 45 | self.assertEqual(len(retrieved), 1, "Should have 1 trace.") |
| 46 | self.assertEqual(retrieved[0].id, saved.id, "Incorrect trace retrieved.") |
| 47 | |
| 48 | # Add another trace with same trace_tag and confirm that we support. |
| 49 | # This is most likley an anti-pattern for the trace_tag but it is an unknown. |
| 50 | saved = TraceDBTest._create_save_trace( |
| 51 | trace_tag="test_trace", |
| 52 | action_executions=[str(bson.ObjectId()) for _ in range(2)], |
| 53 | rules=[str(bson.ObjectId()) for _ in range(4)], |
| 54 | trigger_instances=[str(bson.ObjectId()) for _ in range(3)], |
| 55 | ) |
| 56 | retrieved = Trace.query(trace_tag=saved.trace_tag) |
| 57 | self.assertEqual(len(retrieved), 2, "Should have 2 traces.") |
| 58 | |
| 59 | def test_update(self): |
| 60 | saved = TraceDBTest._create_save_trace( |
nothing calls this directly
no test coverage detected