From 688df9d2d29023f398fd11c3cfa99700a346e8ca Mon Sep 17 00:00:00 2001
From: Andrew Svetlov <andrew.svetlov@gmail.com>
Date: Thu, 19 Nov 2015 14:17:34 +0200
Subject: [PATCH] Convert test_no_identity into pytest style

---
 tests/test_no_identity.py | 97 +++++++++++----------------------------
 1 file changed, 28 insertions(+), 69 deletions(-)

diff --git a/tests/test_no_identity.py b/tests/test_no_identity.py
index e45fd12..36cce51 100644
--- a/tests/test_no_identity.py
+++ b/tests/test_no_identity.py
@@ -1,82 +1,41 @@
 import asyncio
-import socket
-import unittest
+import pytest
 
-import aiohttp
 from aiohttp import web
 from aiohttp_security import remember, forget
 
 
-class TestNoAuth(unittest.TestCase):
-
-    def setUp(self):
-        self.loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(None)
-        self.client = aiohttp.ClientSession(loop=self.loop)
-
-    def tearDown(self):
-        self.client.close()
-        self.loop.run_until_complete(self.handler.finish_connections())
-        self.srv.close()
-        self.loop.run_until_complete(self.srv.wait_closed())
-        self.loop.close()
-
-    def find_unused_port(self):
-        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        s.bind(('127.0.0.1', 0))
-        port = s.getsockname()[1]
-        s.close()
-        return port
+@pytest.mark.run_loop
+def test_remember(create_app_and_client):
 
     @asyncio.coroutine
-    def create_server(self):
-        app = web.Application(loop=self.loop)
+    def do_remember(request):
+        response = web.Response()
+        yield from remember(request, response, 'Andrew')
 
-        port = self.find_unused_port()
-        self.handler = app.make_handler(
-            debug=False, keep_alive_on=False)
-        srv = yield from self.loop.create_server(
-            self.handler, '127.0.0.1', port)
-        url = "http://127.0.0.1:{}/".format(port)
-        self.srv = srv
-        return app, srv, url
+    app, client = yield from create_app_and_client()
+    app.router.add_route('POST', '/', do_remember)
+    resp = yield from client.post('/')
+    assert 500 == resp.status
+    assert (('Security subsystem is not initialized, '
+             'call aiohttp_security.setup(...) first') ==
+            resp.reason)
+    yield from resp.release()
 
-    def test_remember(self):
 
-        @asyncio.coroutine
-        def do_remember(request):
-            response = web.Response()
-            yield from remember(request, response, 'Andrew')
+@pytest.mark.run_loop
+def test_forget(create_app_and_client):
 
-        @asyncio.coroutine
-        def go():
-            app, srv, url = yield from self.create_server()
-            app.router.add_route('POST', '/', do_remember)
-            resp = yield from self.client.post(url)
-            self.assertEqual(500, resp.status)
-            self.assertEqual(('Security subsystem is not initialized, '
-                              'call aiohttp_security.setup(...) first'),
-                             resp.reason)
-            yield from resp.release()
+    @asyncio.coroutine
+    def do_forget(request):
+        response = web.Response()
+        yield from forget(request, response)
 
-        self.loop.run_until_complete(go())
-
-    def test_forget(self):
-
-        @asyncio.coroutine
-        def do_forget(request):
-            response = web.Response()
-            yield from forget(request, response)
-
-        @asyncio.coroutine
-        def go():
-            app, srv, url = yield from self.create_server()
-            app.router.add_route('POST', '/', do_forget)
-            resp = yield from self.client.post(url)
-            self.assertEqual(500, resp.status)
-            self.assertEqual(('Security subsystem is not initialized, '
-                              'call aiohttp_security.setup(...) first'),
-                             resp.reason)
-            yield from resp.release()
-
-        self.loop.run_until_complete(go())
+    app, client = yield from create_app_and_client()
+    app.router.add_route('POST', '/', do_forget)
+    resp = yield from client.post('/')
+    assert 500 == resp.status
+    assert (('Security subsystem is not initialized, '
+             'call aiohttp_security.setup(...) first') ==
+            resp.reason)
+    yield from resp.release()