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)