def test_data_file(self):
purgecounter = PurgeCounter(self.prefs)
purgecounter.write_data(self.counter)
with open(self.filename) as f:
self.assertEqual(f.read(), self.test_string)
def test_no_data_file(self):
purgecounter = PurgeCounter(self.prefs)
self.assertFalse(purgecounter.get_banned_for_life())
self.assertFalse(purgecounter.get_data())
def test_data_file(self):
purgecounter = PurgeCounter(self.prefs)
self.assertEqual(len(purgecounter.get_banned_for_life()), 1)
self.assertEqual(len(purgecounter.get_data()), 1)
def test_increment(self):
purge_counter = PurgeCounter(self.prefs)
self.assertFalse(purge_counter.get_data())
purge_counter.increment(self.hosts)
data = purge_counter.get_data()
self.assertEqual(set(data), self.hosts)
