def test_get_data(self):
self._logger.info('get_data() enter')
self._logger.info('get_data() exit')
self._mox.ReplayAll()
source_data = [('SRC1', 'CAT1', str(self._now), 'some data'), ('SRC2', 'CAT2', self._str_now, 'some another data')]
self._db_manager.execute_nonquery('insert into STAT_DATA(SOURCE, CATEGORY, TIMEMARKER, DATA) values(?, ?, ?, ?)', source_data)
storage = SqliteStorage(self._db_manager.connection_string, self._logger)
actual_data = storage.get_data()
expected_data = [(1, 'SRC1', 'CAT1', self._str_now, 'some data'), (2, 'SRC2', 'CAT2', self._str_now, 'some another data')]
self.assertEquals(expected_data, actual_data)
self._mox.VerifyAll()
def test_clear(self):
self._logger.info('clear((2, 3)) enter')
self._logger.info('clear((2, 3)) exit')
self._mox.ReplayAll()
source_data = [('SRC1', 'CAT1', self._str_now, 'some data'),
('SRC2', 'CAT2', self._str_now, 'some other data'),
('SRC2', 'CAT3', self._str_now, 'some another data'),
('SRC3', 'CAT4', self._str_now, 'yet some other data')]
self._db_manager.execute_nonquery('insert into STAT_DATA(SOURCE, CATEGORY, TIMEMARKER, DATA) values(?, ?, ?, ?)', source_data)
storage = SqliteStorage(self._db_manager.connection_string, self._logger)
storage.clear((2, 3))
actual_data = self._db_manager.execute_query('SELECT ID, SOURCE, CATEGORY, TIMEMARKER, DATA FROM STAT_DATA ORDER BY ID')
expected_data = [(1, 'SRC1', 'CAT1', self._str_now, 'some data'), (4, 'SRC3', 'CAT4', self._str_now, 'yet some other data')]
self.assertEquals(expected_data, actual_data)
self._mox.VerifyAll()
