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)