Added delete user api (#9)

tests/test_api.py

@@ -1,14 +1,10 @@
-import contextlib
-import requests
-import time
-import unittest
 import os
 import sys
-import shutil
-import copy
-from unittest.mock import patch
-import cloudpickle
 import threading
+import unittest
+
+import cloudpickle
+import requests
 from waitress.server import create_server
 
 sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
@@ -278,6 +274,27 @@ def test_disable_admin(self):
 
         self.assertEqual(the_user.is_admin, False)
 
+    def test_delete_user(self):
+        id = "test_delete_user"
+        the_user = AccessKey(id)
+        the_user.enable()
+
+        self.assertEqual(the_user.is_enable, True)
+        self.assertEqual(the_user.is_admin, False)
+
+        id_admin = "test_delete_user_admin"
+        the_admin_access_key = AccessKey(id_admin)
+        the_admin_access_key.enable()
+        the_admin_access_key.set_is_admin(True)
+
+        # Adding the id as user with add_admin_url endpoint
+        data = {"key": id}
+        response = requests.post("http://localhost:7777" + delete_user_url, auth=HTTPBasicAuth("", id_admin),
+                                 data=data)
+
+        self.assertEqual(the_user.is_enable, False)
+        self.assertEqual(the_user.is_admin, False)
+
 
 
 backup = sys.argv

upsonic_on_prem/api/operations/admin.py

@@ -1,12 +1,10 @@
-from upsonic_on_prem.api import app
-
-from upsonic_on_prem.api.urls import *
-
-from upsonic_on_prem.utils import storage, AccessKey
-
 from flask import jsonify
 from flask import request
 
+from upsonic_on_prem.api import app
+from upsonic_on_prem.api.urls import *
+from upsonic_on_prem.utils import AccessKey
+
 
 @app.route(get_admins_url, methods=["get"])
 def get_admins():
@@ -75,3 +73,12 @@ def disable_admin():
 
     return jsonify(True)
 
+
+@app.route(delete_user_url, methods=["POST"])
+def delete_user():
+    key = request.form.get("key")
+
+    user = AccessKey(key)
+    user.delete()
+
+    return jsonify(True)

upsonic_on_prem/api/urls.py

@@ -15,4 +15,6 @@
 
 enable_admin_url = "/enable_admin"
 
-disable_admin_url = "/disable_admin"
+disable_admin_url = "/disable_admin"
+
+delete_user_url = "/delete_user"