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_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