def test_write_file_consistency_present(self, create_mock, create_share_mock, put_file_from_path_mock, get_file_properties_mock): from azure.common import AzureMissingResourceHttpError # given items_to_write = self.get_batch() options = self.get_writer_config() options['options']['check_consistency'] = True get_file_properties_mock.side_effect = AzureMissingResourceHttpError( '', 404) # when: writer = AzureFileWriter(options, meta()) try: writer.write_batch(items_to_write) writer.flush() finally: writer.close() with self.assertRaisesRegexp(InconsistentWriteState, 'Missing file'): writer.finish_writing() # then: self.assertEqual(writer.get_metadata('items_count'), 2)
def test_write_file_consistency_size(self, create_mock, create_share_mock, put_file_from_path_mock, get_file_properties_mock): from azure.storage.file.models import File, FileProperties # given items_to_write = self.get_batch() options = self.get_writer_config() options['options']['check_consistency'] = True fake_properties = FileProperties() fake_properties.content_length = 999 get_file_properties_mock.return_value = File(props=fake_properties) # when: writer = AzureFileWriter(options, ExportMeta(options)) try: writer.write_batch(items_to_write) writer.flush() finally: writer.close() with self.assertRaises(InconsistentWriteState): writer.finish_writing()
def test_write_file_consistency_present(self, create_mock, create_share_mock, put_file_from_path_mock, get_file_properties_mock): from azure.common import AzureMissingResourceHttpError # given items_to_write = self.get_batch() options = self.get_writer_config() options['options']['check_consistency'] = True get_file_properties_mock.side_effect = AzureMissingResourceHttpError('', 404) # when: writer = AzureFileWriter(options, meta()) try: writer.write_batch(items_to_write) writer.flush() finally: writer.close() with self.assertRaisesRegexp(InconsistentWriteState, 'Missing file'): writer.finish_writing() # then: self.assertEqual(writer.get_metadata('items_count'), 2)