def test_done_data(self, os_mock):
data = self.create_data(os_mock, self.contributor, self.process)
os_mock.path.isfile.return_value = True
os_mock.path.getsize.return_value = 42000
hydrate_size(data)
self.assertEqual(data.output['test_file']['size'], 42000)
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_data_with_refs(self, os_mock):
os_mock.path.isfile.return_value = True
os_mock.path.getsize.side_effect = [42000, 8000, 42]
self.data.output = {
'test_file': {
'file': 'test_file.tmp',
'refs': ['ref_file1.tmp', 'ref_file2.tmp']
}
}
hydrate_size(self.data)
self.assertEqual(self.data.output['test_file']['size'], 42000)
self.assertEqual(self.data.output['test_file']['total_size'], 50042)
self.assertEqual(self.data.size, 50042)
def test_data_with_refs(self, os_mock):
data = self.create_data(os_mock, self.contributor, self.process)
os_mock.path.isfile.return_value = True
os_mock.path.getsize.side_effect = [42000, 8000, 42]
data.output = {
'test_file': {
'file': 'test_file.tmp',
'refs': ['ref_file1.tmp', 'ref_file2.tmp']
}
}
hydrate_size(data)
self.assertEqual(data.output['test_file']['size'], 42000)
self.assertEqual(data.output['test_file']['total_size'], 50042)
self.assertEqual(data.size, 50042)
def test_list(self, os_mock):
os_mock.path.isfile.return_value = True
os_mock.path.getsize.side_effect = [34, 42000, 42]
self.data.output = {
'file_list': [
{
'file': 'test_01.tmp',
},
{
'file': 'test_02.tmp',
'refs': ['ref_file1.tmp']
},
]
}
hydrate_size(self.data)
self.assertEqual(self.data.output['file_list'][0]['size'], 34)
self.assertEqual(self.data.output['file_list'][0]['total_size'], 34)
self.assertEqual(self.data.output['file_list'][1]['size'], 42000)
self.assertEqual(self.data.output['file_list'][1]['total_size'], 42042)
self.assertEqual(self.data.size, 34 + 42042)
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_change_size(self, os_mock):
"""Size is not changed after object is done."""
data = self.create_data(os_mock, self.contributor, self.process)
os_mock.path.isfile.return_value = True
os_mock.path.getsize.return_value = 42000
hydrate_size(data)
self.assertEqual(data.output['test_file']['size'], 42000)
os_mock.path.getsize.return_value = 43000
hydrate_size(data)
self.assertEqual(data.output['test_file']['size'], 43000)
data.status = Data.STATUS_DONE
os_mock.path.getsize.return_value = 44000
hydrate_size(data)
self.assertEqual(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)
def test_missing_file(self, os_mock):
data = self.create_data(os_mock, self.contributor, self.process)
os_mock.path.isfile.return_value = False
with self.assertRaises(ValidationError):
hydrate_size(data)
