def test_get_multiple_trace_db_by_rule(self): rule = DummyComponent(id_=self.trace2.rules[0].object_id) trace_dbs = trace_service.get_trace_db_by_rule(rule=rule) self.assertEqual(len(trace_dbs), 2, 'Expected 2 trace_db.') result = [trace_db.id for trace_db in trace_dbs] self.assertEqual(result, [self.trace2.id, self.trace3.id], 'Incorrect trace_dbs returned.')
def test_get_trace_db_by_rule(self): rule = DummyComponent(id_=self.trace1.rules[0].object_id) trace_dbs = trace_service.get_trace_db_by_rule(rule=rule) self.assertEqual(len(trace_dbs), 1, 'Expected 1 trace_db.') self.assertEqual(trace_dbs[0].id, self.trace1.id, 'Incorrect trace_db returned.')