def testIsNew(self, getStubMock):
matcherFalse = opencue.wrappers.filter.Matcher(
filter_pb2.Matcher(id=TEST_MATCHER_ID, input='test'))
self.assertFalse(matcherFalse.isNew())
matcherTrue = opencue.wrappers.filter.Matcher()
self.assertTrue(matcherTrue.isNew())
def testCommit(self, getStubMock):
stubMock = mock.Mock()
stubMock.Commit.return_value = filter_pb2.MatcherCommitResponse()
getStubMock.return_value = stubMock
matcher = opencue.wrappers.filter.Matcher(
filter_pb2.Matcher(id=TEST_MATCHER_ID, input='test'))
matcher.commit()
stubMock.Commit.assert_called_with(
filter_pb2.MatcherCommitRequest(matcher=matcher.data), timeout=mock.ANY)
def testDelete(self, getStubMock):
stubMock = mock.Mock()
stubMock.Delete.return_value = filter_pb2.MatcherDeleteResponse()
getStubMock.return_value = stubMock
matcher = opencue.wrappers.filter.Matcher(
filter_pb2.Matcher(id=TEST_MATCHER_ID))
matcher.delete()
stubMock.Delete.assert_called_with(
filter_pb2.MatcherDeleteRequest(matcher=matcher.data), 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 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)
