def testGetChunkBlocks3(self):
"""
Test the `getChunkBlocks` method and especially the parent/child
relationship
"""
primDict = {"block_A": {"blockSize": 1, "locations": ["Site_A"]},
"block_B": {"blockSize": 2, "locations": ["Site_B"]}}
parentDict = {"parent_A": {"blockSize": 11, "locations": ["Site_A"]},
"parent_B": {"blockSize": 12, "locations": ["Site_B"]},
"parent_C": {"blockSize": 13, "locations": ["Site_A", "Site_B"]}}
parentage = {"block_A": ["parent_B", "parent_D"], # parent_D has no replicas!
"block_B": ["parent_A", "parent_C"]}
wflow = Workflow("workflow_1", {"RequestType": "TaskChain",
"InputDataset": "Dataset_name_XXX"})
# now set a parent
wflow.setParentDataset("Parent_dataset_XXX")
wflow.setPrimaryBlocks(primDict)
wflow.setParentBlocks(parentDict)
wflow.setChildToParentBlocks(parentage)
blockChunks, sizeChunks = wflow.getChunkBlocks(1)
self.assertEqual(len(blockChunks), 1)
self.assertItemsEqual(blockChunks[0], {"block_A", "block_B", "parent_A", "parent_B", "parent_C"})
self.assertEqual(len(sizeChunks), 1)
self.assertEqual(sizeChunks[0], 39)
blockChunks, sizeChunks = wflow.getChunkBlocks(2)
self.assertEqual(len(blockChunks), 2)
self.assertItemsEqual(blockChunks[0], {"block_B", "parent_A", "parent_C"})
self.assertItemsEqual(blockChunks[1], {"block_A", "parent_B"})
self.assertEqual(len(sizeChunks), 2)
self.assertEqual(sizeChunks[0], 26)
self.assertEqual(sizeChunks[1], 13)
def testParentageRelationship(self):
"""
Test methods related to the primary and parent datasets and blocks
"""
primDict = {"block_A": {"blockSize": 1, "locations": ["Site_A"]},
"block_B": {"blockSize": 2, "locations": ["Site_B"]}}
parentDict = {"parent_A": {"blockSize": 11, "locations": ["Site_A"]},
"parent_B": {"blockSize": 12, "locations": ["Site_B"]},
"parent_C": {"blockSize": 13, "locations": ["Site_A", "Site_B"]}}
parentage = {"block_A": ["parent_B", "parent_D"], # parent_D has no replicas!
"block_B": ["parent_A", "parent_C"]}
wflow = Workflow("workflow_1", {"RequestType": "TaskChain",
"InputDataset": "Dataset_name_XXX",
"IncludeParents": True})
self.assertEqual(wflow.getParentDataset(), "")
wflow.setParentDataset("Parent_dataset_XXX")
self.assertEqual(wflow.getParentDataset(), "Parent_dataset_XXX")
self.assertEqual(wflow.getPrimaryBlocks(), {})
wflow.setPrimaryBlocks(primDict)
self.assertItemsEqual(list(wflow.getPrimaryBlocks()), ["block_A", "block_B"])
self.assertEqual(wflow.getParentBlocks(), {})
wflow.setParentBlocks(parentDict)
self.assertItemsEqual(list(wflow.getParentBlocks()), ["parent_A", "parent_B", "parent_C"])
self.assertEqual(wflow.getChildToParentBlocks(), {})
wflow.setChildToParentBlocks(parentage)
self.assertItemsEqual(wflow.getChildToParentBlocks(), parentage)
