diff --git a/scripts/test_e2e.py b/scripts/test_e2e.py index 079b75ef..8a767d00 100644 --- a/scripts/test_e2e.py +++ b/scripts/test_e2e.py @@ -44,14 +44,18 @@ def main(args): "Content-Type": "application/json", } - user_login = "my_user" - user_pwd = "my_pwd" # noqa S105 + # Create an organization + org_name = "my_org" + org_id = api_request("post", f"{args.endpoint}/organizations/", superuser_auth, {"name": org_name})["id"] + + agent_login = "my_user" + agent_pwd = "my_pwd" # noqa S105 # create a user - payload = {"organization_id": 1, "login": user_login, "password": user_pwd, "role": "agent"} + payload = {"organization_id": org_id, "login": agent_login, "password": agent_pwd, "role": "agent"} user_id = api_request("post", f"{args.endpoint}/users/", superuser_auth, payload)["id"] agent_auth = { - "Authorization": f"Bearer {get_token(args.endpoint, user_login, user_pwd)}", + "Authorization": f"Bearer {get_token(args.endpoint, agent_login, agent_pwd)}", "Content-Type": "application/json", } # Get & Fetch access @@ -63,11 +67,11 @@ def main(args): new_pwd = "my_new_pwd" # noqa S105 api_request("patch", f"{args.endpoint}/users/{user_id}/", superuser_auth, {"password": new_pwd}) - # Create a camera (as admin until #79 is closed) + # Create a camera camera_name = "my_device" payload = { "name": camera_name, - "organization_id": 1, + "organization_id": org_id, "angle_of_view": 70.0, "elevation": 100, "lat": 44.7, @@ -111,7 +115,7 @@ def main(args): api_request("delete", f"{args.endpoint}/detections/{detection_id}/", superuser_auth) api_request("delete", f"{args.endpoint}/cameras/{cam_id}/", superuser_auth) api_request("delete", f"{args.endpoint}/users/{user_id}/", superuser_auth) - + api_request("delete", f"{args.endpoint}/organizations/{org_id}/", superuser_auth) print(f"SUCCESS in {time.time() - start_ts:.3}s") return