From 1c58ae3b25120ac670b897665bac7d8f18156220 Mon Sep 17 00:00:00 2001
From: Georgios Andreadis <info@gandreadis.com>
Date: Tue, 30 Jun 2020 14:01:17 +0200
Subject: Fix more violations and ensure experimentIds handling

---
 .../opendc/api/v2/simulations/simulationId/experiments/endpoint.py     | 3 +++
 .../api/v2/simulations/simulationId/experiments/test_endpoint.py       | 3 +++
 2 files changed, 6 insertions(+)

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

diff --git a/web-server/opendc/api/v2/simulations/simulationId/experiments/endpoint.py b/web-server/opendc/api/v2/simulations/simulationId/experiments/endpoint.py
index 2b8aac4d..637b8011 100644
--- a/web-server/opendc/api/v2/simulations/simulationId/experiments/endpoint.py
+++ b/web-server/opendc/api/v2/simulations/simulationId/experiments/endpoint.py
@@ -29,4 +29,7 @@ def POST(request):
 
     experiment.insert()
 
+    simulation.obj['experimentIds'].append(experiment.obj['_id'])
+    simulation.update()
+
     return Response(200, 'Successfully added Experiment.', experiment.obj)
diff --git a/web-server/opendc/api/v2/simulations/simulationId/experiments/test_endpoint.py b/web-server/opendc/api/v2/simulations/simulationId/experiments/test_endpoint.py
index 6a459a3c..1fe09b10 100644
--- a/web-server/opendc/api/v2/simulations/simulationId/experiments/test_endpoint.py
+++ b/web-server/opendc/api/v2/simulations/simulationId/experiments/test_endpoint.py
@@ -46,6 +46,7 @@ def test_add_experiment(client, mocker):
                         return_value={
                             '_id': '1',
                             'simulationId': '1',
+                            'experimentIds': ['1'],
                             'authorizations': [{
                                 'simulationId': '1',
                                 'authorizationLevel': 'EDIT'
@@ -54,6 +55,7 @@ def test_add_experiment(client, mocker):
     mocker.patch.object(DB,
                         'insert',
                         return_value={
+                            '_id': '1',
                             'topologyId': '1',
                             'traceId': '1',
                             'schedulerName': 'default',
@@ -61,6 +63,7 @@ def test_add_experiment(client, mocker):
                             'state': 'QUEUED',
                             'lastSimulatedTick': 0,
                         })
+    mocker.patch.object(DB, 'update', return_value=None)
     res = client.post(
         '/api/v2/simulations/1/experiments',
         json={'experiment': {
-- 
cgit v1.2.3