Manage error when GNS3VM failed to start

Fix https://github.com/GNS3/gns3-gui/issues/1446

tests/controller/test_compute.py

@@ -92,6 +92,33 @@ def test_compute_httpQueryNotConnected(compute, controller, async_run):
     controller.notification.emit.assert_called_with("compute.updated", compute.__json__())
 
 
+
+def test_compute_httpQueryNotConnectedGNS3vmNotRunning(compute, controller, async_run):
+    """
+    We are not connected to the remote and it's a GNS3 VM. So we need to start it
+    """
+    controller._notification = MagicMock()
+    controller.gns3vm = AsyncioMagicMock()
+    controller.gns3vm.running = False
+
+    compute._id = "vm"
+    compute._connected = False
+    response = AsyncioMagicMock()
+    response.read = AsyncioMagicMock(return_value=json.dumps({"version": __version__}).encode())
+    response.status = 200
+    with asyncio_patch("aiohttp.ClientSession.request", return_value=response) as mock:
+        async_run(compute.post("/projects", {"a": "b"}))
+        mock.assert_any_call("GET", "https://example.com:84/v2/compute/capabilities", headers={'content-type': 'application/json'}, data=None, auth=None, chunked=False)
+        mock.assert_any_call("POST", "https://example.com:84/v2/compute/projects", data='{"a": "b"}', headers={'content-type': 'application/json'}, auth=None, chunked=False)
+
+    assert controller.gns3vm.start.called
+    assert compute._connected
+    assert compute._capabilities["version"] == __version__
+    controller.notification.emit.assert_called_with("compute.updated", compute.__json__())
+
+
+
+
 def test_compute_httpQueryNotConnectedInvalidVersion(compute, async_run):
     compute._connected = False
     response = AsyncioMagicMock()

tests/controller/test_controller.py

@@ -289,7 +289,12 @@ def test_getProject(controller, async_run):
 
 
 def test_start(controller, async_run):
-    async_run(controller.start())
+    controller.gns3vm.settings = {
+        "enable": False,
+        "engine": "vmware"
+    }
+    with asyncio_patch("gns3server.controller.compute.Compute.connect") as mock:
+        async_run(controller.start())
     assert len(controller.computes) == 1  # Local compute is created
     assert controller.computes["local"].name == socket.gethostname()
 
@@ -303,8 +308,9 @@ def test_start_vm(controller, async_run):
         "engine": "vmware"
     }
     with asyncio_patch("gns3server.controller.gns3vm.vmware_gns3_vm.VMwareGNS3VM.start") as mock:
-        async_run(controller.start())
-        assert mock.called
+        with asyncio_patch("gns3server.controller.compute.Compute.connect") as mock_connect:
+            async_run(controller.start())
+            assert mock.called
     assert "local" in controller.computes
     assert "vm" in controller.computes
     assert len(controller.computes) == 2  # Local compute and vm are created

tests/controller/test_gns3vm.py

@@ -16,9 +16,31 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 import pytest
-from tests.utils import asyncio_patch
+from tests.utils import asyncio_patch, AsyncioMagicMock
 
 from gns3server.controller.gns3vm import GNS3VM
+from gns3server.controller.gns3vm.gns3_vm_error import GNS3VMError
+
+@pytest.fixture
+def dummy_engine():
+    engine = AsyncioMagicMock()
+    engine.running = False
+    engine.ip_address = "vm.local"
+    engine.protocol = "https"
+    engine.port = 8442
+    engine.user = "hello"
+    engine.password = "world"
+    return engine
+
+
+@pytest.fixture
+def dummy_gns3vm(controller, dummy_engine):
+    vm = GNS3VM(controller)
+    vm._settings["engine"] = "dummy"
+    vm._settings["vmname"] = "Test VM"
+    vm._settings["enable"] = True
+    vm._engines["dummy"] = dummy_engine
+    return vm
 
 
 def test_list(async_run, controller):
@@ -50,3 +72,26 @@ def test_update_settings(controller, async_run):
     assert "vm" in controller.computes
     async_run(vm.update_settings({"enable": False}))
     assert "vm" not in controller.computes
+
+
+def test_auto_start(async_run, controller, dummy_gns3vm, dummy_engine):
+    """
+    When start the compute should be add to the controller
+    """
+    async_run(dummy_gns3vm.auto_start_vm())
+    assert dummy_engine.start.called
+    assert controller.computes["vm"].name == "GNS3 VM (Test VM)"
+    assert controller.computes["vm"].host == "vm.local"
+    assert controller.computes["vm"].port == 8442
+    assert controller.computes["vm"].protocol == "https"
+    assert controller.computes["vm"].user == "hello"
+    assert controller.computes["vm"].password == "world"
+
+
+def test_auto_start_with_error(async_run, controller, dummy_gns3vm, dummy_engine):
+    dummy_engine.start.side_effect = GNS3VMError("Dummy error")
+
+    async_run(dummy_gns3vm.auto_start_vm())
+    assert dummy_engine.start.called
+    assert controller.computes["vm"].name == "GNS3 VM (Test VM)"
+