def test_get_projects(temp_app, temp_db):
'''Tests getting all projects by an authenticated user'''
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
res = get_projects(auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 200
assert res_data['message'] == 'Success'
assert isinstance(res_data['projects'], list)
def test_get_access_token(temp_app, temp_db):
'''Tests verifying a valid jwt'''
login_res = login_hello(temp_app)
auth_token = login_res['refreshToken']
res = get_authenticate(auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 200
assert set(['message', 'userId', 'email', 'accessToken']) == set(res_data)
assert res_data['email'] == '*****@*****.**'
def test_get_project(temp_app, temp_db):
'''Tests getting a specific project (with tracks)'''
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
res = get_project(1, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 200
assert isinstance(res_data['project'], dict)
assert 'id' in res_data
assert 'name' not in res_data
def test_save_new_project_fail(temp_app, temp_db):
'''Tests various failure cases when trying to save a new project'''
# Tests trying to save a project with a duplicate name under the same user
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
data = generate_temp_project()
data['name'] = 'New Project 1'
res = post_save(data, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 400
assert res_data['error'] == 'A project with that name already exists'
# Tests trying to save project with no auth header
data = generate_temp_project()
res = temp_app.post('/save',
data=json.dumps(data),
content_type='application/json')
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'No authentication provided'
# Tests trying to save project with no auth token
res = post_save(data, '', temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'No authentication provided'
# Tests trying to save project with an expired token
token = generate_expired_token('access',
temp_app.application.config['SECRET_KEY'])
res = post_save(data, token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'Invalid token'
# Tests trying to save project with a token signed with the wrong key
token = generate_invalid_token('access')
res = post_save(data, token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'Invalid token'
# Tests trying to use a refresh token to access projects
token = encode_auth_token('refresh', 1, datetime.timedelta(days=3),
temp_app.application.config['SECRET_KEY'])
res = post_save(data, token.decode(), temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'Invalid token type'
def test_delete_project(temp_app, temp_db):
'''Tests deleting a project'''
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
res = delete_project(1, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 200
assert res_data['message'] == 'Success'
cursor = temp_db.cursor(cursor_factory=psycopg2.extras.DictCursor)
cursor.execute('SELECT * FROM projects WHERE id = %s', (1, ))
db_project = cursor.fetchall()
assert not db_project
def test_save_new_project(temp_app, temp_db):
'''Tests saving a new project'''
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
data = generate_temp_project()
res = post_save(data, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 201
assert isinstance(res_data, dict)
assert res_data['message'] == 'Success'
assert isinstance(res_data['projectId'], int)
# ensure it's in the database
cursor = temp_db.cursor(cursor_factory=psycopg2.extras.DictCursor)
cursor.execute('SELECT * FROM projects WHERE id = %(projectId)s', res_data)
db_project = cursor.fetchall()[0]
assert db_project['name'] == 'My New Project'
assert db_project['data']['bpm'] == 120
cursor.close()
def test_save_existing_project(temp_app, temp_db):
''' Tests making changes to an existing project '''
login_res = login_hello(temp_app)
auth_token = login_res['accessToken']
project = generate_temp_project()
project['name'] = 'Fancy Beatz'
project['shared'] = True
project['id'] = 1
res = put_save(project, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 200
assert res_data['message'] == 'Success'
cursor = temp_db.cursor(cursor_factory=psycopg2.extras.DictCursor)
cursor.execute('SELECT * FROM projects WHERE id = %(id)s', project)
db_project = cursor.fetchone()
assert db_project['name'] == 'Fancy Beatz'
assert db_project['shared'] is True
assert db_project['data']['bpm'] == 120
assert db_project['data']['name'] == 'Fancy Beatz'
cursor.close()