From 5166d894eb4433d0e1222ce7232a3ff64b7991e7 Mon Sep 17 00:00:00 2001
From: Julien Duponchelle <julien@gns3.net>
Date: Mon, 14 Nov 2016 18:45:44 +0100
Subject: [PATCH] Do not reload a project via /load if the project is already
 opened

Fix #785
---
 gns3server/controller/__init__.py |  6 +-
 scripts/random_query.py           | 92 +++++++++++++++++++++++++------
 2 files changed, 78 insertions(+), 20 deletions(-)

diff --git a/gns3server/controller/__init__.py b/gns3server/controller/__init__.py
index d275795b..150427d5 100644
--- a/gns3server/controller/__init__.py
+++ b/gns3server/controller/__init__.py
@@ -363,9 +363,9 @@ class Controller:
         topo_data.pop("type")
 
         if topo_data["project_id"] in self._projects:
-            self.remove_project(self._projects[topo_data["project_id"]])
-
-        project = yield from self.add_project(path=os.path.dirname(path), status="closed", filename=os.path.basename(path), **topo_data)
+            project = self._projects[topo_data["project_id"]]
+        else:
+            project = yield from self.add_project(path=os.path.dirname(path), status="closed", filename=os.path.basename(path), **topo_data)
         if load or project.auto_open:
             yield from project.open()
         return project
diff --git a/scripts/random_query.py b/scripts/random_query.py
index d4db7f5b..9eb5bf9c 100644
--- a/scripts/random_query.py
+++ b/scripts/random_query.py
@@ -25,7 +25,7 @@ import math
 import aiohttp
 import aiohttp.web
 import asyncio
-import async_timeout
+import random
 
 import coloredlogs
 import logging
@@ -73,17 +73,16 @@ async def query(method, path, body=None, **kwargs):
     if body:
         kwargs["data"] = json.dumps(body)
 
-    with async_timeout.timeout(10):
-        async with session.request(method, "http://localhost:3081/v2" + path, **kwargs) as response:
-            if response.status == 409:
-                raise HTTPConflict(method, path, response)
-            elif response.status >= 300:
-                raise HTTPError(method, path, response)
-            log.info("%s %s %d", method, path, response.status)
-            if response.headers["content-type"] == "application/json":
-                return await response.json()
-            else:
-                return "{}"
+    async with session.request(method, "http://localhost:3081/v2" + path, **kwargs) as response:
+        if response.status == 409:
+            raise HTTPConflict(method, path, response)
+        elif response.status >= 300:
+            raise HTTPError(method, path, response)
+        log.info("%s %s %d", method, path, response.status)
+        if response.headers["content-type"] == "application/json":
+            return await response.json()
+        else:
+            return "{}"
 
 
 async def post(path, **kwargs):
@@ -107,8 +106,10 @@ async def create_project():
             await delete("/projects/" + project["project_id"])
         elif project["project_id"] == PROJECT_ID:
             project_exists = True
+            tasks = []
             for node in await get("/projects/" + PROJECT_ID + "/nodes"):
-                await delete("/projects/" + PROJECT_ID + "/nodes/" + node["node_id"])
+                tasks.append(delete_node(project, node))
+            await asyncio.gather(*tasks)
     if project_exists:
         response = await post("/projects/" + PROJECT_ID + "/open")
     else:
@@ -119,8 +120,9 @@ async def create_project():
 async def create_node(project):
     global node_i
     response = await post("/projects/{}/nodes".format(project["project_id"]), body={
-        "node_type": "vpcs",
+        "node_type": "ethernet_switch",
         "compute_id": "local",
+        "symbol": ":/symbols/ethernet_switch.svg",
         "name": "Node{}".format(node_i),
         "x": (math.floor((node_i - 1) % 12.0) * 100) - 500,
         "y": (math.ceil((node_i) / 12.0) * 100) - 300
@@ -129,15 +131,71 @@ async def create_node(project):
     return response
 
 
+async def delete_node(project, node):
+    await delete("/projects/{}/nodes/{}".format(project["project_id"], node["node_id"]))
+
+
+async def create_link(project, nodes):
+    """
+    Create all possible link of a node
+    """
+    node1 = random.choice(list(nodes.values()))
+
+    for port in range(0, 8):
+        node2 = random.choice(list(nodes.values()))
+
+        if node1 == node2:
+            continue
+
+        data = {"nodes":
+                [
+                    {
+                        "adapter_number": 0,
+                        "node_id": node1["node_id"],
+                        "port_number": port
+                    },
+                    {
+                        "adapter_number": 0,
+                        "node_id": node2["node_id"],
+                        "port_number": port
+                    }
+                ]
+                }
+        try:
+            await post("/projects/{}/links".format(project["project_id"]), body=data)
+        except HTTPConflict:
+            pass
+
+
 async def build_topology():
     global node_i
 
     nodes = {}
     project = await create_project()
     while True:
-        node = await create_node(project)
-        nodes[node["node_id"]] = node
-        await asyncio.sleep(0.5)
+        rand = random.randint(0, 1000)
+        if rand < 500:  # chance to create a new node
+            if len(nodes.keys()) < 255:  # Limit of VPCS:
+                node = await create_node(project)
+                nodes[node["node_id"]] = node
+        elif rand < 950:  # create a link
+            if len(nodes.keys()) >= 2:
+                await create_link(project, nodes)
+        elif rand < 999:  # chance to delete a node
+            continue
+            if len(nodes.keys()) > 0:
+                node = random.choice(list(nodes.values()))
+                await delete_node(project, node)
+                del nodes[node["node_id"]]
+        elif len(nodes.keys()) > 0:  # % chance to delete all nodes
+            continue
+            node_i = 1
+            tasks = []
+            for node in nodes.values():
+                tasks.append(delete_node(project, node))
+            await asyncio.gather(*tasks)
+            nodes = {}
+        await asyncio.sleep(0.2)
 
 async def main(loop):
     global session