def test_get_trace_ignore_trace_tag(self): trace_context = {'trace_tag': self.trace1.trace_tag} trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, 'Incorrect trace_db returned.') trace_context = {'trace_tag': self.trace1.trace_tag} trace_db = trace_service.get_trace(trace_context, ignore_trace_tag=True) self.assertEqual(trace_db, None, 'Should be None.')
def test_get_trace_by_trace_context(self): trace_context = TraceContext(**{'id_': str(self.trace1.id)}) trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, 'Incorrect trace_db returned.') trace_context = TraceContext(**{'trace_tag': self.trace1.trace_tag}) trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, 'Incorrect trace_db returned.')
def test_get_trace_by_trace_context(self): trace_context = TraceContext(**{"id_": str(self.trace1.id)}) trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, "Incorrect trace_db returned.") trace_context = TraceContext(**{"trace_tag": self.trace1.trace_tag}) trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, "Incorrect trace_db returned.")
def test_get_trace_by_dict(self): trace_context = {'id_': str(self.trace1.id)} trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, 'Incorrect trace_db returned.') trace_context = {'id_': str(bson.ObjectId())} self.assertRaises(StackStormDBObjectNotFoundError, trace_service.get_trace, trace_context) trace_context = {'trace_tag': self.trace1.trace_tag} trace_db = trace_service.get_trace(trace_context) self.assertEqual(trace_db.id, self.trace1.id, 'Incorrect trace_db returned.')