MONGODB_HOST = '127.0.0.1'
MONGODB_PORT = 27017
MONGODB_USER = 'aaa'
MONGODB_PASSWD = 'bbb'
MONGODB_NAME = 'proj'

TEST_RUNNER = 'utils.mongotest.MongoTestSuiteRunner'

mongoengine.connect(MONGODB_NAME, host=MONGODB_HOST,
                    port=MONGODB_PORT, username=MONGODB_USER, password=MONGODB_PASSWD)

from django.test.runner import DiscoverRunner
from mongoengine.connection import connect, disconnect, get_connection

from daodao import settings


class MongoTestSuiteRunner(DiscoverRunner):
    def setup_databases(self, **kwargs):
        self._test_dbname = settings.MONGODB_TEST_NAME
        disconnect()

        connect(
            self._test_dbname, host=settings.MONGODB_HOST,
            port=settings.MONGODB_PORT,
            username=settings.MONGODB_TEST_USER,
            password=settings.MONGODB_TEST_PASSWD
        )

        return super(MongoTestSuiteRunner, self).setup_databases(**kwargs)

    def teardown_databases(self, old_config, **kwargs):
        conn = get_connection()
        print 'drop mongo database %s...' % self._test_dbname
        conn.drop_database(self._test_dbname)

        super(MongoTestSuiteRunner, self).teardown_databases(old_config, **kwargs)

from __future__ import absolute_import

from .celery import app as celery_app

__all__ = ['celery_app']

# proj/celery.py
from __future__ import absolute_import

import os

from celery import Celery
from django.conf import settings

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'daodao.settings')

app = Celery('daodao')

app.config_from_object('django.conf:settings')
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)

from mongoengine import Document, StringField


class User(Document):
    nickname = StringField(max_length=30)

from __future__ import absolute_import
from celery import shared_task
from app.models import User

@shared_task
def debug_shared_task():
    # this will print the count of users in database xxx, not test_xxx
    print User.objects.all().count()