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)
