def testPushEvents(self):
"""Tests the PushEvents function."""
event_heap = event_heaps.EventHeap()
self.assertEqual(len(event_heap._heap), 0)
event1 = containers_test_lib.TestEvent(5134324321)
event2 = containers_test_lib.TestEvent(2345871286)
event_heap.PushEvents([event1, event2])
self.assertEqual(len(event_heap._heap), 2)
def testPushEvent(self):
"""Tests the PushEvent function."""
event_heap = event_heaps.EventHeap()
self.assertEqual(len(event_heap._heap), 0)
event = containers_test_lib.TestEvent(5134324321)
event_heap.PushEvent(event)
self.assertEqual(len(event_heap._heap), 1)
def testPushEvent(self):
"""Tests the PushEvent function."""
event_heap = psort.PsortEventHeap()
self.assertEqual(len(event_heap._heap), 0)
event = containers_test_lib.TestEvent(
5134324321, attributes=self._TEST_EVENT_ATTRIBUTES)
event_heap.PushEvent(event)
self.assertEqual(len(event_heap._heap), 1)
def testPopEvents(self):
"""Tests the PopEvents function."""
event_heap = event_heaps.EventHeap()
self.assertEqual(len(event_heap._heap), 0)
test_events = list(event_heap.PopEvents())
self.assertEqual(len(test_events), 0)
event1 = containers_test_lib.TestEvent(5134324321)
event_heap.PushEvent(event1)
event2 = containers_test_lib.TestEvent(2345871286)
event_heap.PushEvent(event2)
self.assertEqual(len(event_heap._heap), 2)
test_events = list(event_heap.PopEvents())
self.assertEqual(len(test_events), 2)
self.assertEqual(len(event_heap._heap), 0)
def testPopEvent(self):
"""Tests the PopEvent function."""
event_heap = event_heaps.EventHeap()
self.assertEqual(len(event_heap._heap), 0)
test_event = event_heap.PopEvent()
self.assertIsNone(test_event)
event1 = containers_test_lib.TestEvent(5134324321)
event_heap.PushEvent(event1)
event2 = containers_test_lib.TestEvent(2345871286)
event_heap.PushEvent(event2)
self.assertEqual(len(event_heap._heap), 2)
test_event = event_heap.PopEvent()
self.assertIsNotNone(test_event)
self.assertEqual(len(event_heap._heap), 1)
def testPopEvents(self):
"""Tests the PopEvents function."""
event_heap = psort.PsortEventHeap()
self.assertEqual(len(event_heap._heap), 0)
test_events = list(event_heap.PopEvents())
self.assertEqual(len(test_events), 0)
event1 = containers_test_lib.TestEvent(
5134324321, attributes=self._TEST_EVENT_ATTRIBUTES)
event_heap.PushEvent(event1)
event2 = containers_test_lib.TestEvent(
2345871286, attributes=self._TEST_EVENT_ATTRIBUTES)
event_heap.PushEvent(event2)
self.assertEqual(len(event_heap._heap), 2)
test_events = list(event_heap.PopEvents())
self.assertEqual(len(test_events), 2)
self.assertEqual(len(event_heap._heap), 0)
def testGetEventIdentifiers(self):
"""Tests the _GetEventIdentifiers function."""
event_heap = psort.PsortEventHeap()
event = containers_test_lib.TestEvent(
5134324321, attributes=self._TEST_EVENT_ATTRIBUTES)
macb_group_identifier, content_identifier = (
event_heap._GetEventIdentifiers(event))
expected_identifier = 'data_type: test:event'
self.assertEqual(macb_group_identifier, expected_identifier)
expected_identifier = 'Metadata Modification Time, data_type: test:event'
self.assertEqual(content_identifier, expected_identifier)
def testNotInEventAndNoParent(self):
"""Call to an attribute that does not exist."""
event = test_lib.TestEvent(0, {})
with self.assertRaises(AttributeError):
getattr(event, u'doesnotexist')
