def test__create_subject_set(self, mockSave, mockFind):
operation = UploadSubject(None)
created_set = operation._create_subject_set(1, 'some name')
mockFind.assert_called_once_with(1)
mockSave.assert_called_once()
assert (created_set.display_name == 'some name')
assert (mockFind.return_value == created_set.links.project.id) # weird
def test__create_subject_no_metadata(self, mockAdd, mockSave, mockFind):
operation = UploadSubject(None)
created_subject = operation._create_subject(1, 'some_file')
mockFind.assert_called_once_with(1)
mockAdd.assert_called_once_with(created_subject, 'some_file') # weird
mockSave.assert_called_once()
assert (mockFind.return_value == created_subject.links.project.id
) # weird
def test__connect(self, mockSecret, mockId, mockUrl, mockConnect):
operation = UploadSubject(JobBundle())
operation._connect()
mockUrl.assert_called_once()
mockId.assert_called_once()
mockSecret.assert_called_once()
mockConnect.assert_called_once_with(endpoint='sample url',
client_id='sample id',
client_secret='sample secret')
def test_apply_multiple(self, mockAdd, mockCreate, mockGet, mockConnect,
mockGetName, *args):
project = Project(id=8)
pipeline = Pipeline(project=project, multiple_subject_sets=True)
bundle = JobBundle(pipeline=pipeline)
operation = UploadSubject(bundle)
operation.apply(['some_file'])
mockConnect.assert_called_once()
mockGetName.assert_called_once()
mockGet.assert_called_once_with(bundle, 8, 'bundle name')
mockCreate.assert_called_once_with(8, 'some_file')
mockAdd.assert_called_once_with(mockCreate.return_value)
def test_apply_single(self, mockOpen, mockAdd, mockCreate, mockGet,
mockGetName, mockIncludeMetadata, *args):
project = Project(id=8)
pipeline = Pipeline(project=project)
bundle = JobBundle(pipeline=pipeline)
operation = UploadSubject(bundle)
operation.apply(['some_file'])
mockOpen.assert_called_once()
mockGetName.assert_called_once()
mockGet.assert_called_once_with(pipeline, 8, 'pipeline name')
mockCreate.assert_called_once_with(8, 'some_file', metadata={})
mockAdd.assert_called_once_with(mockCreate.return_value)
def test__get_subject_set(self, mockFind, mockCreateSet, *args):
mockFind.reset_mock()
mockCreateSet.reset_mock()
emptyJobBundle = JobBundle()
linkedJobBundle = JobBundle(subject_set_id=3)
emptyPipeline = Pipeline()
linkedPipeline = Pipeline(subject_set_id=3)
operation = UploadSubject(emptyJobBundle)
result = operation._get_subject_set(emptyJobBundle, 8, 'some name')
mockFind.assert_not_called()
mockCreateSet.assert_called_once_with(8, 'some name')
mockFind.reset_mock()
mockCreateSet.reset_mock()
operation = UploadSubject(linkedJobBundle)
result = operation._get_subject_set(linkedJobBundle, 8, 'some name')
mockFind.assert_called_once_with(3)
mockCreateSet.assert_not_called()
mockFind.reset_mock()
mockCreateSet.reset_mock()
operation = UploadSubject(emptyPipeline)
result = operation._get_subject_set(emptyPipeline, 8, 'some name')
mockFind.assert_not_called()
mockCreateSet.assert_called_once_with(8, 'some name')
mockFind.reset_mock()
mockCreateSet.reset_mock()
operation = UploadSubject(linkedPipeline)
result = operation._get_subject_set(linkedPipeline, 8, 'some name')
mockFind.assert_called_once_with(3)
mockCreateSet.assert_not_called()