def test_update_via_list_push(self):
no_action_executions = 4
no_rules = 4
no_trigger_instances = 5
saved = TraceDBTest._create_save_trace(
trace_tag='test_trace',
action_executions=[
str(bson.ObjectId()) for _ in range(no_action_executions)
],
rules=[str(bson.ObjectId()) for _ in range(no_rules)],
trigger_instances=[
str(bson.ObjectId()) for _ in range(no_trigger_instances)
])
# push updates
Trace.push_action_execution(
saved,
action_execution=TraceComponentDB(object_id=str(bson.ObjectId())))
Trace.push_rule(saved,
rule=TraceComponentDB(object_id=str(bson.ObjectId())))
Trace.push_trigger_instance(
saved,
trigger_instance=TraceComponentDB(object_id=str(bson.ObjectId())))
retrieved = Trace.get(id=saved.id)
self.assertEqual(retrieved.id, saved.id, 'Incorrect trace retrieved.')
self.assertEqual(len(retrieved.action_executions),
no_action_executions + 1)
self.assertEqual(len(retrieved.rules), no_rules + 1)
self.assertEqual(len(retrieved.trigger_instances),
no_trigger_instances + 1)
def test_get(self):
saved = TraceDBTest._create_save_trace(
trace_tag='test_trace',
action_executions=[str(bson.ObjectId()) for _ in range(4)],
rules=[str(bson.ObjectId()) for _ in range(4)],
trigger_instances=[str(bson.ObjectId()) for _ in range(5)])
retrieved = Trace.get(id=saved.id)
self.assertEquals(retrieved.id, saved.id, 'Incorrect trace retrieved.')
def test_get(self):
saved = TraceDBTest._create_save_trace(
trace_tag='test_trace',
action_executions=[str(bson.ObjectId()) for _ in range(4)],
rules=[str(bson.ObjectId()) for _ in range(4)],
trigger_instances=[str(bson.ObjectId()) for _ in range(5)])
retrieved = Trace.get(id=saved.id)
self.assertEquals(retrieved.id, saved.id, 'Incorrect trace retrieved.')
def test_update_via_list_push(self):
no_action_executions = 4
no_rules = 4
no_trigger_instances = 5
saved = TraceDBTest._create_save_trace(
trace_tag='test_trace',
action_executions=[str(bson.ObjectId()) for _ in range(no_action_executions)],
rules=[str(bson.ObjectId()) for _ in range(no_rules)],
trigger_instances=[str(bson.ObjectId()) for _ in range(no_trigger_instances)])
# push updates
Trace.push_action_execution(
saved, action_execution=TraceComponentDB(object_id=str(bson.ObjectId())))
Trace.push_rule(saved, rule=TraceComponentDB(object_id=str(bson.ObjectId())))
Trace.push_trigger_instance(
saved, trigger_instance=TraceComponentDB(object_id=str(bson.ObjectId())))
retrieved = Trace.get(id=saved.id)
self.assertEquals(retrieved.id, saved.id, 'Incorrect trace retrieved.')
self.assertEquals(len(retrieved.action_executions), no_action_executions + 1)
self.assertEquals(len(retrieved.rules), no_rules + 1)
self.assertEquals(len(retrieved.trigger_instances), no_trigger_instances + 1)
