def test_get_all_source_paths_ignore_nonsense(self): bucket = Bucket(self.workdir) open(os.path.join(self.workdir, 'sources', 'source3'), 'wb').write('Hi') open(os.path.join(self.workdir, 'sources', 'foo_3'), 'wb').write('Hi') paths1 = list(bucket.getAllSourcePaths()) bucket.storeResult( self.src_path1, self.result_path1, suffix='foo') paths2 = list(bucket.getAllSourcePaths()) self.assertEqual(paths1, []) self.assertEqual(len(paths2), 1)
def test_get_all_source_paths(self): bucket = Bucket(self.workdir) paths = bucket.getAllSourcePaths() self.assertTrue( isinstance(paths, types.GeneratorType)) paths = list(paths) self.assertEqual(paths, []) bucket.storeResult( self.src_path1, self.result_path1, suffix='foo') paths = list(bucket.getAllSourcePaths()) self.assertTrue(paths[0].endswith('source_1')) self.assertEqual(len(paths), 1) bucket.storeResult( self.src_path2, self.result_path2, suffix='bar') paths = list(bucket.getAllSourcePaths()) self.assertEqual(len(paths), 2) return