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

---
 .../api/v2/simulations/simulationId/experiments/test_endpoint.py       | 3 +++
 1 file changed, 3 insertions(+)

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

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