API for list & download files of a project

tests/handlers/api/test_project.py

@@ -20,12 +20,14 @@ This test suite check /project endpoint
 """
 
 import uuid
+import os
 import asyncio
 import aiohttp
 from unittest.mock import patch
 from tests.utils import asyncio_patch
 
 from gns3server.handlers.api.project_handler import ProjectHandler
+from gns3server.modules.project_manager import ProjectManager
 
 
 def test_create_project_with_path(server, tmpdir):
@@ -175,6 +177,42 @@ def test_notification(server, project, loop):
     assert response.body == b'{"action": "ping"}\n{"action": "vm.created", "event": {"a": "b"}}\n'
 
 
-def test_notification_invalid_id(server, project):
+def test_notification_invalid_id(server):
     response = server.get("/projects/{project_id}/notifications".format(project_id=uuid.uuid4()))
     assert response.status == 404
+
+
+def test_list_files(server, project):
+    files = [
+        {
+            "path": "test.txt",
+            "md5sum": "ad0234829205b9033196ba818f7a872b"
+        },
+        {
+            "path": "vm-1/dynamips/test.bin",
+            "md5sum": "098f6bcd4621d373cade4e832627b4f6"
+        }
+    ]
+    with asyncio_patch("gns3server.modules.project.Project.list_files", return_value=files) as mock:
+        response = server.get("/projects/{project_id}/files".format(project_id=project.id), example=True)
+        assert response.status == 200
+        assert response.json == files
+
+
+def test_get_file(server, tmpdir):
+
+    with patch("gns3server.config.Config.get_section_config", return_value={"project_directory": str(tmpdir)}):
+        project = ProjectManager.instance().create_project()
+
+    with open(os.path.join(project.path, "hello"), "w+") as f:
+        f.write("world")
+
+    response = server.get("/projects/{project_id}/files/hello".format(project_id=project.id), raw=True, example=True)
+    assert response.status == 200
+    assert response.body == b"world"
+
+    response = server.get("/projects/{project_id}/files/false".format(project_id=project.id), raw=True)
+    assert response.status == 404
+
+    response = server.get("/projects/{project_id}/files/../hello".format(project_id=project.id), raw=True)
+    assert response.status == 403