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_save_existing_project_fail(temp_app, temp_db):
'''Tests various failure cases when saving an existing project'''
# Tests trying to save a nonexistent project
login_res = login_mackland(temp_app)
auth_token = login_res['accessToken']
project = generate_temp_project()
project['id'] = 0
res = put_save(project, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 404
assert res_data['error'] == 'Project does not exist'
# Tests trying to save a project created by a different user
project['id'] = 1
res = put_save(project, auth_token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 403
assert res_data['error'] == 'Forbidden: project belongs to another user'
# Tests trying to save a project with no auth header
res = temp_app.put('/save',
data=json.dumps(project),
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 a project with no auth token
res = put_save(project, '', temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'No authentication provided'
# Tests trying to save a project with an expired auth token
token = generate_expired_token('access',
temp_app.application.config['SECRET_KEY'])
res = put_save(project, token, temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'Invalid token'
# Tests trying to save a project with an invalid auth token
token = generate_invalid_token('access')
res = put_save(project, 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 = put_save(project, token.decode(), temp_app)
res_data = json.loads(res.data)
assert res.status_code == 401
assert res_data['error'] == 'Invalid token type'
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()
def temp_db(temp_app):
'''Populates the db with dummy data and yields the db connection'''
user_a = create_hashed_user({
'email': '*****@*****.**',
'password': '******'
})
user_a['id'] = 1
user_b = create_hashed_user({
'email': '*****@*****.**',
'password': '******'
})
user_b['id'] = 2
# db_conn = connect_to_db('beatbucket_test')
db_conn = connect_to_db(temp_app.application)
cursor = db_conn.cursor()
cursor.execute(
'''
INSERT INTO users (id, email, password, salt)
VALUES (%(id)s, %(email)s, %(password)s, %(salt)s)
''', user_a)
cursor.execute(
'''
INSERT INTO users (id, email, password, salt)
VALUES (%(id)s, %(email)s, %(password)s, %(salt)s)
''', user_b)
project_data = generate_temp_project()
project_data['name'] = 'New Project 1'
json_data = json.dumps(project_data)
project = {
'name': 'New Project 1',
'id': 1,
'user_id': 1,
'data': json_data,
'shared': False
}
cursor.execute(
'''
INSERT INTO projects (id, name, user_id, data, shared)
VALUES (%(id)s, %(name)s, %(user_id)s, %(data)s, %(shared)s)
''', project)
project['shared'] = True
project['name'] = 'Test Project'
project['id'] = 2
project_data['shared'] = True
project_data['name'] = 'Test Project'
project_data['id'] = 2
project['data'] = json.dumps(project_data)
cursor.execute(
'''
INSERT INTO projects (id, name, user_id, data, shared)
VALUES (%(id)s, %(name)s, %(user_id)s, %(data)s, %(shared)s)
''', project)
db_conn.commit()
cursor.close()
yield db_conn
cursor = db_conn.cursor()
cursor.execute('DELETE FROM projects')
cursor.execute('DELETE FROM users')
db_conn.commit()
cursor.close()
db_conn.close()