def test_get_events(self):
descriptions = PicklableMock()
descriptions.formatter = HistoryEventFormatter()
items = {
'events': [{
'type': 'field_event',
'scenario': 'name',
'node': 'node',
'results': 'filled',
'content': {
'field': 'field_name'
}
}]
}
expected = [{
'no': 1,
'scenarioName': 'name',
'scenarioVersion': None,
'results': 'filled',
'eventType': 'field_event',
'eventContent': {
'field': 'field_name'
}
}]
history = History(items, descriptions, None)
events = history.get_events()
self.assertDictEqual(events[0], expected[0])
def test_history_expire(self):
now = time()
descriptions = PicklableMock()
descriptions.event_expiration_delay = 5
items = {
'events': [{
'type': 'event_type_1',
'content': {
'foo': 'bar'
},
'created_time': now - 1
}, {
'type': 'event_type_2',
'content': {
'foo': 'bar'
},
'created_time': now - 5
}]
}
expected_keys = {
'event_type_1',
}
history = History(items, descriptions, None)
history.expire()
history_raw = history.raw
events_raw = history_raw["events"]
event_keys = {event_raw.get("type") for event_raw in events_raw}
self.assertSetEqual(event_keys, expected_keys)
def test_history_raw(self):
now = time()
descriptions = PicklableMock()
items = {
'events': [{
'type': 'event_type_1',
'content': {
'foo': 'bar'
},
'created_time': now
}]
}
expected = {
'events': [{
'type': 'event_type_1',
'content': {
'foo': 'bar'
},
'node': None,
'results': None,
'scenario': None,
'scenario_version': None,
'created_time': now
}]
}
history = History(items, descriptions, None)
self.assertDictEqual(history.raw, expected)
def test_history_expire(self):
now = time()
descriptions = Mock()
descriptions.event_expiration_delay = 5
items = {
'events': [{
'type': 'event_type_1',
'content': {
'foo': 'bar'
},
'created_time': now + 7
}, {
'type': 'event_type_2',
'content': {
'foo': 'bar'
},
'created_time': now + 6
}, {
'type': 'event_type_3',
'content': {
'foo': 'bar'
},
'created_time': now - 5
}]
}
expected = {
'events': [{
'type': 'event_type_3',
'content': {
'foo': 'bar'
},
'node': None,
'results': None,
'scenario': None,
'scenario_version': None,
'created_time': now - 5
}]
}
history = History(items, descriptions, None)
history.expire()
self.assertDictEqual(history.raw, expected)
def test_history_add_event(self):
descriptions = PicklableMock()
item = {'type': 'event_type', 'content': {'foo': 'bar'}}
history = History({}, descriptions, None)
expected = Event(**item)
history.add_event(expected)
self.assertEqual(len(history.get_raw_events()), 1)
self.assertEqual(history.get_raw_events()[0], expected)
def test_history_clear(self):
descriptions = PicklableMock()
items = {
'events': [{
'type': 'event_type_1',
'content': {
'foo': 'bar'
}
}, {
'type': 'event_type_2',
'content': {
'foo': 'bar'
}
}]
}
history = History(items, descriptions, None)
self.assertEqual(len(history.get_raw_events()), 2)
history.clear()
self.assertEqual(len(history.get_raw_events()), 0)