def get_seed_output_files(self):
"""Get the list of SeedOutputFiles typed results
:return: list of output files elements
:rtype: [`job.seed.types.SeedOutputFiles`]
"""
return [SeedOutputFiles(x) for x in self.get_output_files()]
def test_capture_output_files_multiple(self, get_files):
output_files = [SeedOutputFiles(self.test_output_snippet)]
name = 'OUTPUT_TIFFS'
get_files.return_value = ['outfile0.tif', 'outfile1.tif']
outputs = JobResults()._capture_output_files(output_files)
self.assertIn(name, outputs)
files = outputs[name]
self.assertEqual(len(files), 2)
self.assertEqual(files[0].__dict__, ProductFileMetadata(name, 'outfile0.tif', media_type='image/tiff').__dict__)
self.assertEqual(files[1].__dict__, ProductFileMetadata(name, 'outfile1.tif', media_type='image/tiff').__dict__)
def test_capture_output_files_metadata(self, get_files):
output_files = [SeedOutputFiles(self.test_output_snippet)]
name = 'OUTPUT_TIFFS'
get_files.return_value = ['outfile0.tif']
metadata = {
'type': 'Feature',
'geometry': {
'type': 'Point',
'coordinates': [0, 1]
},
'properties': {
'dataStarted': '2018-06-01T00:00:00Z',
'dataEnded': '2018-06-01T01:00:00Z',
'sourceStarted': '2018-06-01T00:00:00Z',
'sourceEnded': '2018-06-01T06:00:00Z',
'sourceSensorClass': 'Platform',
'sourceSensor': 'X1',
'sourceCollection': '12345A',
'sourceTask': 'Calibration'
}
}
metadata_name = 'outfile0.tif.metadata.json'
with open(metadata_name, 'w') as metadata_file:
json.dump(metadata, metadata_file)
outputs = JobResults()._capture_output_files(output_files)
os.remove(metadata_name)
self.assertIn(name, outputs)
files = outputs[name]
self.assertEqual(len(files), 1)
self.assertDictEqual(
files[0].__dict__,
ProductFileMetadata(output_name=name,
local_path='outfile0.tif',
media_type='image/tiff',
data_start='2018-06-01T00:00:00Z',
data_end='2018-06-01T01:00:00Z',
geojson=metadata,
source_started='2018-06-01T00:00:00Z',
source_ended='2018-06-01T06:00:00Z',
source_sensor_class='Platform',
source_sensor='X1',
source_collection='12345A',
source_task='Calibration').__dict__)