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)