Пример #1
0
class TestEventFileWriter(unittest.TestCase):
    def setUp(self):
        basedir = os.path.dirname(os.path.realpath(__file__))
        self.events_file = os.path.join(basedir, 'events_file.bin')
        self.evt = MockEvent(id=1, name='foo')
        self.fw = EventFileWriter(self.events_file)

    def test_event_file_creation(self):
        self.fw.on_event(self.evt)
        self.assertTrue(os.path.isfile(self.events_file))

    def test_event_correctness(self):
        self.fw.on_event(self.evt)
        with open(self.events_file, 'rb') as fp:
            event_file_content = fp.read()
        fp.close()

        event = MockEvent.deserialize(event_file_content)
        self.assertEqual(event.id, 1)
        self.assertEqual(event.name, 'foo')

    def tearDown(self):
        os.unlink(self.events_file)
Пример #2
0
 def setUp(self):
     basedir = os.path.dirname(os.path.realpath(__file__))
     self.events_file = os.path.join(basedir, 'events_file.bin')
     self.evt = MockEvent(id=1, name='foo')
     self.fw = EventFileWriter(self.events_file)