def test_dataset(self):
dataset = 'dataset1'
filters = {
'dataset': dataset,
'year': 2015,
'month': 9,
'day': 1,
}
current_level = 'dataset'
directories = ['dataset1', 'dataset2', 'dataset3']
dry_run = False
root = "/root"
process_directories(root, directories, filters, current_level, dry_run)
self.assertEqual(len(directories), 1)
self.assertEqual(directories[0], 'dataset1')
def test_no_directories_to_process(self, remove_directory_mock):
dataset = None
filters = {
'dataset': dataset,
'year': 2014,
'month': 9,
'day': 1,
}
current_level = 'day'
directories = []
root = "/root"
dry_run = False
process_directories(root, directories, filters, current_level, dry_run)
self.assertEqual(len(directories), 0)
self.assertEqual(remove_directory_mock.call_count, 0)
def test_year(self, remove_directory_mock):
dataset = None
filters = {
'dataset': dataset,
'year': 2014,
'month': 9,
'day': 1,
}
current_level = 'year'
directories = ['2013', '2014', '2015']
dry_run = False
root = "/root/dataset"
process_directories(root, directories, filters, current_level, dry_run)
self.assertEqual(len(directories), 1)
self.assertTrue('2014' in directories)
remove_directory_mock.assert_has_calls([mock.call('/root/dataset/2013', False)])
