def test_post_assign_to_not_existing_user(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.post(self.ENDPOINT,
headers=self.auth_header,
data={'user_id': 100, 'task_id': 1})
self.assert400(res)
def test_get_task_filtering_invalid_method_argument(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT + '?status=wrong',
headers=self.auth_header)
self.assert400(res)
self.assertIn('status', res.json['message'])
def test_delete_task(self):
task = Task(title='title1', status='in_progress')
task.assign_user(1)
db.session.add(task)
db.session.commit()
self.assertEqual(Task.query.count(), 1)
res = self.client.delete(self.ENDPOINT + '/1',
headers=self.auth_header)
self.assertEqual(res.status_code, 204)
self.assertEqual(Task.query.count(), 0)
def test_post_assign_integrity_error(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.post(self.ENDPOINT,
headers=self.auth_header,
data={'user_id': 1, 'task_id': 1})
self.assert400(res)
self.assertIn('Integrity', res.json['message'])
def test_get_tasks_count_for_current_user(self):
res = self.client.get(self.ENDPOINT, headers=self.auth_header)
self.assert200(res)
self.assertEqual(len(res.json['items']), 0)
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT, headers=self.auth_header)
self.assertEqual(len(res.json['items']), 1)
def post(self):
args = post_task.parse_args()
task = Task(**args)
task.assign_user(current_app.user.id)
try:
db.session.add(task)
db.session.commit()
except DataError as e:
db.session.rollback()
abort(400, str(e))
return task, 201
def test_get_tasks_count_for_other_user(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
user2 = User(name='user2', email='*****@*****.**')
user2.password = hashlib.md5('1').hexdigest()
db.session.add(user2)
db.session.commit()
token = self.login('user2', '1').json['token']
res = self.client.get(self.ENDPOINT, headers={'token': token})
self.assertEqual(len(res.json['items']), 0)
def test_post_assign_valid(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
user2 = User(name='user2', email='*****@*****.**')
user2.password = hashlib.md5('1').hexdigest()
db.session.add(user2)
db.session.commit()
res = self.client.post(self.ENDPOINT,
headers=self.auth_header,
data={'user_id': 2, 'task_id': 1})
self.assertEqual(res.status_code, 201)
def test_put_user_task(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
self.assertEqual(Task.query.get(1).status, 'in_progress')
res = self.client.get(self.ENDPOINT + '/1', headers=self.auth_header)
self.assert200(res)
res = self.client.put(self.ENDPOINT + '/1',
headers=self.auth_header,
data={'status': 'completed'})
self.assert200(res)
self.assertEqual(Task.query.get(1).status, 'completed')
def test_put_task_with_no_status(self):
task = Task(title='title1', status='in_progress')
task.assign_user(1)
db.session.add(task)
db.session.commit()
res = self.client.get(self.ENDPOINT + '/1', headers=self.auth_header)
self.assert200(res)
self.assertEqual(res.json['status'], 'in_progress')
self.assertEqual(Task.query.get(1).status, 'in_progress')
res = self.client.put(self.ENDPOINT + '/1',
headers=self.auth_header)
self.assert400(res)
self.assertEqual(Task.query.get(1).status, 'in_progress')
def test_get_assigned(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT + '/1', headers=self.auth_header)
self.assert200(res)
self.assertEqual(res.json['id'], 1)
self.assertEqual(res.json['status'], 'in_progress')
self.assertEqual(res.json['title'], 'title1')
self.assertEqual(res.json['users'], '[1]')
self.assertIn('2016', res.json['created_at'])
self.assertIn('2016', res.json['updated_at'])
def test_get_task_fields_default(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT, headers=self.auth_header)
self.assertDictEqual(res.json, {'limit': 20,
'offset': 0,
'items': [{'title': 'title1',
'status': 'in_progress',
'id': 1,
'users': '[1]'}],
'total': 1,
'user': 1})
def test_post_not_assigned_task(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
user2 = User(name='user2', email='*****@*****.**')
user2.password = hashlib.md5('1').hexdigest()
db.session.add(user2)
db.session.commit()
token_for_2_user = self.login('user2', '1').json['token']
res = self.client.post(self.ENDPOINT,
headers={'token': token_for_2_user},
data={'user_id': 2, 'task_id': 1})
self.assert403(res)
def test_get_forbidden(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT + '/1', headers=self.auth_header)
self.assert200(res)
user2 = User(name='user2', email='*****@*****.**')
user2.password = hashlib.md5('1').hexdigest()
db.session.add(user2)
db.session.commit()
token = self.login('user2', '1').json['token']
res = self.client.get(self.ENDPOINT + '/1', headers={'token': token})
self.assert403(res)
def test_post_assign_not_valid_fields(self):
DATA = {
'user_id': '1',
'task_id': '1'
}
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
for key in DATA.keys():
params = DATA.copy()
del params[key]
self.assert400(self.client.post(self.ENDPOINT,
headers=self.auth_header,
data=params))
def test_put_other_user_task(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
self.assertEqual(Task.query.get(1).status, 'in_progress')
res = self.client.get(self.ENDPOINT + '/1', headers=self.auth_header)
self.assert200(res)
user2 = User(name='user2', email='*****@*****.**')
user2.password = hashlib.md5('1').hexdigest()
db.session.add(user2)
db.session.commit()
token = self.login('user2', '1').json['token']
res = self.client.put(self.ENDPOINT + '/1',
headers={'token': token},
data={'status': 'completed'})
self.assert403(res)
self.assertEqual(Task.query.get(1).status, 'in_progress')
def test_get_task_fields_offset_limit(self):
for i in range(30):
task1 = Task(title='title' + str(i), status='in_progress')
task1.assign_user(1)
db.session.add(task1)
db.session.commit()
res = self.client.get(self.ENDPOINT, headers=self.auth_header)
self.assertEqual(res.json['limit'], 20)
self.assertEqual(res.json['offset'], 0)
self.assertEqual(len(res.json['items']), 20)
self.assertEqual(res.json['items'][0]['id'], 1)
self.assertEqual(res.json['total'], 30)
self.assertEqual(res.json['user'], 1)
res = self.client.get(self.ENDPOINT + '?offset=10&limit=5',
headers=self.auth_header)
self.assertEqual(res.json['limit'], 5)
self.assertEqual(res.json['offset'], 10)
self.assertEqual(len(res.json['items']), 5)
self.assertEqual(res.json['items'][0]['id'], 11)
self.assertEqual(res.json['total'], 30)
self.assertEqual(res.json['user'], 1)
def test_get_task_valid_filtering(self):
task1 = Task(title='title1', status='in_progress')
task1.assign_user(1)
task2 = Task(title='title2', status='completed')
task2.assign_user(1)
db.session.add(task1)
db.session.add(task2)
db.session.commit()
res = self.client.get(self.ENDPOINT,
headers=self.auth_header)
self.assert200(res)
self.assertEqual(len(res.json['items']), 2)
res = self.client.get(self.ENDPOINT + '?status=in_progress',
headers=self.auth_header)
self.assert200(res)
self.assertEqual(len(res.json['items']), 1)
self.assertEqual(res.json['items'][0]['status'], 'in_progress')
res = self.client.get(self.ENDPOINT + '?status=completed',
headers=self.auth_header)
self.assert200(res)
self.assertEqual(len(res.json['items']), 1)
self.assertEqual(res.json['items'][0]['status'], 'completed')
