Serialize NIO

tests/api/test_version.py

@@ -21,7 +21,7 @@ It's also used for unittest the HTTP implementation.
 """
 
 from tests.utils import asyncio_patch
-from tests.api.base import server, loop
+from tests.api.base import server, loop, port_manager
 from gns3server.version import __version__
 
 

tests/api/test_vpcs.py

@@ -15,6 +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/>.
 
+from unittest.mock import patch
 from tests.api.base import server, loop, port_manager
 from tests.utils import asyncio_patch
 from gns3server import modules
@@ -29,13 +30,26 @@ def test_vpcs_create(server):
     assert response.json['vpcs_id'] == 84
 
 
-def test_vpcs_nio_create(server):
-    response = server.post('/vpcs/42/ports/0/nio', {
-            'type': 'nio_unix',
-            'local_file': '/tmp/test',
-            'remote_file': '/tmp/remote'
+def test_vpcs_nio_create_udp(server):
+    vm = server.post('/vpcs', {'name': 'PC TEST 1'})
+    response = server.post('/vpcs/{}/ports/0/nio'.format(vm.json["vpcs_id"]), {
+            'type': 'nio_udp',
+            'lport': 4242,
+            'rport': 4343,
+            'rhost': '127.0.0.1'
         },
         example=True)
     assert response.status == 200
     assert response.route == '/vpcs/{vpcs_id}/ports/{port_id}/nio'
-    assert response.json['name'] == 'PC 2'
+    assert response.json['type'] == 'nio_udp'
+
+@patch("gns3server.modules.vpcs.vpcs_device.has_privileged_access", return_value=True)
+def test_vpcs_nio_create_tap(mock, server):
+    vm = server.post('/vpcs', {'name': 'PC TEST 1'})
+    response = server.post('/vpcs/{}/ports/0/nio'.format(vm.json["vpcs_id"]), {
+            'type': 'nio_tap',
+            'tap_device': 'test',
+        })
+    assert response.status == 200
+    assert response.route == '/vpcs/{vpcs_id}/ports/{port_id}/nio'
+    assert response.json['type'] == 'nio_tap'