def test_list(self, os_mock):
os_mock.path.isfile.return_value = True
os_mock.path.getsize.side_effect = [34, 42000]
self.data.output = {
'file_list': [
{'file': 'test_01.tmp'},
{'file': 'test_02.tmp'}
]
}
hydrate_size(self.data)
self.assertEqual(self.data.output['file_list'][0]['size'], 34)
self.assertEqual(self.data.output['file_list'][1]['size'], 42000)
def test_change_size(self, os_mock):
"""Size is not changed after object is done."""
os_mock.path.isfile.return_value = True
os_mock.path.getsize.return_value = 42000
hydrate_size(self.data)
self.assertEqual(self.data.output['test_file']['size'], 42000)
os_mock.path.getsize.return_value = 43000
hydrate_size(self.data)
self.assertEqual(self.data.output['test_file']['size'], 43000)
self.data.status = Data.STATUS_DONE
os_mock.path.getsize.return_value = 44000
hydrate_size(self.data)
self.assertEqual(self.data.output['test_file']['size'], 43000)
def test_missing_file(self, os_mock):
os_mock.path.isfile.return_value = False
with self.assertRaises(ValidationError):
hydrate_size(self.data)
def test_done_data(self, os_mock):
os_mock.path.isfile.return_value = True
os_mock.path.getsize.return_value = 42000
hydrate_size(self.data)
self.assertEqual(self.data.output['test_file']['size'], 42000)
