class Test_test_cached_data(unittest.TestCase): def setUp(self): self.cache_o = Cached_data('test.csv', carTags) self.cache_o.load() def test_get_values_empty(self): row = self.cache_o.get_values('New ID') assert row == dict() def test_set_value(self): self.cache_o.set_value('New ID', 'Type', '1') row = self.cache_o.get_values('New ID') assert row['DB file ID'] == 'New ID' assert row['Type'] == '1' def test_set_value_new(self): self.cache_o.set_value('New ID 2', 'Aero', '1') row = self.cache_o.get_values('New ID 2') assert row['DB file ID'] == 'New ID 2' assert row['Type'] == '' assert row['Aero'] == '1' def test_write(self): if os.path.isfile('test.csv'): os.remove('test.csv') self.test_set_value() self.cache_o.write() assert os.path.isfile('test.csv') self.cache2_o = Cached_data('test.csv', carTags) self.cache2_o.load() row = self.cache2_o.get_values('New ID') assert row['DB file ID'] == 'New ID' assert row['Type'] == '1' os.remove('test.csv')
def savePressed(self): # Write the data to a file named by the unique ID field messagebox.askokcancel( 'Sorry', 'Save not yet implemented' ) return if 'Track Name' in self.fields: _cd_o = Cached_data(trackCacheDataFile, trackTags) else: _cd_o = Cached_data(carCacheDataFile, carTags) _cd_o.load() for i, tag in enumerate(self.fields): if self.label[i]['text'] == 'DB file ID': ident = self.entry[i].get() _cd_o.delete_entry(ident) break for i, tag in enumerate(self.fields): _cd_o.set_value(ident, self.label[i]['text'], self.entry[i].get()) _cd_o.write() reloadAllData()