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)
