def test_counts_secondary_files(self):
file = self.make_file(1000)
file['secondaryFiles'] = [
self.make_file(100),
self.make_file(200),
self.make_file(300)
]
self.assertEqual(total_size(file), 1600)
def test_counts_in_array(self):
outputs = {
'files':
[self.make_file(100),
self.make_file(200),
self.make_file(300)]
}
self.assertEqual(total_size(outputs), 600)
def test_counts_nested(self):
outputs = {
'files':
[self.make_file(100),
self.make_file(200),
self.make_file(300)],
'things': {
'nested_files': [
self.make_file(1000),
self.make_file(2000),
self.make_file(3000)
]
}
}
self.assertEqual(total_size(outputs), 6600)
def test_defaults_zero_when_no_size(self):
file = self.make_file()
self.assertNotIn('size', file)
self.assertEqual(total_size(file), 0)
def test_total_size_file_objects(self):
outputs = {
'file1': self.make_file(100, 'file1'),
'file2': self.make_file(200, 'file2')
}
self.assertEqual(total_size(outputs), 300)
def test_total_size_direct(self):
file = self.make_file(100)
self.assertEqual(total_size(file), 100)