hobo/tests/test_version.py

import json

import pytest

import hobo.scrutiny.wsgi.middleware

pytestmark = pytest.mark.django_db


def test_version_middleware(settings, client):
    hobo.scrutiny.wsgi.middleware.VersionMiddleware.ENTROUVERT_PACKAGES = ['pytest', 'pytest-django']
    # disable apt_cache to get stable test results
    hobo.scrutiny.wsgi.middleware.apt_cache = None

    settings.MIDDLEWARE = ('hobo.middleware.version.VersionMiddleware',) + tuple(settings.MIDDLEWARE)
    json_response = client.get('/__version__').json()
    assert set(json_response.keys()) == set(['pytest', 'pytest-django'])