def test_encode_auth_token(db):
user = User("*****@*****.**", "pass123", admin=True)
db.session.add(user)
db.session.commit()
auth_token = user.encode_auth_token(user.id)
assert isinstance(auth_token, bytes)
def test_decode_auth_token(db):
user = User("*****@*****.**", "pass123", admin=True)
db.session.add(user)
db.session.commit()
auth_token = user.encode_auth_token(user.id)
decoded = User.decode_auth_token(auth_token)
assert decoded == user.id
def get(self):
"""
Get information about all jobs.
"""
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return JobService.get_jobs_by_user_id(user_id)
def get(self):
"""
Get all scripts.
"""
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return ScriptService.get_all_by_user(user_id)
def get(self, job_id: str):
"""
Get information about the given job ID.
"""
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return JobService.get_job_by_id(job_id=job_id, user_id=user_id)
def post(self, script_name: str):
"""
Execute a given script. Accepts an arbitrary JSON object
"""
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return ScriptService.execute(user_id,
script_name,
placeholder_dict=request.get_json())
def post(self):
"""
Register a script with the backend.
"""
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
script_obj = request.parsed_obj
script_obj["user"] = user_id
return ScriptService.create(script_obj)
def delete(self, script_name: str):
"""
Delete the script.
"""
version = request.args.get("version")
delete_all = request.args.get("delete_all")
if (version is not None) and (delete_all is not None):
raise UserException(
"Please specify only one of `version` or `delete_all`.", 400)
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return ScriptService.delete(user_id,
script_name,
version=version,
delete_all=delete_all)
from unittest.mock import patch
from cli_api.auth.model import User
from cli_api.auth.service import UserService
@patch.object(UserService, "register_user",
lambda register_req: User(**register_req))
def test_register_user_success(app):
with app.test_client() as client:
payload = {"email": "*****@*****.**", "password": "******"}
result = client.post("/auth/register", json=payload)
assert result.status_code == 201
result_json = result.get_json()
print(result_json)
assert result_json["email"] == "*****@*****.**"
assert result_json["registered_on"]
@patch.object(UserService, "login_user", lambda register_req: b"token123")
def test_login_success(app):
with app.test_client() as client:
payload = {"email": "*****@*****.**", "password": "******"}
result = client.post("/auth/login", json=payload)
assert result.status_code == 200
result_json = result.get_json()
assert result_json["message"] == "Successfully logged in"
def test_user_create(db):
user = User("*****@*****.**", "pass123", admin=True)
assert user
def get(self, script_name: str):
auth_token = get_bearer_token()
user_id = User.decode_auth_token(auth_token)
return ScriptService.get_script_by_user_and_name(
user_id, script_name, version=request.args.get("version"))