X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=tests%2Fintegration%2Ftest_unit.py;h=1604c311cb5f403b61e61cf9d94cb4a109f34ed9;hb=fbd25b458d70f0ca0743be60ff3d6ab21335707f;hp=e9116ce33db53e62016603427b81f4cb32baea6d;hpb=6ec9ae9638c417de432734853d7e04f57d965ff7;p=osm%2FN2VC.git

diff --git a/tests/integration/test_unit.py b/tests/integration/test_unit.py
index e9116ce..1604c31 100644
--- a/tests/integration/test_unit.py
+++ b/tests/integration/test_unit.py
@@ -1,5 +1,8 @@
+import asyncio
 import pytest
 
+from tempfile import NamedTemporaryFile
+
 from .. import base
 
 
@@ -44,3 +47,32 @@ async def test_run_action(event_loop):
             action = await run_action(unit)
             assert action.results == {'dir': '/var/git/myrepo.git'}
             break
+
+
+@base.bootstrapped
+@pytest.mark.asyncio
+async def test_scp(event_loop):
+    async with base.CleanModel() as model:
+        app = await model.deploy('ubuntu')
+
+        await asyncio.wait_for(
+            model.block_until(lambda: app.units),
+            timeout=60)
+        unit = app.units[0]
+        await asyncio.wait_for(
+            model.block_until(lambda: unit.machine),
+            timeout=60)
+        machine = unit.machine
+        await asyncio.wait_for(
+            model.block_until(lambda: (machine.status == 'running' and
+                                       machine.agent_status == 'started')),
+            timeout=480)
+
+        with NamedTemporaryFile() as f:
+            f.write(b'testcontents')
+            f.flush()
+            await unit.scp_to(f.name, 'testfile')
+
+        with NamedTemporaryFile() as f:
+            await unit.scp_from('testfile', f.name)
+            assert f.read() == b'testcontents'