def test_mongodb_connect_single_host(self, mongoengine_mock): connection = { 'HOST': 'host', 'PORT': 0, 'NAME': 'foo' } alias = 'bar' mongodb_connect(connection, alias) calls = [call('foo', host='mongodb://host:0/foo', alias='bar')] self.assertEqual(mongoengine_mock.connect.call_args_list, calls)
def test_mongodb_connect_multiple_host(self, mongoengine_mock): connection = { 'HOST': ['host1', 'host2'], 'PORT': [0, 1], 'NAME': 'foo', 'REPLICA_SET': 'replica' } alias = 'bar' mongodb_connect(connection, alias) calls = [call('foo', host='mongodb://host1:0,host2:1/foo?replicaSet=replica', alias='bar')] self.assertEqual(mongoengine_mock.connect.call_args_list, calls)
def test_mongodb_connect_with_auth(self, mongoengine_mock): connection = { 'USER': '******', 'PASSWORD': '******', 'HOST': 'host', 'PORT': 0, 'NAME': 'foo' } alias = 'bar' mongodb_connect(connection, alias) calls = [call('foo', host='mongodb://*****:*****@host:0/foo', alias='bar')] self.assertEqual(mongoengine_mock.connect.call_args_list, calls)
CUSTOM_PROVIDER = getattr(settings, 'AUDIT_CUSTOM_PROVIDER', {'audit': 'audit.middleware.custom_provider'}) # List of models that will be logged for audit LOGGED_MODELS = getattr(settings, 'AUDIT_LOGGED_MODELS', ()) # Activate or deactivate logging ACTIVATE = getattr(settings, 'AUDIT_ACTIVATE', False) # Database alias DB_ALIAS = getattr(settings, 'AUDIT_DB_ALIAS', 'audit') # Database connection DB_CONNECTION = getattr(settings, 'AUDIT_DB_CONNECTION', {'HOST': 'localhost', 'PORT': 27017, 'NAME': 'audit', 'USER': '', 'PASSWORD': ''}) # Translate URLs TRANSLATE_URLS = getattr(settings, 'AUDIT_TRANSLATE_URLS', False) # Additional indexes for the accesses. ACCESS_INDEXES = getattr(settings, 'AUDIT_ACCESS_INDEXES', []) # Additional indexes for the processes. PROCESS_INDEXES = getattr(settings, 'AUDIT_PROCESS_INDEXES', []) # Additional indexes for the model actions. MODEL_ACTION_INDEXES = getattr(settings, 'AUDIT_MODEL_ACTION_INDEXES', []) if ACTIVATE: # Create connection to database mongodb_connect(connection=DB_CONNECTION, alias=DB_ALIAS)
ACTIVATE = getattr(settings, 'AUDIT_ACTIVATE', False) # Database alias DB_ALIAS = getattr(settings, 'AUDIT_DB_ALIAS', 'audit') # Database connection DB_CONNECTION = getattr( settings, 'AUDIT_DB_CONNECTION', { 'HOST': 'localhost', 'PORT': 27017, 'NAME': 'audit', 'USER': '', 'PASSWORD': '' }) # Translate URLs TRANSLATE_URLS = getattr(settings, 'AUDIT_TRANSLATE_URLS', False) # Additional indexes for the accesses. ACCESS_INDEXES = getattr(settings, 'AUDIT_ACCESS_INDEXES', []) # Additional indexes for the processes. PROCESS_INDEXES = getattr(settings, 'AUDIT_PROCESS_INDEXES', []) # Additional indexes for the model actions. MODEL_ACTION_INDEXES = getattr(settings, 'AUDIT_MODEL_ACTION_INDEXES', []) if ACTIVATE: # Create connection to database mongodb_connect(connection=DB_CONNECTION, alias=DB_ALIAS)