def test_reset_Arctic(mongo_host, library_name):
arctic = Arctic(mongo_host=mongo_host)
arctic.list_libraries()
arctic.initialize_library(library_name, VERSION_STORE)
arctic[library_name]
c = arctic._conn
arctic.reset()
assert len(c.nodes) == 0
def test_reset_Arctic(mongo_host, library_name):
arctic = Arctic(mongo_host=mongo_host)
arctic.list_libraries()
arctic.initialize_library(library_name, VERSION_STORE)
c = arctic._conn
assert arctic[library_name]._arctic_lib._curr_conn is c
arctic.reset()
assert c is not arctic._conn
assert len(c.nodes) == 0
assert arctic[library_name]._arctic_lib._curr_conn is arctic._conn
def test_reset():
c = MagicMock()
with patch('pymongo.MongoClient', return_value=c, autospec=True) as mc:
store = Arctic('hostname')
# do something to trigger lazy arctic init
store.list_libraries()
store.reset()
# Doesn't matter how many times we call it:
store.reset()
c.close.assert_called_once()
def test_reset_Arctic(mongo_host, library_name):
arctic = Arctic(mongo_host=mongo_host)
arctic.list_libraries()
arctic.initialize_library(library_name, VERSION_STORE)
c = arctic._conn
assert arctic[library_name]._arctic_lib._curr_conn is c
arctic.reset()
assert c is not arctic._conn
assert len(c.nodes) == 0
assert arctic[library_name]._arctic_lib._curr_conn is arctic._conn
def test_reset():
c = MagicMock()
with patch('pymongo.MongoClient', return_value=c, autospec=True) as mc:
store = Arctic('hostname')
# do something to trigger lazy arctic init
store.list_libraries()
store.reset()
# Doesn't matter how many times we call it:
store.reset()
c.close.assert_called_once()
def test_re_authenticate_on_arctic_reset(mongo_host, library_name):
from collections import namedtuple
Cred = namedtuple('Cred', 'user, password')
with patch('arctic.arctic.authenticate') as auth_mock, \
patch('arctic.arctic.get_auth') as get_auth_mock:
auth_mock.return_value = True
get_auth_mock.return_value = Cred(user='******', password='******')
arctic = Arctic(mongo_host=mongo_host)
arctic.initialize_library(library_name, VERSION_STORE)
vstore = arctic[library_name]
vstore.list_symbols()
auth_mock.reset_mock()
arctic.reset()
assert auth_mock.call_count > 0
auth_mock.reset_mock()
vstore.list_symbols()
assert auth_mock.call_count == 0
def test_re_authenticate_on_arctic_reset(mongo_host, library_name):
from collections import namedtuple
Cred = namedtuple('Cred', 'user, password')
with patch('arctic.arctic.authenticate') as auth_mock, \
patch('arctic.arctic.get_auth') as get_auth_mock:
auth_mock.return_value = True
get_auth_mock.return_value = Cred(user='******', password='******')
arctic = Arctic(mongo_host=mongo_host)
arctic.initialize_library(library_name, VERSION_STORE)
vstore = arctic[library_name]
vstore.list_symbols()
auth_mock.reset_mock()
arctic.reset()
assert auth_mock.call_count > 0
auth_mock.reset_mock()
vstore.list_symbols()
assert auth_mock.call_count == 0
