def test_get_arctic_connection_strings():
with patch('arctic.arctic.Arctic') as Arctic:
with patch('pymongo.MongoClient') as MongoClient:
get_arctic_lib("foo@bar")
get_arctic_lib("foo.sheep@bar")
get_arctic_lib("foo.sheep@bar:123")
get_arctic_lib("[email protected]:123")
def test_get_arctic_connection_strings():
with patch('arctic.arctic.Arctic') as Arctic:
with patch('pymongo.MongoClient') as MongoClient:
get_arctic_lib("foo@bar")
get_arctic_lib("foo.sheep@bar")
get_arctic_lib("foo.sheep@bar:123")
get_arctic_lib("[email protected]:123")
def test_get_arctic_lib_with_known_host():
with patch('arctic.arctic.Arctic') as Arctic:
get_arctic_lib("foo@bar")
assert Arctic.call_args_list == [call('bar')]
def test_get_arctic_malformed_connection_strings(string):
with pytest.raises(ValueError):
get_arctic_lib(string)
def test_get_arctic_lib_with_unknown_host():
with patch('arctic.arctic.Arctic') as Arctic:
with patch('pymongo.MongoClient') as MongoClient:
get_arctic_lib("foo@bar:123")
assert Arctic.call_args_list == [call("bar:123")]
def test_get_arctic_lib_with_known_host():
with patch('arctic.arctic.Arctic') as Arctic:
get_arctic_lib("foo@bar")
assert Arctic.call_args_list == [call('bar')]
def test_get_arctic_malformed_connection_strings(string):
with pytest.raises(ValueError):
get_arctic_lib(string)
def test_get_arctic_lib_with_unknown_host():
with patch('arctic.arctic.Arctic') as Arctic:
with patch('pymongo.MongoClient') as MongoClient:
get_arctic_lib("foo@bar:123")
assert Arctic.call_args_list == [call("bar:123")]
