def test_integration_validate_path_success(self):
"""
Test whether overriden validate method validates submitted path.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
# initiate a Swift service connection
conn = swiftclient.Connection(
user=settings.SWIFT_USERNAME,
key=settings.SWIFT_KEY,
authurl=settings.SWIFT_AUTH_URL,
)
# create container in case it doesn't already exist
conn.put_container(settings.SWIFT_CONTAINER_NAME)
# upload file to Swift storage
with io.StringIO("test file") as file1:
conn.put_object(settings.SWIFT_CONTAINER_NAME, path, contents=file1.read(),
content_type='text/plain')
for _ in range(20):
object_list = conn.get_container(settings.SWIFT_CONTAINER_NAME, prefix=path)[1]
if object_list:
break
time.sleep(0.2)
self.assertEqual(servicefiles_serializer.validate(data).get('path'), path)
# delete file from Swift storage
conn.delete_object(settings.SWIFT_CONTAINER_NAME, path)
def test_validate_service_name_failure_registered_service(self):
"""
Test whether overriden validate_name method validates whether submitted
unregistered service name is actually the name of a registered service.
"""
servicefiles_serializer = ServiceFileSerializer()
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate_service_name('PACS')
def test_integration_validate_path_failure_does_not_exist(self):
"""
Test whether overriden validate method validates that submitted path exists
in internal storage.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate(data)
def test_validate_failure_path_does_not_start_with_SERVICES_PACS(self):
"""
Test whether overriden validate method validates submitted path must start
with the 'SERVICES/<service_name>/' string.
"""
path = 'SERVICES/Other/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate(data)
def test_validate_service_name_success(self):
"""
Test whether overriden validate_name method successfully returns a valid
unregistered service name.
"""
Service.objects.get_or_create(identifier='NewService')
servicefiles_serializer = ServiceFileSerializer()
self.assertEqual(servicefiles_serializer.validate_service_name('MyService'),
'MyService')
self.assertEqual(servicefiles_serializer.validate_service_name('NewService'),
'NewService')
def test_validate_failure_path_does_not_exist(self):
"""
Test whether overriden validate method validates that submitted path exists
in internal storage.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
with mock.patch.object(SwiftManager, 'obj_exists',
return_value=False) as obj_exists_mock:
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate(data)
obj_exists_mock.assert_called_with(path.strip(' ').strip('/'))
def test_validate_updates_validated_data(self):
"""
Test whether overriden validate method updates validated data with the descriptors
embedded in the path string.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
with mock.patch.object(SwiftManager, 'obj_exists',
return_value=True) as obj_exists_mock:
new_data = servicefiles_serializer.validate(data)
self.assertIn('service', new_data)
self.assertNotIn('service_name', new_data)
self.assertEqual(new_data.get('path'), path.strip(' ').strip('/'))
obj_exists_mock.assert_called_with(new_data.get('path'))
def test_validate_validates_path_has_not_already_been_registered(self):
"""
Test whether overriden validate method validates that the submitted path
has not been already registered.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
service = Service(identifier='MyService')
service.save()
service_file = ServiceFile(service=service)
service_file.fname.name = path
service_file.save()
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate(data)
def test_integration_validate_path_success(self):
"""
Test whether overriden validate method validates submitted path.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
swift_manager = SwiftManager(settings.SWIFT_CONTAINER_NAME,
settings.SWIFT_CONNECTION_PARAMS)
# upload file to Swift storage
with io.StringIO("test file") as file1:
swift_manager.upload_obj(path, file1.read(), content_type='text/plain')
for _ in range(20):
if swift_manager.obj_exists(path):
break
time.sleep(0.2)
self.assertEqual(servicefiles_serializer.validate(data).get('path'), path)
# delete file from Swift storage
swift_manager.delete_obj(path)
def test_validate_failure_path_does_not_exist(self):
"""
Test whether overriden validate method validates that submitted path exists
in internal storage.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
object_list = []
container_data = ['', object_list]
with mock.patch.object(swiftclient.Connection, '__init__',
return_value=None) as conn_init_mock:
with mock.patch.object(swiftclient.Connection, 'get_container',
return_value=container_data) as conn_get_container_mock:
with self.assertRaises(serializers.ValidationError):
servicefiles_serializer.validate(data)
conn_init_mock.assert_called_with(user=settings.SWIFT_USERNAME,
key=settings.SWIFT_KEY,
authurl=settings.SWIFT_AUTH_URL)
conn_get_container_mock.assert_called_with(settings.SWIFT_CONTAINER_NAME,
prefix=path)
def test_validate_updates_validated_data(self):
"""
Test whether overriden validate method updates validated data with the descriptors
embedded in the path string.
"""
path = 'SERVICES/MyService/123456-crazy/brain_crazy_study/brain_crazy_mri/file1.dcm'
data = {'service_name': 'MyService', 'path': path}
servicefiles_serializer = ServiceFileSerializer()
object_list = [{'name': path}]
container_data = ['', object_list]
with mock.patch.object(swiftclient.Connection, '__init__',
return_value=None) as conn_init_mock:
with mock.patch.object(swiftclient.Connection, 'get_container',
return_value=container_data) as conn_get_container_mock:
new_data = servicefiles_serializer.validate(data)
self.assertIn('service', new_data)
self.assertNotIn('service_name', new_data)
self.assertEqual(new_data.get('path'), path.strip(' ').strip('/'))
conn_init_mock.assert_called_with(user=settings.SWIFT_USERNAME,
key=settings.SWIFT_KEY,
authurl=settings.SWIFT_AUTH_URL)
conn_get_container_mock.assert_called_with(settings.SWIFT_CONTAINER_NAME,
prefix=path)