def test_generate_sql(self):
transaction = Transaction(self.from_id, self.to_id, self.category,
self.amount)
result = transaction.generate_sql()
self.assertEqual(
"INSERT INTO transactions (from_entity, to_entity, type, balance) "
"values (31, 21, 'bet', 50)", result)
def test_batch_generate_sql(self):
transaction = Transaction(self.from_id, self.to_id, self.category,
self.amount)
result = transaction.generate_sql(is_batch=True)
self.assertEqual("(31, 21, 'bet', 50)", result)