Fix ext.flask tests

This commit is contained in:
Roman Mogylatov 2020-07-10 17:20:37 -04:00
parent d916bd4ce3
commit 541534ec1d

View File

@ -40,14 +40,14 @@ class Application(containers.DeclarativeContainer):
class ApplicationTests(unittest.TestCase): class ApplicationTests(unittest.TestCase):
def setUp(self) -> None: def setUp(self):
application = Application() application = Application()
self.app = application.app() self.app = application.app()
self.app.config['SERVER_NAME'] = 'test-server.com' self.app.config['SERVER_NAME'] = 'test-server.com'
self.client = self.app.test_client() self.client = self.app.test_client()
self.client.__enter__() self.client.__enter__()
def tearDown(self) -> None: def tearDown(self):
self.client.__exit__(None, None, None) self.client.__exit__(None, None, None)
def test_index(self): def test_index(self):