python-dependency-injector/tests/unit/ext/test_flask_py2_py3.py

"""Dependency injector Flask extension unit tests."""

import 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')