def _create_tmp_cache_file(self, tmp_dir, serializer):
"""
Helper function that creates a temporary cache file using localfs.store. This
is to used to create DRY unit tests for the localfs cache.
"""
with patch.dict(localfs.__opts__, {"cachedir": tmp_dir}):
with patch.dict(localfs.__context__, {"serial": serializer}):
localfs.store(bank="bank", key="key", data="payload data")
def _create_tmp_cache_file(self, tmp_dir, serializer):
'''
Helper function that creates a temporary cache file using localfs.store. This
is to used to create DRY unit tests for the localfs cache.
'''
with patch.dict(localfs.__opts__, {'cachedir': tmp_dir}):
with patch.dict(localfs.__context__, {'serial': serializer}):
localfs.store(bank='bank', key='key', data='payload data')
def _create_tmp_cache_file(self, tmp_dir):
"""
Helper function that creates a temporary cache file using localfs.store. This
is to used to create DRY unit tests for the localfs cache.
"""
self.addCleanup(shutil.rmtree, tmp_dir)
with patch.dict(localfs.__opts__, {"cachedir": tmp_dir}):
localfs.store(bank="bank", key="key", data="payload data", cachedir=tmp_dir)
def test_mix_of_utf8_and_non_utf8_can_be_round_tripped(self):
tmp_dir = tempfile.mkdtemp(dir=RUNTIME_VARS.TMP)
data = {
# Any unicode, which ideally is invalid ascii.
"unicode": "áéí",
# Any bytes so long as they're not valid utf-8
"bytes": b"\xfe\x99\x00\xff",
}
bank = "bank"
key = "key"
self.addCleanup(shutil.rmtree, tmp_dir)
with patch.dict(localfs.__opts__, {"cachedir": tmp_dir}):
localfs.store(bank, key, data, tmp_dir)
actual = localfs.fetch(bank, key, tmp_dir)
self.assertEqual(data, actual)