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_contains(self):
cm = CacheManager(self.workdir)
self.assertRaises(TypeError, cm.contains)
self.assertRaises(TypeError, cm.contains, path='foo', marker='bar')
self.assertFalse(
cm.contains(marker='737b337e605199de28b3b64c674f9422_1'))
cm.registerDoc(
self.src_path1, self.result_path1, suffix=None)
cm.registerDoc(
self.src_path2, self.result_path2, suffix='foo')
result1 = cm.contains(path=self.src_path1)
result2 = cm.contains(marker='737b337e605199de28b3b64c674f9422_1')
result3 = cm.contains(marker='737b337e605199de28b3b64c674f9422_1',
suffix='foo')
result4 = cm.contains(marker='737b337e605199de28b3b64c674f9422_1',
suffix='bar')
result5 = cm.contains(marker='d5aa51d7fb180729089d2de904f7dffe_1',
suffix='foo')
self.assertTrue(result1)
self.assertTrue(result2)
self.assertFalse(result3)
self.assertFalse(result4)
self.assertTrue(result5)
return
def test_register_doc(self):
cm = CacheManager(self.workdir)
marker1 = cm.registerDoc(
self.src_path1, self.result_path1, suffix=None)
marker2 = cm.registerDoc(
self.src_path1, self.result_path1, suffix=None)
marker3 = cm.registerDoc(
self.src_path1, self.result_path2, suffix='foo')
marker4 = cm.registerDoc(
self.src_path2, self.result_path2, suffix='foo')
self.assertEqual(marker1, '737b337e605199de28b3b64c674f9422_1')
self.assertEqual(marker2, '737b337e605199de28b3b64c674f9422_1')
self.assertEqual(marker3, '737b337e605199de28b3b64c674f9422_1')
self.assertEqual(marker4, 'd5aa51d7fb180729089d2de904f7dffe_1')
return
def test_get_cached_file_from_marker(self):
cm = CacheManager(self.workdir)
path1 = cm.getCachedFileFromMarker('not-a-marker')
path2 = cm.getCachedFileFromMarker('737b337e605199de28b3b64c674f9422_1')
path3 = cm.getCachedFileFromMarker('not-a-marker-with_underscore')
marker1 = cm.registerDoc(
self.src_path1, self.result_path1, suffix=None)
path4 = cm.getCachedFileFromMarker(marker1)
marker2 = cm.registerDoc(
self.src_path1, self.result_path2, suffix='foo')
path5 = cm.getCachedFileFromMarker(marker2, suffix='foo')
self.assertTrue(path1 is None)
self.assertTrue(path2 is None)
self.assertTrue(path3 is None)
self.assertTrue(path4 is not None)
self.assertTrue(path5 is not None)
return
def test_get_cached_file(self):
cm = CacheManager(self.workdir)
path = cm.getCachedFile(self.src_path1)
self.assertTrue(path is None)
self.assertEqual(os.listdir(self.workdir), [])
cm.registerDoc(self.src_path1, self.result_path1, suffix=None)
path1 = cm.getCachedFile(self.src_path1)
path2 = cm.getCachedFile(self.src_path1, suffix='bar')
path3 = cm.getCachedFile(self.src_path1, suffix='foo')
self.assertTrue(path1 is not None)
self.assertTrue(path2 is None)
self.assertTrue(path3 is None)
cm.registerDoc(self.src_path2, self.result_path1, suffix='foo')
path1 = cm.getCachedFile(self.src_path2)
path2 = cm.getCachedFile(self.src_path2, suffix='bar')
path3 = cm.getCachedFile(self.src_path2, suffix='foo')
self.assertTrue(path1 is None)
self.assertTrue(path2 is None)
self.assertTrue(path3 is not None)
return
