Ejemplo n.º 1
0
class TestEventQueue(TestCase):
    def setUp(self):
        self.events = EventQueue()

    def test_empty_on_init(self):
        self.assertTrue(self.events.empty())

    def test_add_event(self):
        self.events.add_event(Event(5))
        self.assertFalse(self.events.empty())
        self.assertEqual(len(self.events._queue), 1)

    def test_add_events(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertFalse(self.events.empty())
        self.assertEqual(5, len(self.events._queue))

    def test_len(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertEqual(5, len(self.events))

    def test_get_event(self):
        self.events.add_event(Event(5))
        self.assertEqual(len(self.events._queue), 1)
        e = self.events.get_event()
        self.assertTrue(self.events.empty())
        self.assertEqual(5, e.timestamp)

    def test_get_current_events_empty(self):
        events = []
        self.events.add_events(events)
        curr_events = self.events.get_current_events(3)
        self.assertEqual(curr_events, [])

    def test_get_current_events(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        curr_events = self.events.get_current_events(3)
        self.assertEqual(len(curr_events), 3)
        for event, timestamp in zip(curr_events, [1, 2, 3]):
            self.assertEqual(event.timestamp, timestamp)

    def test_get_last_timestamp(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertEqual(5, self.events.get_last_timestamp())
        self.events.add_event(Event(8))
        self.assertEqual(8, self.events.get_last_timestamp())
        # Check that the last timestamp is unaltered from a call to
        # get_current_events at an earlier timestamp.
        _ = self.events.get_current_events(3)
        self.assertEqual(8, self.events.get_last_timestamp())

    def test_get_last_timestamp_no_events(self):
        events = []
        self.events.add_events(events)
        self.assertIsNone(self.events.get_last_timestamp())
Ejemplo n.º 2
0
class TestEventQueue(TestCase):
    def setUp(self):
        self.events = EventQueue()

    def test_empty_on_init(self):
        self.assertTrue(self.events.empty())

    def test_add_event(self):
        self.events.add_event(Event(5))
        self.assertFalse(self.events.empty())
        self.assertEqual(len(self.events._queue), 1)

    def test_add_events(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertFalse(self.events.empty())
        self.assertEqual(5, len(self.events._queue))

    def test_len(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertEqual(5, len(self.events))

    def test_get_event(self):
        self.events.add_event(Event(5))
        self.assertEqual(len(self.events._queue), 1)
        e = self.events.get_event()
        self.assertTrue(self.events.empty())
        self.assertEqual(5, e.timestamp)

    def test_get_current_events(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        curr_events = self.events.get_current_events(3)
        self.assertEqual(len(curr_events), 3)

    def test_get_last_timestamp(self):
        events = [Event(i) for i in range(1, 6)]
        self.events.add_events(events)
        self.assertEqual(5, self.events.get_last_timestamp())
        self.events.add_event(Event(8))
        self.assertEqual(8, self.events.get_last_timestamp())
        curr_events = self.events.get_current_events(3)
        self.assertEqual(8, self.events.get_last_timestamp())

    def test_get_last_timestamp_no_events(self):
        events = []
        self.events.add_events(events)
        self.assertIsNone(self.events.get_last_timestamp())