Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_engine_override_test(self):
from aldjemy import core
self.assertEqual(core.get_connection_string(), 'sqlite+pysqlite://')
def test_engines_cache(self):
from aldjemy.core import Cache, get_engine
self.assertEqual(get_engine('default'), Cache.engines['default'])
self.assertEqual(get_engine('logs'), Cache.engines['logs'])
self.assertEqual(get_engine(), Cache.engines['default'])
self.assertNotEqual(get_engine('default'), get_engine('logs'))
def test_engines_cache(self):
from aldjemy.core import Cache, get_engine
self.assertEqual(get_engine('default'), Cache.engines['default'])
self.assertEqual(get_engine('logs'), Cache.engines['logs'])
self.assertEqual(get_engine(), Cache.engines['default'])
self.assertNotEqual(get_engine('default'), get_engine('logs'))
def test_engines_cache(self):
from aldjemy.core import Cache, get_engine
self.assertEqual(get_engine('default'), Cache.engines['default'])
self.assertEqual(get_engine('logs'), Cache.engines['logs'])
self.assertEqual(get_engine(), Cache.engines['default'])
self.assertNotEqual(get_engine('default'), get_engine('logs'))
def test_many_to_many_through_self_aliased(self):
"""Make sure aliased recursive through tables work."""
from sample.models import Person
through_field = Person._meta.get_field('parents')
if django.VERSION < (1, 9):
through = through_field.rel.through
else:
through = through_field.remote_field.through
metadata = MetaData(schema='unique')
sa_models = construct_models(metadata)
aliased(sa_models[through])
def test_custom_metadata_schema_aliased(self):
"""Make sure the aliased command works with the schema."""
# This was an issue that cropped up after things seemed
# to be generating properly, so we want to test it and
# make sure that it stays working.
from sample.models import Log
metadata = MetaData(schema='pseudorandom')
sa_models = construct_models(metadata)
aliased(sa_models[Log])
def test_custom_metadata_schema(self):
"""Use a custom MetaData instance to add a schema."""
# The use-case for this functionality is to allow using
# Foreign Data Wrappers, each with a full set of Django
# tables, to copy between databases using SQLAlchemy
# and the automatically generation of aldjemy.
from sample.models import Log
metadata = MetaData(schema='arbitrary')
sa_models = construct_models(metadata)
self.assertEqual(sa_models[Log].table.schema, 'arbitrary')
def test_many_to_many_through_self(self):
"""Make sure recursive through tables work."""
from sample.models import Person
through_field = Person._meta.get_field('parents')
if django.VERSION < (1, 9):
through = through_field.rel.through
else:
through = through_field.remote_field.through
metadata = MetaData(schema='unique')
sa_models = construct_models(metadata)
self.assertEqual(sa_models[through].table.schema, 'unique')
def test_sessions(self):
from aldjemy.orm import get_session
from django.db import connections
session_default = get_session()
session_default2 = get_session('default')
self.assertEqual(session_default, session_default2)
session_logs = get_session('logs')
self.assertEqual(connections['default'].sa_session, session_default)
self.assertEqual(connections['logs'].sa_session, session_logs)
self.assertNotEqual(session_default, session_logs)
def test_sessions(self):
from aldjemy.orm import get_session
from django.db import connections
session_default = get_session()
session_default2 = get_session('default')
self.assertEqual(session_default, session_default2)
session_logs = get_session('logs')
self.assertEqual(connections['default'].sa_session, session_default)
self.assertEqual(connections['logs'].sa_session, session_logs)
self.assertNotEqual(session_default, session_logs)