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_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)
