def testCreateAction(self, getStubMock):
actionId = 'aaa-aaaa-aaa'
stubMock = mock.Mock()
stubMock.CreateAction.return_value = filter_pb2.FilterCreateActionResponse(
action=filter_pb2.Action(id=actionId))
getStubMock.return_value = stubMock
actionType = filter_pb2.PAUSE_JOB
value = 10
filter = opencue.wrappers.filter.Filter(
filter_pb2.Filter(name=TEST_FILTER_NAME))
action = filter.createAction(actionType, value)
actionData = opencue.wrappers.filter.ActionData(
type=actionType,
value_type=filter_pb2.INTEGER_TYPE,
group_value=None,
string_value=None,
integer_value=value,
float_value=0.0,
boolean_value=False)
stubMock.CreateAction.assert_called_with(
filter_pb2.FilterCreateActionRequest(filter=filter.data,
data=actionData),
timeout=mock.ANY)
self.assertEqual(action.id(), actionId)
def testOrderLast(self, getStubMock):
stubMock = mock.Mock()
stubMock.OrderLast.return_value = filter_pb2.FilterOrderLastResponse()
getStubMock.return_value = stubMock
filterInst = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterInst.orderLast()
stubMock.OrderLast.assert_called_with(
filter_pb2.FilterOrderLastRequest(filter=filterInst.data), timeout=mock.ANY)
def testDelete(self, getStubMock):
stubMock = mock.Mock()
stubMock.Delete.return_value = filter_pb2.FilterDeleteResponse()
getStubMock.return_value = stubMock
filterToDelete = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToDelete.delete()
stubMock.Delete.assert_called_with(
filter_pb2.FilterDeleteRequest(filter=filterToDelete.data), timeout=mock.ANY)
def testSetType(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetType.return_value = filter_pb2.FilterSetTypeResponse()
getStubMock.return_value = stubMock
value = filter_pb2.MATCH_ALL
filterToSet = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToSet.setType(value)
stubMock.SetType.assert_called_with(
filter_pb2.FilterSetTypeRequest(filter=filterToSet.data, type=value), timeout=mock.ANY)
def testRaiseOrder(self, getStubMock):
stubMock = mock.Mock()
stubMock.RaiseOrder.return_value = filter_pb2.FilterRaiseOrderResponse()
getStubMock.return_value = stubMock
filter = opencue.wrappers.filter.Filter(
filter_pb2.Filter(name=TEST_FILTER_NAME))
filter.raiseOrder()
stubMock.RaiseOrder.assert_called_with(
filter_pb2.FilterRaiseOrderRequest(filter=filter.data), timeout=mock.ANY)
def testRunFilterOnGroup(self, getStubMock):
stubMock = mock.Mock()
stubMock.RunFilterOnGroup.return_value = filter_pb2.FilterRunFilterOnGroupResponse()
getStubMock.return_value = stubMock
group = opencue.wrappers.group.Group(job_pb2.Group(name='testGroup'))
filterToRun = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToRun.runFilterOnGroup(group)
stubMock.RunFilterOnGroup.assert_called_with(
filter_pb2.FilterRunFilterOnGroupRequest(filter=filterToRun.data, group=group.data),
timeout=mock.ANY)
def testSetEnabled(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetEnabled.return_value = filter_pb2.FilterSetEnabledResponse()
getStubMock.return_value = stubMock
value = True
filterToEnable = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToEnable.setEnabled(value)
stubMock.SetEnabled.assert_called_with(
filter_pb2.FilterSetEnabledRequest(
filter=filterToEnable.data, enabled=value), timeout=mock.ANY)
def testGetParentFilter(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetParentFilter.return_value = filter_pb2.MatcherGetParentFilterResponse(
filter=filter_pb2.Filter(name=TEST_FILTER_NAME))
getStubMock.return_value = stubMock
matcher = opencue.wrappers.filter.Matcher(filter_pb2.Matcher(id=TEST_MATCHER_ID))
filterReturns = matcher.getParentFilter()
stubMock.GetParentFilter.assert_called_with(
filter_pb2.MatcherGetParentFilterRequest(matcher=matcher.data), timeout=mock.ANY)
self.assertEqual(filterReturns.name(), TEST_FILTER_NAME)
def testGetParentFilter(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetParentFilter.return_value = filter_pb2.ActionGetParentFilterResponse(
filter=filter_pb2.Filter(name=TEST_FILTER_NAME))
getStubMock.return_value = stubMock
action = opencue.wrappers.filter.Action(filter_pb2.Action(id=TEST_ACTION_ID))
filterReturned = action.getParentFilter()
stubMock.GetParentFilter.assert_called_with(
filter_pb2.ActionGetParentFilterRequest(action=action.data), timeout=mock.ANY)
self.assertEqual(filterReturned.name(), TEST_FILTER_NAME)
def testSetOrder(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetOrder.return_value = filter_pb2.FilterSetOrderResponse()
getStubMock.return_value = stubMock
value = 2
filter = opencue.wrappers.filter.Filter(
filter_pb2.Filter(name=TEST_FILTER_NAME))
filter.setOrder(value)
stubMock.SetOrder.assert_called_with(
filter_pb2.FilterSetOrderRequest(filter=filter.data, order=value), timeout=mock.ANY)
def testFindFilter(self, getStubMock):
filterName = 'arbitrary-name'
stubMock = mock.Mock()
stubMock.FindFilter.return_value = filter_pb2.FilterFindFilterResponse(
filter=filter_pb2.Filter(name=filterName))
getStubMock.return_value = stubMock
filterReturned = opencue.api.findFilter(TEST_SHOW_NAME, filterName)
stubMock.FindFilter.assert_called_with(
filter_pb2.FilterFindFilterRequest(show=TEST_SHOW_NAME, name=filterName),
timeout=mock.ANY)
self.assertEqual(filterName, filterReturned.name())
def testFindFilter(self, getStubMock):
stubMock = mock.Mock()
stubMock.FindFilter.return_value = show_pb2.ShowFindFilterResponse(
filter=filter_pb2.Filter(name=TEST_FILTER_NAME))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
filter_found = show.findFilter(TEST_FILTER_NAME)
stubMock.FindFilter.assert_called_with(show_pb2.ShowFindFilterRequest(
show=show.data, name=TEST_FILTER_NAME),
timeout=mock.ANY)
self.assertEqual(filter_found.name(), TEST_FILTER_NAME)
def testGetFilters(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetFilters.return_value = show_pb2.ShowGetFiltersResponse(
filters=filter_pb2.FilterSeq(filters=[filter_pb2.Filter(name=TEST_FILTER_NAME)]))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
filters = show.getFilters()
stubMock.GetFilters.assert_called_with(
show_pb2.ShowGetFiltersRequest(show=show.data), timeout=mock.ANY)
self.assertEqual(len(filters), 1)
self.assertEqual(filters[0].name(), TEST_FILTER_NAME)
def testRunFilterOnJobs(self, getStubMock):
stubMock = mock.Mock()
stubMock.RunFilterOnJobs.return_value = filter_pb2.FilterRunFilterOnJobsResponse()
getStubMock.return_value = stubMock
jobs = [opencue.wrappers.job.Job(job_pb2.Job(name='testJob'))]
jobSeq = job_pb2.JobSeq(jobs=[job.data for job in jobs])
filterToRun = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToRun.runFilterOnJobs(jobs)
stubMock.RunFilterOnJobs.assert_called_with(
filter_pb2.FilterRunFilterOnJobsRequest(filter=filterToRun.data, jobs=jobSeq),
timeout=mock.ANY)
def testGetActions(self, getStubMock):
actionId = 'aaa-aaaa-aaa'
stubMock = mock.Mock()
stubMock.GetActions.return_value = filter_pb2.FilterGetActionsResponse(
actions=filter_pb2.ActionSeq(actions=[filter_pb2.Action(id=actionId)]))
getStubMock.return_value = stubMock
filterForActions = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
actions = filterForActions.getActions()
stubMock.GetActions.assert_called_with(
filter_pb2.FilterGetActionsRequest(filter=filterForActions.data), timeout=mock.ANY)
self.assertEqual(len(actions), 1)
self.assertEqual(actions[0].id(), actionId)
def testGetMatchers(self, getStubMock):
matcherId = 'mmm-mmmm-mmm'
stubMock = mock.Mock()
stubMock.GetMatchers.return_value = filter_pb2.FilterGetMatchersResponse(
matchers=filter_pb2.MatcherSeq(matchers=[filter_pb2.Matcher(id=matcherId)]))
getStubMock.return_value = stubMock
filterForMatchers = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
matchers = filterForMatchers.getMatchers()
stubMock.GetMatchers.assert_called_with(
filter_pb2.FilterGetMatchersRequest(filter=filterForMatchers.data), timeout=mock.ANY)
self.assertEqual(len(matchers), 1)
self.assertEqual(matchers[0].id(), matcherId)
def testCreateMatcher(self, getStubMock):
matcherId = 'mmm-mmmm-mmm'
stubMock = mock.Mock()
stubMock.CreateMatcher.return_value = filter_pb2.FilterCreateMatcherResponse(
matcher=filter_pb2.Matcher(id=matcherId))
getStubMock.return_value = stubMock
queryStr = 'john'
subject = filter_pb2.USER
matcherType = filter_pb2.IS_NOT
matcherData = opencue.wrappers.filter.MatcherData(
subject=subject, type=matcherType, input=queryStr)
filterForMatcher = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
matcher = filterForMatcher.createMatcher(subject, matcherType, queryStr)
stubMock.CreateMatcher.assert_called_with(
filter_pb2.FilterCreateMatcherRequest(filter=filterForMatcher.data, data=matcherData),
timeout=mock.ANY)
self.assertEqual(matcher.id(), matcherId)
