Add functional tests for Model.add_machine()

[osm/N2VC.git] / tests / client / test_connection.py

diff --git a/tests/client/test_connection.py b/tests/client/test_connection.py
index 18b5863..0a7402b 100644 (file)
--- a/tests/client/test_connection.py
+++ b/tests/client/test_connection.py
@@ -1,15 +1,13 @@
-import asyncio
-import unittest
+import pytest
 
 from juju.client.connection import Connection
 from ..base import bootstrapped
 
 
 @bootstrapped
-class FunctionalConnectionTest(unittest.TestCase):
-    def test_connect_current(self):
-        loop = asyncio.get_event_loop()
-        conn = loop.run_until_complete(
-            Connection.connect_current())
+@pytest.mark.asyncio
+async def test_connect_current(event_loop):
+    conn = await Connection.connect_current()
 
-        self.assertIsInstance(conn, Connection)
+    assert isinstance(conn, Connection)
+    await conn.close()