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