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