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)