def test_add_flow_1000_bad_category_transport_in_base(self):
self.session.add(Category('работа', TypeFlow.PROFIT))
self.session.commit()
text_answer('Wallet добавь 1000 в работа')
flow = self.session.query(Flow).all()[0]
self.assertEqual(1000, flow.money)
self.assertEqual('работа', flow.category.name)
def test_add_bad_category(self):
answer = text_answer('Wallet добавь категорию работа')
self.assertEqual('Не понимаю что вы хотите.', answer)
def test_add_bad_type_category(self):
answer = text_answer('Wallet добавь категорию приходов работа')
self.assertEqual('Не понял тип категории попробуйте снова.', answer)
def test_add_category_work_in_base(self):
text_answer('Wallet добавь категорию доходов работа')
category = self.session.query(Category).first()
self.assertIsNotNone(category)
self.assertEqual('работа', category.name)
self.assertEqual(TypeFlow.PROFIT, category.type)
def test_add_category_transport_in_base(self):
text_answer('Wallet добавь категорию расходов транспорт')
category = self.session.query(Category).first()
self.assertIsNotNone(category)
self.assertEqual('транспорт', category.name)
self.assertEqual(TypeFlow.LESS, category.type)
def test_add_category_work(self):
answer = text_answer('Wallet добавь категорию доходов работа')
self.assertEqual('Категория работа добавлена!', answer)
def test_add_category_transport(self):
answer = text_answer('Wallet добавь категорию расходов транспорт')
self.assertEqual('Категория транспорт добавлена!', answer)
def test_add_flow_2000_category_transport(self):
self.session.add(Category('транспорт', TypeFlow.LESS))
self.session.commit()
answer = text_answer('Wallet добавь 2000 в транспорт')
self.assertEqual('В транспорт добавлена 2000р!', answer)
def test_add_flow_1000_category_work(self):
self.session.add(Category('работа', TypeFlow.PROFIT))
self.session.commit()
answer = text_answer('Wallet добавь 1000 в работа')
self.assertEqual('В работа добавлена 1000р!', answer)
def test_add_flow_1000_bad_category_transport(self):
answer = text_answer('Wallet добавь 1000 в транспорт')
self.assertEqual('Категория транспорт не найдена', answer)