From f119fc78dda4d1e828dde04f378a63a93e3a0a7e Mon Sep 17 00:00:00 2001
From: Georgios Andreadis <info@gandreadis.com>
Date: Thu, 2 Jul 2020 18:39:28 +0200
Subject: Add current progress on frontend port

---
 web-server/opendc/api/v2/simulations/endpoint.py       | 3 +++
 web-server/opendc/api/v2/users/endpoint.py             | 1 +
 web-server/opendc/api/v2/users/userId/endpoint.py      | 8 ++++++++
 web-server/opendc/api/v2/users/userId/test_endpoint.py | 2 +-
 4 files changed, 13 insertions(+), 1 deletion(-)

(limited to 'web-server/opendc/api/v2')

diff --git a/web-server/opendc/api/v2/simulations/endpoint.py b/web-server/opendc/api/v2/simulations/endpoint.py
index bcdea650..993e317a 100644
--- a/web-server/opendc/api/v2/simulations/endpoint.py
+++ b/web-server/opendc/api/v2/simulations/endpoint.py
@@ -22,6 +22,9 @@ def POST(request):
     simulation.set_property('experimentIds', [])
     simulation.insert()
 
+    topology.set_property('simulationId', simulation.get_id())
+    topology.update()
+
     user = User.from_google_id(request.google_id)
     user.obj['authorizations'].append({'simulationId': simulation.get_id(), 'authorizationLevel': 'OWN'})
     user.update()
diff --git a/web-server/opendc/api/v2/users/endpoint.py b/web-server/opendc/api/v2/users/endpoint.py
index c597732f..0dcf2463 100644
--- a/web-server/opendc/api/v2/users/endpoint.py
+++ b/web-server/opendc/api/v2/users/endpoint.py
@@ -26,4 +26,5 @@ def POST(request):
     user.check_already_exists()
 
     user.insert()
+
     return Response(200, 'Successfully created user.', user.obj)
diff --git a/web-server/opendc/api/v2/users/userId/endpoint.py b/web-server/opendc/api/v2/users/userId/endpoint.py
index eedb2e3c..7a4fb104 100644
--- a/web-server/opendc/api/v2/users/userId/endpoint.py
+++ b/web-server/opendc/api/v2/users/userId/endpoint.py
@@ -1,3 +1,4 @@
+from opendc.models.simulation import Simulation
 from opendc.models.user import User
 from opendc.util.rest import Response
 
@@ -46,6 +47,13 @@ def DELETE(request):
     user.check_exists()
     user.check_correct_user(request.google_id)
 
+    for authorization in user.obj['authorizations']:
+        if authorization['authorizationLevel'] != 'OWN':
+            continue
+
+        simulation = Simulation.from_id(authorization['simulationId'])
+        simulation.delete()
+
     old_object = user.delete()
 
     return Response(200, 'Successfully deleted user.', old_object)
diff --git a/web-server/opendc/api/v2/users/userId/test_endpoint.py b/web-server/opendc/api/v2/users/userId/test_endpoint.py
index 0d590129..cdff2229 100644
--- a/web-server/opendc/api/v2/users/userId/test_endpoint.py
+++ b/web-server/opendc/api/v2/users/userId/test_endpoint.py
@@ -46,7 +46,7 @@ def test_delete_user_different_user(client, mocker):
 
 
 def test_delete_user(client, mocker):
-    mocker.patch.object(DB, 'fetch_one', return_value={'_id': '1', 'googleId': 'test'})
+    mocker.patch.object(DB, 'fetch_one', return_value={'_id': '1', 'googleId': 'test', 'authorizations': []})
     mocker.patch.object(DB, 'delete_one', return_value={'googleId': 'test'})
     res = client.delete('/api/v2/users/1')
     assert 'googleId' in res.json['content']
-- 
cgit v1.2.3