/django-test-addons

Testing support for different database system like Mongo, Redis, Neo4j, Memcache, Django Rest Framework for django

Primary LanguagePythonMIT LicenseMIT

Full Documentation:

Tutorial

This tutorial provides a step-by-step description on how to use django test addons for testing different database systems.

Getting Started

It is recommended to have local installation of respective databases just for testing. Staging or shared database or any database with critical data should never be used in testing, as database is cleaned after each test is ran. It is recommended to use a separate settings file for testing.

Warning

Be Careful to use correct settings for test databases. Using staging or any other database may result in cleaning of the entire database.

If you haven't installed django test addons already, use

pip install django-test-addons

Testing Mongodb

Defining test settings

Make sure you have running installation of mongodb and have mongoengine installed. Just specify the settings for connection to mongodb instance in the settings file. Define TEST_MONGO_DATABASE dict in your test file containing connection information.

Example:

Add this code to test settings file -

TEST_MONGO_DATABASE = {
    'db': 'test',
    'host': ['localhost'],
    'port': 27017,
}

Make sure to use same test database for all mongo database aliases. To clarify, say you have following mongo connection settings in your development/production settings containing two mongodb aliases.

MONGO_DATABASES = {
    'default': {
        'db': 'main',
        'host': ['193.34.32.11'], # random development server
        'port': 27017,
    },
    'miscellaneous': {
        'DB_NAME': 'misc',
        'HOST': ['193.34.32.11'],
        'PORT': 27017,
    }
}

In your test settings, make sure to disconnect all existing connections and connect all mongodb aliases to test db.

# import MONGO_DATABASES variable from development settings file or just use the
# variable if you are using single file for testing with some environment settings.

import mongoengine

TEST_MONGO_DATABASE = {
    'db': 'test',
    'host': ['localhost'],
    'port': 27017,
}

map(lambda connection: mongoengine.connection.disconnect(connection), MONGO_DATABASES.keys())

MONGO_DATABASES = {connection: TEST_MONGO_DATABASE for connection in MONGO_DATABASES.keys()}

for connection_name, attrs in MONGO_DATABASES.items():
    mongoengine.connect(**dict(zip(['alias'] + attrs.keys(), [connection_name] + attrs.values())))

Writing Tests

Just import MongoTestCase from test_addons, and inherit test class from it.

Example

import test_addons

class TestSomething(test_addons.MongoTestCase):

    def test_instantiation(self):
        pass

Testing Memcache

Just specify CLEAR_CACHE=TRUE in your test class, if you want to clear cache too(it could be Memcache or Redis or any other caching framework that works with django). You must have CACHES configured in your test settings for this to work.

Example

import test_addons

class TestSomething(test_addons.MongoTestCase):

    CLEAR_CACHE = True

    def test_instantiation(self):
        pass

Testing Redis

Defining test settings

Make sure you have redis db installed and a running redis server. Just specify TEST_CACHES dictionary in your test settings containing redis connection info.

Example:

TEST_CACHES = {
    'default': {
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "127.0.0.1:6379:0",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },
    'redis1': {
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": "127.0.0.1:6379:1",
        "OPTIONS": {
            "CLIENT_CLASS": "django_redis.client.DefaultClient",
        }
    },
}

Note

'django_redis.cache.ShardClient' does not allow flushing all db as of now, so make sure not to use it. Sharding is not required in testing environment anyway.

Writing Tests

Just import RedisTestCase from test_addons, and inherit test class from it.

Example

import test_addons

class TestSomething(test_addons.RedisTestCase):

    def test_instantiation(self):
        pass

Testing Neo4j Graph database

Defining test settings

Make sure you have neo4j graph installed and a running neo4j server. Just specify NEO4J_TEST_LINK pointing to ip address of running neo4j server in your test settings file.

Example

NEO4J_TEST_LINK = 'http://localhost:7474/db/data'

Note

Since neo4j 2.0, it requires authentication to connection to your neo4j server. Considering it is unnecessary for testing environment, make sure to set 'dbms.security.auth_enabled=false' in your neo4j-server.properties file

Writing Tests

Just import Neo4jTestCase from test_addons, and inherit test class from it.

Example

import test_addons

class TestSomething(test_addons.Neo4jTestCase):

    def test_instantiation(self):
        pass

Testing Django Rest Framework APIs

It provides support for testing Django rest framework api's along with one or more databases.

Note

Test cases described above would have worked for apis as well, but they use default Test Client provided by Django, whereas it uses Test Client provided by DRF having some additional facilities like forcing authentication.

Writing Tests

Just import APITestCase for the specific database you are using (specify settings accordingly).

Available options are:

  • APIRedisTestCase
  • APIMongoTestCase
  • APINeo4jTestCase
  • APIMongoRedisTestCase
  • APIRedisMongoNeo4jTestCase

Example Say we want to use test DRF apis along with mongodb.

import test_addons

class TestSomething(test_addons.APIMongoTestCase):

    def test_instantiation(self):
        pass

Composite Testing

Often multiple databases are used simulataneously, thereby creating the need of testing them simulataneously. Just to cater this need, django test addons provide different combinations of TestCases for respective database combinations.

Composite Test Cases:

  • MongoNeo4jTestCase
  • MongoRedisTestCase
  • RedisMongoNeo4jTestCase
  • APIRedisTestCase
  • APIMongoTestCase
  • APINeo4jTestCase
  • APIMongoRedisTestCase
  • APIRedisMongoNeo4jTestCase

Facing Issues

Make sure you have defined settings exactly as mentioned. If you still can't resolve the issue, you can use Django test addons mailing list or raise an issue on github or just mail me directly at hspandher@outlook.com