def testCommitNew(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock action = opencue.wrappers.filter.Action() with self.assertRaises(Exception): action.commit()
def testSetTypeAndValueFail(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID)) with self.assertRaises(Exception): action.setTypeAndValue('foo', 'bar')
def testCommit(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID)) action.commit() stubMock.Commit.assert_called_with( filter_pb2.ActionCommitRequest(action=action.data), timeout=mock.ANY)
def testSetTypeAndValueNone(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock testData = (filter_pb2.STOP_PROCESSING, None) action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID)) action.setTypeAndValue(testData[0], testData[1]) stubMock.Commit.assert_called_with( filter_pb2.ActionCommitRequest(action=action.data), timeout=mock.ANY)
def testSetTypeAndValueString(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock testData = (filter_pb2.SET_ALL_RENDER_LAYER_TAGS, 'testString') action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID)) action.setTypeAndValue(testData[0], testData[1]) stubMock.Commit.assert_called_with( filter_pb2.ActionCommitRequest(action=action.data), timeout=mock.ANY)
def testSetTypeAndValueGroup(self, getStubMock): stubMock = mock.Mock() stubMock.Commit.return_value = filter_pb2.ActionCommitResponse() getStubMock.return_value = stubMock testData = (filter_pb2.MOVE_JOB_TO_GROUP, job_pb2.Group(id='testGroup')) action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID)) action.setTypeAndValue(testData[0], testData[1]) stubMock.Commit.assert_called_with( filter_pb2.ActionCommitRequest(action=action.data), timeout=mock.ANY)