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)