pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

get user perms or create them on login

Browse Source
This commit is contained in:
dandds 2018-07-18 10:44:42 -04:00
parent ac150d1af3
commit 0c0aa44468
2 changed files with 56 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
atst/handler.py
View File

@ -24,9 +24,16 @@ class BaseHandler(tornado.web.RequestHandler):
@tornado.gen.coroutine
def _get_user_permissions(self, user_id):
response = yield self.authz_client.post(
"/users", json={"id": user_id, "atat_role": "ccpo"}
response = yield self.authz_client.get(
"/users/{}".format(user_id), raise_error=False
)
if response.code == 404:
response = yield self.authz_client.post(
"/users", json={"id": user_id, "atat_role": "developer"}
)
return response.json
else:
return response.json
def get_current_user(self):

15
tests/mocks.py
View File

@ -5,6 +5,7 @@ from atst.api_client import ApiClient
class MockApiClient(ApiClient):
def __init__(self, service):
self.service = service
@ -43,6 +44,7 @@ class MockApiClient(ApiClient):
class MockRequestsClient(MockApiClient):
@tornado.gen.coroutine
def get(self, path, **kwargs):
json = {
@ -64,9 +66,7 @@ class MockRequestsClient(MockApiClient):
class MockAuthzClient(MockApiClient):
@tornado.gen.coroutine
def post(self, path, **kwargs):
json = {
_json = {
"atat_permissions": [
"view_original_jedi_request",
"review_and_approve_jedi_workspace_request",
@ -104,4 +104,11 @@ class MockAuthzClient(MockApiClient):
"username": None,
"workspace_roles": [],
}
return self._get_response("POST", path, 200, json=json)
@tornado.gen.coroutine
def post(self, path, **kwargs):
return self._get_response("POST", path, 200, json=self._json)
@tornado.gen.coroutine
def get(self, path, **kwargs):
return self._get_response("POST", path, 200, json=self._json)