Add admin functionality to API (#3)

tests/test_api.py

@@ -175,6 +175,33 @@ def my_function():
 
 
 
+
+    def test_add_admin(self):
+        id = "test_add_admin"
+        id_admin = "test_add_admin_admin"
+        the_admin_access_key =  AccessKey(id_admin)
+        the_admin_access_key.enable()
+        the_admin_access_key.set_is_admin(True)
+
+
+        # Testing to access get_admins_url and expected to get 403
+        response = requests.get("http://localhost:7777"+get_admins_url, auth=HTTPBasicAuth("", id), )
+        self.assertEqual(response.status_code, 403)
+
+
+        # Adding the id as user with add_admin_url endpoint
+        data = {"key": id}
+        response = requests.post("http://localhost:7777"+add_admin_url, auth=HTTPBasicAuth("", id_admin), data=data)
+
+
+
+        # Testing to access get_admins_url and expected to get 200
+        response = requests.get("http://localhost:7777"+get_admins_url, auth=HTTPBasicAuth("", id), )
+        self.assertEqual(response.status_code, 200)
+
+
+
+
 backup = sys.argv
 sys.argv = [sys.argv[0]]
 unittest.main(exit=False)

upsonic_on_prem/api/operations/admin.py

@@ -11,3 +11,21 @@
 @app.route(get_admins_url, methods=["get"])
 def get_admins():
     return jsonify(AccessKey.get_admins())
+
+
+
+
+
+
+@app.route(add_admin_url, methods=["POST"])
+def add_admin():
+    key = request.form.get("key")
+
+    user = AccessKey(key)
+    user.enable()
+
+    user.set_is_admin(True)
+
+    return jsonify(True)
+
+

upsonic_on_prem/api/urls.py

@@ -4,4 +4,6 @@
 dump_url = "/dump"
 load_url = "/load"
 
-get_admins_url = "/get_admins"
+get_admins_url = "/get_admins"
+
+add_admin_url = "/add_admin"