List images return images on compute and controller

Fix #506

tests/controller/test_compute.py

@@ -15,7 +15,7 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-
+import os
 import pytest
 import json
 import aiohttp
@@ -247,3 +247,18 @@ def test_forward_post(compute, async_run):
     with asyncio_patch("aiohttp.ClientSession.request", return_value=response) as mock:
         async_run(compute.forward("POST", "qemu", "img", data={"id": 42}))
         mock.assert_called_with("POST", "https://example.com:84/v2/compute/qemu/img", auth=None, data='{"id": 42}', headers={'content-type': 'application/json'}, chunked=False)
+
+
+def test_images(compute, async_run, images_dir):
+    """
+    Will return image on compute and on controller
+    """
+    response = MagicMock()
+    response.status = 200
+    response.read = AsyncioMagicMock(return_value=json.dumps([{"filename": "linux.qcow2", "path": "linux.qcow2"}]).encode())
+    open(os.path.join(images_dir, "asa.qcow2"), "w+").close()
+    with asyncio_patch("aiohttp.ClientSession.request", return_value=response) as mock:
+        images = async_run(compute.images("qemu"))
+        mock.assert_called_with("GET", "https://example.com:84/v2/compute/qemu/images", auth=None, data=None, headers={'content-type': 'application/json'}, chunked=False)
+
+    assert images == [{"filename": "linux.qcow2", "path": "linux.qcow2"}, {"filename": "asa.qcow2", "path": "asa.qcow2"}]

tests/handlers/api/controller/test_compute.py

@@ -169,10 +169,10 @@ def test_compute_list_images(http_controller, controller):
     response = http_controller.post("/computes", params)
     assert response.status == 201
 
-    with asyncio_patch("gns3server.controller.compute.Compute.forward", return_value=[]) as mock:
-        response = http_controller.get("/computes/my_compute/qemu/images")
-        assert response.json == []
-        mock.assert_called_with("GET", "qemu", "images")
+    with asyncio_patch("gns3server.controller.compute.Compute.images", return_value=[{"filename": "linux.qcow2"}, {"filename": "asav.qcow2"}]) as mock:
+        response = http_controller.get("/computes/my_compute/qemu/images", example=True)
+        assert response.json == [{"filename": "linux.qcow2"}, {"filename": "asav.qcow2"}]
+        mock.assert_called_with("qemu")
 
 
 def test_compute_list_vms(http_controller, controller):

tests/utils/test_images.py

@@ -20,7 +20,7 @@ from unittest.mock import patch
 
 
 from gns3server.utils import force_unix_path
-from gns3server.utils.images import md5sum, remove_checksum, images_directories
+from gns3server.utils.images import md5sum, remove_checksum, images_directories, scan_for_images
 
 
 def test_images_directories(tmpdir):
@@ -90,3 +90,30 @@ def test_remove_checksum(tmpdir):
     assert not os.path.exists(str(tmpdir / 'hello.md5sum'))
 
     remove_checksum(str(tmpdir / 'not_exists'))
+
+
+def test_scan_for_images(tmpdir):
+    path1 = tmpdir / "images1" / "IOS" / "test1.image"
+    path1.write("1", ensure=True)
+    path1 = force_unix_path(str(path1))
+
+    path2 = tmpdir / "images2" / "test2.image"
+    path2.write("1", ensure=True)
+    path2 = force_unix_path(str(path2))
+
+    path3 = tmpdir / "images1" / "IOU" / "test3.bin"
+    path3.write("1", ensure=True)
+    path3 = force_unix_path(str(path3))
+
+    path4 = tmpdir / "images1" / "QEMU" / "test4.qcow2"
+    path4.write("1", ensure=True)
+    path4 = force_unix_path(str(path4))
+
+    with patch("gns3server.config.Config.get_section_config", return_value={
+            "images_path": str(tmpdir / "images1"),
+            "additional_images_path": "/tmp/null24564:{}".format(tmpdir / "images2"),
+            "local": False}):
+
+        assert scan_for_images("dynamips") == [str(path1), str(path2)]
+        assert scan_for_images("iou") == [str(path3)]
+        assert scan_for_images("qemu") == [str(path4)]