mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-28 12:33:59 +03:00
e0d81c2d28
* Add aiohttp extension module * Add giphynav-aiohttp app * Add missing docstrings * Remove print() call * Remove not needed import from flask extension tests * Improve coroutine provider tests * Add aiohttp extension tests * Update tox.ini * Add aiohttp extras * Try fix Python 3.4 tests * Try fix 3.6 tests * Stop running coroutine tests for Python 3.4 * Rename tests * Remove type hints * Fix pypy and change python version for coverage job to 3.8 * Fix coveralls job * Try fix Python 3.4, 3.5 tests * Make coverage job to run 3.5+ tests * Add tests * Add readme * Update the readmes * Add API docs * Add API docs page * Update changelog
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
"""Dependency injector Flask extension unit tests."""
|
|
|
|
import unittest2 as unittest
|
|
from flask import Flask, url_for
|
|
from flask.views import MethodView
|
|
|
|
from dependency_injector import containers
|
|
from dependency_injector.ext import flask
|
|
|
|
|
|
def index():
|
|
return 'Hello World!'
|
|
|
|
|
|
def test():
|
|
return 'Test!'
|
|
|
|
|
|
class Test(MethodView):
|
|
def get(self):
|
|
return 'Test class-based!'
|
|
|
|
|
|
class ApplicationContainer(containers.DeclarativeContainer):
|
|
|
|
app = flask.Application(Flask, __name__)
|
|
|
|
index_view = flask.View(index)
|
|
test_view = flask.View(test)
|
|
test_class_view = flask.ClassBasedView(Test)
|
|
|
|
|
|
def create_app():
|
|
container = ApplicationContainer()
|
|
app = container.app()
|
|
app.container = container
|
|
app.add_url_rule('/', view_func=container.index_view.as_view())
|
|
app.add_url_rule('/test', 'test-test', view_func=container.test_view.as_view())
|
|
app.add_url_rule('/test-class', view_func=container.test_class_view.as_view('test-class'))
|
|
return app
|
|
|
|
|
|
class ApplicationTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.app = create_app()
|
|
self.app.config['SERVER_NAME'] = 'test-server.com'
|
|
self.client = self.app.test_client()
|
|
self.client.__enter__()
|
|
|
|
def tearDown(self):
|
|
self.client.__exit__(None, None, None)
|
|
|
|
def test_index(self):
|
|
response = self.client.get('/')
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data, b'Hello World!')
|
|
|
|
def test_test(self):
|
|
response = self.client.get('/test')
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data, b'Test!')
|
|
|
|
def test_test_class_based(self):
|
|
response = self.client.get('/test-class')
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data, b'Test class-based!')
|
|
|
|
def test_endpoints(self):
|
|
with self.app.app_context():
|
|
self.assertEqual(url_for('index'), 'http://test-server.com/')
|
|
self.assertEqual(url_for('test-test'), 'http://test-server.com/test')
|
|
self.assertEqual(url_for('test-class'), 'http://test-server.com/test-class')
|