class TestModels(TestCase):
def setUp(self):
self.task = Tasks(task_name='test_task', author='author')
self.task.create()
@property
def __fields_list(self):
return loads(self.task.fields_list)['fields']
def test_create_repeatedly(self):
with self.assertRaises(TaskIsAlreadyExistsException):
Tasks(task_name='test_task', author='author').create()
def test_add_field(self):
self.task.add_field('field_1')
self.task.add_field('field_2')
self.task.refresh_from_db()
fields = self.__fields_list
self.assertEquals(fields[0]['name'], 'field_1')
self.assertEquals(fields[0]['field_id'], 0)
self.assertEquals(fields[1]['name'], 'field_2')
self.assertEquals(fields[1]['field_id'], 1)
def test_remark_field(self):
self.task.add_field('field_1')
self.task.remark_field(0)
self.task.refresh_from_db()
fields = self.__fields_list
self.assertEquals(fields[0]['marked'], 1)
self.task.remark_field(0)
fields = self.__fields_list
self.assertEquals(fields[0]['marked'], 0)
def test_delete_field(self):
self.task.add_field('field_1')
self.task.delete_field(0)
self.task.refresh_from_db()
self.assertIsNone(self.task.fields_list)
