def setUp(self): self.mock = FlagServiceMock() self.rfid1 = '*****@*****.**' self.rfid2 = '*****@*****.**' self.person1 = self.mock.find_person(self.rfid1) self.person2 = self.mock.find_person(self.rfid2) self.ds = DataService(self.mock)
class DataServiceTestCase(unittest.TestCase): def setUp(self): self.mock = FlagServiceMock() self.rfid1 = '*****@*****.**' self.rfid2 = '*****@*****.**' self.person1 = self.mock.find_person(self.rfid1) self.person2 = self.mock.find_person(self.rfid2) self.ds = DataService(self.mock) def test_is_developer(self): self.assertFalse(self.ds.is_developer(self.person1)) self.mock.data[self.rfid1]['coffee|developer'] = '1' self.assertTrue(self.ds.is_developer(self.person1)) def test_beer_per_day(self): self.ds._set_default_credits_per_day(self.person1, 10) self.assertEqual(10, self.ds._get_credits_per_day(self.person1)) self.ds._set_default_credits_per_day(self.person1, 20) self.assertEqual(10, self.ds._get_credits_per_day(self.person1)) def test_beer_remaining(self): self.ds.set_credits(self.person1, 10) self.assertEqual(10, self.ds._get_credits(self.person1)) self.ds.set_credits(self.person1, 20) self.assertEqual(20, self.ds._get_credits(self.person1)) def test_beer_last(self): self.assertEqual(None, self.ds._get_last(self.person1)) nearly_now = datetime.datetime.now().replace(microsecond=0) self.ds.set_last(self.person1, nearly_now) self.assertEqual(nearly_now, self.ds._get_last(self.person1)) def test_total_beers(self): self.assertEqual(0, self.ds.get_total_beers(self.person1)) self.ds.set_total_beers(self.person1, 10) self.assertEqual(10, self.ds.get_total_beers(self.person1)) self.ds.set_total_beers(self.person1, 15) self.assertEqual(15, self.ds.get_total_beers(self.person1))