def setUpForDjango(test):
connect(crate_host)
import os
os.environ["DJANGO_SETTINGS_MODULE"] = "crate.client.django.tests.settings"
from django.test.runner import setup_databases
setup_databases(3, False)
def setUpForDjango(test):
connect(crate_host)
import os
os.environ["DJANGO_SETTINGS_MODULE"] = "crate.client.django.tests.settings"
from django.test.runner import setup_databases
setup_databases(3, False)
def handle(self, *args, **options):
if len(args) < 1:
args = ["8001"]
if not options.get('settings'):
os.environ['DJANGO_SETTINGS_MODULE'] = 'mainsite.settings_testserver'
setup_databases(verbosity=1, interactive=False)
call_command("loaddata", os.path.join(TOP_DIR, "fixtures", "testserver.json"))
return super(Command, self).handle(*args, **options)
def handle(self, *args, **options):
if len(args) < 1:
args = ["8001"]
if not options.get('settings'):
os.environ[
'DJANGO_SETTINGS_MODULE'] = 'mainsite.settings_testserver'
setup_databases(verbosity=1, interactive=False)
call_command("loaddata",
os.path.join(TOP_DIR, "fixtures", "testserver.json"))
return super(Command, self).handle(*args, **options)
def _set_template(self):
verbosity = 0
old_config = setup_databases(verbosity=verbosity, interactive=False)
try:
if self.settings.database:
storage = self.settings.database.storage(u'django.fixtures')
else:
storage = None
self.template = best_satisfying_template(
search_strategy=self.strategy,
random=Random(),
condition=self.template_condition,
settings=self.settings,
storage=storage,
max_parameter_tries=1)
finally:
if django.VERSION[:2] < (1, 9):
old_names, _ = old_config
else:
old_names = old_config
for connection, old_name, destroy in old_names:
if destroy:
connection.creation.destroy_test_db(old_name, verbosity)
def _set_template(self):
verbosity = 0
old_config = setup_databases(
verbosity=verbosity, interactive=False
)
try:
if self.settings.database:
storage = self.settings.database.storage(u'django.fixtures')
else:
storage = None
self.template = best_satisfying_template(
search_strategy=self.strategy,
random=Random(),
condition=self.template_condition,
settings=self.settings,
storage=storage,
max_parameter_tries=1
)
finally:
if django.VERSION[:2] < (1, 9):
old_names, _ = old_config
else:
old_names = old_config
for connection, old_name, destroy in old_names:
if destroy:
connection.creation.destroy_test_db(
old_name, verbosity)
def setup_databases(self):
"""Setup the test databases."""
django.setup()
self._old_config = setup_databases(self.verbosity, self.interactive)
# Load the fixture into the database.
call_command("loaddata",
self.FIXTURE,
verbosity=self.verbosity,
database=DEFAULT_DB_ALIAS,
skip_validation=True)
def pytest_xdist_setupnodes(config):
"""called once before any remote node is set up. """
if not config.getvalue('xdist_one_db'):
return
_setup_django()
from django.conf import settings
if settings.DATABASES['default']['ENGINE'] == 'django.db.backends.sqlite3':
return
_handle_south()
if config.getvalue('nomigrations'):
_disable_native_migrations()
db_args = {}
if get_django_version() >= (1, 8):
db_args['keepdb'] = True
else:
monkey_patch_creation_for_db_reuse()
from django.test.runner import setup_databases
# Create the database
setup_databases(verbosity=config.option.verbose, interactive=False, **db_args)
def test_setup_databases(self):
"""
Test that setup_databases() doesn't fail with dummy database backend.
"""
runner = DjangoTestSuiteRunner(verbosity=0)
old_db_connections = db.connections
try:
db.connections = db.ConnectionHandler({})
old_config = runner.setup_databases()
runner.teardown_databases(old_config)
except Exception as e:
self.fail("setup_databases/teardown_databases unexpectedly raised "
"an error: %s" % e)
finally:
db.connections = old_db_connections
def test_setup_databases(self):
"""
Test that setup_databases() doesn't fail with dummy database backend.
"""
runner = DjangoTestSuiteRunner(verbosity=0)
old_db_connections = db.connections
try:
db.connections = db.ConnectionHandler({})
old_config = runner.setup_databases()
runner.teardown_databases(old_config)
except Exception as e:
self.fail("setup_databases/teardown_databases unexpectedly raised "
"an error: %s" % e)
finally:
db.connections = old_db_connections
def django_db_setup(
request,
django_test_environment,
django_db_blocker,
django_db_use_migrations,
django_db_keepdb,
django_db_modify_db_settings,
):
"""Top level fixture to ensure test databases are available"""
from django.test.runner import setup_databases, DiscoverRunner
setup_databases_args = {}
if not django_db_use_migrations:
_disable_native_migrations()
if django_db_keepdb:
if get_django_version() >= (1, 8):
setup_databases_args['keepdb'] = True
else:
# Django 1.7 compatibility
from .db_reuse import monkey_patch_creation_for_db_reuse
with django_db_blocker:
monkey_patch_creation_for_db_reuse()
with django_db_blocker:
db_cfg = setup_databases(
verbosity=pytest.config.option.verbose,
interactive=False,
**setup_databases_args
)
def teardown_database():
with django_db_blocker:
(DiscoverRunner(verbosity=pytest.config.option.verbose, interactive=False)
.teardown_databases(db_cfg))
if not django_db_keepdb:
request.addfinalizer(teardown_database)
testing_type = ' '.join(sys.argv) if len(sys.argv) > 1 else ''
try:
import django
except ImportError:
pass
else:
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/example')
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "project.settings")
django.setup()
try:
from django.test.runner import setup_databases
except ImportError:
from django.test.utils import setup_databases
setup_databases(0, False)
import djburger # noQA
if not testing_type:
from tests.main_tests import *
from tests.django_tests import *
from tests.djside_tests import *
from tests.rest_tests import *
from tests.side_tests import *
else:
if 'main_tests' in testing_type:
from tests import main_tests
if 'django_tests' in testing_type:
from tests import django_tests
if 'djside_tests' in testing_type:
def setup_databases(self, options):
return setup_databases(options['verbosity'], False)
def setUpClass(cls):
django.setup()
cls.old_config = setup_databases(0, False)
super(DatabaseTestCase, cls).setUpClass()
def setup():
dj_setup()
setup_databases(0, False)
def setup():
setup_databases(0, False)
def setup_databases(self, options):
return setup_databases(options['verbosity'], False)
import django
from django.contrib.auth import get_user_model
from django.test import TestCase
from django.test.runner import setup_databases
from mock import patch, Mock
django.setup()
# Only create a database once
setup_databases(1, 1)
User = get_user_model()
class TestDjangoAuthentication(TestCase):
def setUp(self):
from django_pypuppetdb.django_authentication import \
PuppetDBAuthentication
self.user = User.objects.create(
username='******', email='*****@*****.**')
self.auth = PuppetDBAuthentication()
def test_authenticate_without_connection(self):
self.assertIsNone(self.auth.authenticate())
@patch('django_pypuppetdb.user_authentication.UserAuthentication.'
'check_user', Mock(return_value=None))
def test_authenticate_incorrect_username(self):
self.assertIsNone(self.auth.authenticate('test'))
@patch('django_pypuppetdb.user_authentication.UserAuthentication.'
def setUpModule():
global db
setup_test_environment()
db = setup_databases(verbosity, interactive)
