def commit(self): if self.isNew(): raise Exception( "unable to commit action that has not been created, proxy does not exist" ) self.stub.Commit(filter_pb2.ActionCommitRequest(action=self.data), timeout=Cuebot.Timeout)
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)