def test_repr(self):
add_example_earning(session)
expected = f"<Earning('{example_earning['id']}', '{example_earning['user_id']}', " \
f"'{example_earning['category_id']}', '{example_earning['amount_money']}', " \
f"'{example_earning['time_creation']}')>"
earning = session.query(Earning).get(1)
self.assertEqual(repr(earning), expected)
def test_delete_list_earning(self):
for _ in range(2):
add_example_earning(session)
earnings = session.query(Earning).all()
Earning.delete_list_earning(session, earnings)
answer = session.query(Earning).all()
self.assertEqual(answer, [])
def test_make_text_delete_transaction(self):
add_example_category_earning(session)
add_example_earning(session)
earning = session.query(Earning).get(1)
expected = f"Вы уверены, что хотите удалить доход в размере <b>{example_earning['amount_money']} " \
f"рублей</b> категории <b>{example_category_earning['category']}</b>, созданный в " \
f"<b>{earning.get_str_time_creation()}</b>?"
self.assertEqual(self.deleter.make_text_delete_transaction(session), expected)
def test_exist_transactions(self):
now = datetime.datetime(2020, 1, 22, 10, 10)
add_example_user(session)
add_example_category_consumption(session)
add_example_category_earning(session)
add_example_consumption(session, now)
add_example_earning(session, now)
expected = f"Информация на 22.01.2020 10:10, Среда\n\n" \
f"<b>Расходы:</b>\n" \
f"{example_category_consumption['category']} - {example_consumption['amount_money']} р.\n\n" \
f"<b>Доходы:</b>\n" \
f"{example_category_earning['category']} - {example_earning['amount_money']} р.\n\n" \
f"<b>Всего:</b>\n" \
f"Доход: {example_earning['amount_money']}\n" \
f"Расход: {example_consumption['amount_money']}\n\n" \
f"Хорошего дня!"
user = session.query(User).get(1)
actual = make_text_today(session, now, user)
self.assertEqual(actual, expected)
def test_query_earning(self):
add_example_earning(session)
expected = [Earning(id=1, user_id=1, category_id=1, amount_money=0)]
earnings = session.query(Earning).all()
self.assertEqual(earnings, expected)
def test_get_time_creation(self):
add_example_earning(session)
earning = session.query(Earning).get(1)
answer = earning.get_str_time_creation()
expected = '10.01.2020, 12:30'
self.assertEqual(answer, expected)
def test_delete_transaction(self):
add_example_earning(session)
self.deleter.delete_transaction(session)
answer = session.query(Earning).get(1)
expected = None
self.assertEqual(answer, expected)