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