def test_list(self):
machine = Machine.objects.get(title='Troester',
lower='troester',
register=123)
user = User.objects.get(first_name='Anton', telegram_id=123456789)
Loop.objects.create(machine=machine, user=user)
telegram_id = 123456789
cmd = {'cmd': 'loop', 'name': None, 'date': None, 'param': None}
result_test = 'Список отслеживаемого Вами оборудования:\n\n/Troester\n\n'
self.assertEqual(display_loop(telegram_id, cmd), result_test)
def test_add(self):
telegram_id = 123456789
cmd = {
'cmd': 'loop',
'name': 'troester',
'date': None,
'param': None
} # date, name, cmd, param
result_test = 'В Ваш список отслеживания добавлено:\n /Troester\n'
self.assertEqual(display_loop(telegram_id, cmd), result_test)
def test_remove(self):
machine = Machine.objects.get(title='Troester',
lower='troester',
register=123)
user = User.objects.get(first_name='Anton', telegram_id=123456789)
Loop.objects.create(machine=machine, user=user)
telegram_id = 123456789
cmd = {
'cmd': 'loop',
'name': 'troester',
'date': None,
'param': None
} # date, name, cmd, param
result_test = 'Из Вашего списка отслеживания удалено:\n /Troester\n'
self.assertEqual(display_loop(telegram_id, cmd), result_test)
def test_bad_name(self):
telegram_id = 123456789
cmd = {'cmd': 'loop', 'name': 'proton', 'date': None, 'param': None}
result_test = 'Не найдено оорудование с указанным именем!'
self.assertEqual(display_loop(telegram_id, cmd), result_test)
def test_list_pure(self):
telegram_id = 123456789
cmd = {'cmd': 'loop', 'name': None, 'date': None, 'param': None}
result_test = 'У Вас еще не выбрано оборудования для отслеживания!'
self.assertEqual(display_loop(telegram_id, cmd), result_test)
pass