Implement apitest with aiohttp-pytestpyry/apitest

2 files changed, 13 insertions, 22 deletions

diff --git a/apitest/test_version.py b/apitest/test_version.py
new file mode 100644
index 0000000..1d72cd8
--- /dev/null
+++ b/apitest/test_version.py
@@ -0,0 +1,13 @@
+"""Version endpoint test module.
+"""
+
+import aiohttp
+
+
+async def test_version():
+    """Test version endpoint"""
+    async with aiohttp.ClientSession() as session:
+        async with session.get('http://0.0.0.0:8080/version') as resp:
+            assert resp.status == 200
+            resp = await resp.json()
+            assert 'api' in resp.keys()
diff --git a/apitest/version.py b/apitest/version.py
deleted file mode 100644
index 79183f4..0000000
--- a/apitest/version.py
+++ /dev/null
@@ -1,22 +0,0 @@
-"""Version endpoint test module.
-"""
-
-import asyncio
-
-import aiohttp
-
-
-async def main(url):
-    """Test version endpoint"""
-    async with aiohttp.ClientSession() as session:
-        async with session.get(url) as resp:
-            assert resp.status == 200
-            resp = await resp.json()
-            assert 'api' in resp.keys()
-            print("Version test oolrait!")
-
-
-if __name__ == '__main__':
-    LOOP = asyncio.get_event_loop()
-    LOOP.run_until_complete(main('http://0.0.0.0:8080/version'))
-    LOOP.close()