def _get_database_connection(self, config_section=_CONFIG_SECTION):
if config_section == _CONFIG_SECTION:
self._override_config()
db = database_connection.DatabaseConnection(config_section)
self._fake_backend = self.god.create_mock_class(
database_connection._GenericBackend, 'fake_backend')
for exception in database_connection._DB_EXCEPTIONS:
setattr(self._fake_backend, exception, FakeDatabaseError)
self._fake_backend.rowcount = 0
def get_fake_backend(db_type):
self._db_type = db_type
return self._fake_backend
self.god.stub_with(db, '_get_backend', get_fake_backend)
db.reconnect_delay_sec = _RECONNECT_DELAY
return db
def get_migration_manager(db_name, debug, force):
database = database_connection.DatabaseConnection(db_name)
database.debug = debug
database.reconnect_enabled = False
database.connect()
return MigrationManager(database, force=force)