def test_use_shelve_store(unittest, tmpdir, test_data):
initialize_store(test_data)
contents_before = get_store_contents()
type_before = get_store_type()
directory = tmpdir.mkdir("test_use_shelve_store").dirname
global_state.use_shelve_store(directory)
contents_after = get_store_contents()
type_after = get_store_type()
unittest.assertEqual(contents_before, contents_after)
unittest.assertNotEqual(type_before, type_after)
global_state.cleanup(data_id="1")
unittest.assertNotEqual(contents_after, get_store_contents())
def test_use_default_store(unittest, tmpdir, test_data):
"""Make sure flipping back and forth multiple times doesn't corrupt the data"""
initialize_store(test_data)
contents_before = get_store_contents()
type_before = get_store_type()
directory = tmpdir.mkdir('test_use_default_store').dirname
global_state.use_shelve_store(directory)
global_state.use_default_store()
contents_after = get_store_contents()
type_after = get_store_type()
unittest.assertEqual(contents_before, contents_after)
unittest.assertEqual(type_before, type_after)
