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)
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)