Python cachemanager.CacheManager类代码示例

    def test_get_all_sources(self):
        cm = CacheManager(self.workdir)
        result1 = cm.getAllSources()
        self.assertTrue(isinstance(result1, types.GeneratorType))
        self.assertEqual(list(result1), [])

        cm.registerDoc(
            self.src_path1, self.result_path1, suffix=None)
        cm.registerDoc(
            self.src_path2, self.result_path2, suffix='foo')
        result2 = list(cm.getAllSources())
        self.assertTrue(len(result2) == 2)

        open(os.path.join(self.workdir, 'crapfile'), 'wb').write('crap')
        result3 = list(cm.getAllSources())
        self.assertFalse('crap' in result3)

        os.mkdir(os.path.join(self.workdir, 'crapdir'))
        result4 = list(cm.getAllSources())
        self.assertFalse('crapdir' in result4)

        os.makedirs(os.path.join(self.workdir, '66', 'invalid_hashdir'))
        result5 = list(cm.getAllSources())
        self.assertFalse('66' in result5)
        return

 def test_get_cached_file_by_src_failed(self, cache_env):
     # uncached files result in `None` as result
     cm = CacheManager(str(cache_env))
     result, key = cm.get_cached_file_by_source(
         str(cache_env / "src1.txt"))
     assert result is None
     assert key is None

 def test_prepare_cache_dir_broken(self, tmpdir):
     # we fail loudly if we cannot create a cache dir
     cm = CacheManager(str(tmpdir))
     tmpdir.join("not-a-dir.txt").write("foo")     # broken dir
     cm.cache_dir = str(tmpdir / "not-a-dir.txt")
     with pytest.raises(IOError):
         cm._prepare_cache_dir()

 def test_prepare_cache_dir(self, tmpdir):
     # _prepare_cache_dir creates a cache dir normally
     cm = CacheManager(str(tmpdir.join("cache")))
     new_dir = tmpdir.join("new_dir")
     cm.cache_dir = str(new_dir)
     cm._prepare_cache_dir()
     assert new_dir.isdir() is True

 def test_get_bucket_path(self, tmpdir):
     # we can get a bucket path from a hash value
     cm = CacheManager(str(tmpdir.join("cache")))
     tmpdir.join("src.txt").write("source1\n")
     hash_val = cm.get_hash(str(tmpdir / "src.txt"))
     assert cm._get_bucket_path(hash_val) == (
         tmpdir / "cache" / "73" / "737b337e605199de28b3b64c674f9422")

 def test_get_cached_file_by_src(self):
     # we can get a cached file by source file and options
     cm = CacheManager(self.workdir)
     # without a cache key
     my_id1 = cm.register_doc(self.src_path1, self.result_path1)
     result, key = cm.get_cached_file_by_source(self.src_path1)
     assert filecmp.cmp(result, self.result_path1, shallow=False)
     assert key == '737b337e605199de28b3b64c674f9422_1_1'

 def test_get_bucket_path_from_path(self):
     cm = CacheManager(self.workdir)
     path = cm._getBucketPathFromPath(self.src_path1)
     expected_path_end = os.path.join(
         '73', '737b337e605199de28b3b64c674f9422')
     self.assertEqual(os.listdir(self.workdir), [])
     self.assertTrue(path.endswith(expected_path_end))
     return

 def test_get_bucket_from_path(self):
     cache_dir_len1 = len(os.listdir(self.workdir))
     cm = CacheManager(self.workdir)
     bucket1 = cm.getBucketFromPath(self.src_path1)
     cache_dir_len2 = len(os.listdir(self.workdir))
     self.assertTrue(isinstance(bucket1, Bucket))
     self.assertTrue(cache_dir_len2 == cache_dir_len1+1)
     return

 def test_get_hash(self):
     cm = CacheManager(self.workdir)
     hash1 = cm.getHash(self.src_path1)
     hash2 = cm.getHash(self.src_path2)
     self.assertEqual(hash1, '737b337e605199de28b3b64c674f9422')
     self.assertEqual(hash2, 'd5aa51d7fb180729089d2de904f7dffe')
     self.assertRaises(TypeError, cm.getHash)
     return

 def register_fakedoc_in_cache(self, src, options):
     # register a fake doc in cache. Result cache_key is based on
     # path to src document and options given.
     cm = CacheManager(self.cachedir)
     fake_result_path = os.path.join(self.workdir, 'result.html')
     open(fake_result_path, 'w').write('A fake result.')
     marker = get_marker(options)
     cache_key = cm.register_doc(src, fake_result_path, repr_key=marker)
     return cache_key

 def test_get_cached_file(self, cache_env):
     # we can get a file cached before.
     cm = CacheManager(str(cache_env / "cache"))
     cache_key = cm.register_doc(
         str(cache_env / "src1.txt"),
         str(cache_env / "result1.txt"))
     path = cm.get_cached_file(cache_key)
     assert path is not None
     assert open(path, 'r').read() == (
         cache_env / "result1.txt").read()

 def test_markerhandling(self, tmpdir):
     # we can dissolve markers from cache_keys.
     cm = CacheManager(str(tmpdir))
     marker_string = cm._compose_cache_key('somefakedhash', 3)
     assert marker_string == "somefakedhash_3"
     hash_val, bucket_marker = cm._dissolve_cache_key("somefakedhash_3")
     assert hash_val == "somefakedhash"
     assert bucket_marker == "3"
     assert cm._dissolve_cache_key("asd") == (None, None)
     assert cm._dissolve_cache_key(None) == (None, None)

 def test_get_hash(self, cache_env, samples_dir):
     # we can compute a hash for a source file.
     cm = CacheManager(str(cache_env))
     hash1 = cm.get_hash(str(cache_env / "src1.txt"))
     hash2 = cm.get_hash(str(cache_env / "src2.txt"))
     hash3 = cm.get_hash(str(samples_dir / "testdoc1.doc"))
     assert hash1 == '737b337e605199de28b3b64c674f9422'
     assert hash2 == 'd5aa51d7fb180729089d2de904f7dffe'
     assert hash3 == '443a07e0e92b7dc6b21f8be6a388f05f'
     with pytest.raises(TypeError):
         cm.get_hash()

 def test_get_cached_file_w_key_from_file(self, cache_env):
     # we can get a cached file, stored under a key, which is a file
     cm = CacheManager(str(cache_env / "cache"))
     cache_key = cm.register_doc(
         str(cache_env / "src1.txt"),
         str(cache_env / "result1.txt"),
         repr_key=StringIO('foo'))
     path = cm.get_cached_file(cache_key)
     assert path is not None
     assert open(path, 'r').read() == (
         cache_env / "result1.txt").read()

 def test_get_cached(self):
     # we can get cached docs
     cm = CacheManager(self.cachedir)
     fake_result_path = os.path.join(self.src_dir, 'result.txt')
     open(fake_result_path, 'wb').write('The Result\n')
     key = cm.register_doc(self.src_path, fake_result_path, 'somekey')
     assert key == '2b87e29fca6ee7f1df6c1a76cb58e101_1_1'
     result_path = self.proxy.get_cached(key)
     assert result_path is not None
     assert result_path != fake_result_path
     assert filecmp.cmp(result_path, fake_result_path, shallow=False)