def test_deposit_user_in_db(self): # Covers user: user in db econ = Economy() self.add_user(econ, "test_user", 69) econ.deposit("test_user", 351) self.assertEqual(econ.get_balance("test_user"), 420) self.drop_table(econ)
def test_deposit_all(self): econ = Economy() self.add_user(econ, "user1", 69) users = {'user1', 'user2'} econ.deposit_all(users, 69) self.assertEqual(econ.get_balance("user1"), 138) self.assertEqual(econ.get_balance("user2"), 69) self.drop_table(econ)
def test_tx_both_in_db(self): # Covers from_user: user in db, to_user: user in db econ = Economy() self.add_user(econ, "user1", 69) self.add_user(econ, "user2", 69) self.assertTrue(econ.transfer("user1", "user2", 69)) self.assertEqual(econ.get_balance("user1"), 0) self.assertEqual(econ.get_balance("user2"), 138) self.drop_table(econ)
def test_bal_user_not_in_db(self): # Covers user: user not in db econ = Economy() self.assertEqual(econ.get_balance("test_user"), 0)
def test_deposit_user_not_in_db(self): # Covers user: user not in db econ = Economy() econ.deposit("test_user", 420) self.assertEqual(econ.get_balance("test_user"), 420) self.drop_table(econ)