def test_load_client_module_class_missing(import_module):
import_module.return_value = stub(foo='bar')
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'Neo4j client module' in e.message
raise e
def test_load_client_fail_module_import(import_module):
import_module.side_effect = ImportError
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'Could not import' in e.message
raise e
def test_load_client_module_class_missing(import_module):
import_module.return_value = stub(foo='bar')
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'Neo4j client module' in e.message
raise e
def test_load_client_fail_module_import(import_module):
import_module.side_effect = ImportError
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'Could not import' in e.message
raise e
def test_load_client_not_correct_subclass(import_module):
MyClass = type('MyClass', (object, ), {})
import_module.return_value = stub(baz=MyClass)
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'is not a subclass of EnhancedGraphDatabase' in e.message
raise e
def test_load_client_not_correct_subclass(import_module):
MyClass = type('MyClass', (object,), {})
import_module.return_value = stub(baz=MyClass)
try:
utils.load_client('foo.bar.baz')
except ImproperlyConfigured as e:
assert 'is not a subclass of EnhancedGraphDatabase' in e.message
raise e
def test_load_client(import_module):
MyClass = type('MyClass', (EnhancedGraphDatabase, ), {})
import_module.return_value = stub(baz=MyClass)
assert utils.load_client('foo.bar.baz') == MyClass
def test_load_client(import_module):
MyClass = type('MyClass', (EnhancedGraphDatabase,), {})
import_module.return_value = stub(baz=MyClass)
assert utils.load_client('foo.bar.baz') == MyClass
