Project import generated by Copybara.
GitOrigin-RevId: d9e9e3fb4e31372ec1fb43b178994ca78fa8fe70
diff --git a/search/test/query2ast_test.py b/search/test/query2ast_test.py
new file mode 100644
index 0000000..fc92e72
--- /dev/null
+++ b/search/test/query2ast_test.py
@@ -0,0 +1,1041 @@
+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style
+# license that can be found in the LICENSE file or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Tests for the query2ast module."""
+from __future__ import print_function
+from __future__ import division
+from __future__ import absolute_import
+
+import datetime
+import time
+import unittest
+import mock
+
+from proto import ast_pb2
+from proto import tracker_pb2
+from search import query2ast
+from tracker import tracker_bizobj
+
+BOOL = query2ast.BOOL
+DATE = query2ast.DATE
+NUM = query2ast.NUM
+TXT = query2ast.TXT
+
+SUBQUERY = query2ast.SUBQUERY
+LEFT_PAREN = query2ast.LEFT_PAREN
+RIGHT_PAREN = query2ast.RIGHT_PAREN
+OR = query2ast.OR
+
+BUILTIN_ISSUE_FIELDS = query2ast.BUILTIN_ISSUE_FIELDS
+ANY_FIELD = query2ast.BUILTIN_ISSUE_FIELDS['any_field']
+
+EQ = query2ast.EQ
+NE = query2ast.NE
+LT = query2ast.LT
+GT = query2ast.GT
+LE = query2ast.LE
+GE = query2ast.GE
+TEXT_HAS = query2ast.TEXT_HAS
+NOT_TEXT_HAS = query2ast.NOT_TEXT_HAS
+IS_DEFINED = query2ast.IS_DEFINED
+IS_NOT_DEFINED = query2ast.IS_NOT_DEFINED
+KEY_HAS = query2ast.KEY_HAS
+
+MakeCond = ast_pb2.MakeCond
+NOW = 1277762224
+
+
+class QueryParsingUnitTest(unittest.TestCase):
+
+ def setUp(self):
+ self.project_id = 789
+ self.default_config = tracker_bizobj.MakeDefaultProjectIssueConfig(
+ self.project_id)
+
+ def testParseUserQuery_OrClause(self):
+ # an "OR" query, which should look like two separate simple querys
+ # joined together by a pipe.
+ ast = query2ast.ParseUserQuery(
+ 'ham OR fancy', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ conj1 = ast.conjunctions[0]
+ conj2 = ast.conjunctions[1]
+ self.assertEqual([MakeCond(TEXT_HAS, [ANY_FIELD], ['ham'], [])],
+ conj1.conds)
+ self.assertEqual([MakeCond(TEXT_HAS, [ANY_FIELD], ['fancy'], [])],
+ conj2.conds)
+
+ def testParseUserQuery_Words(self):
+ # an "ORTerm" is actually anything appearing on either side of an
+ # "OR" operator. So this could be thought of as "simple" query parsing.
+
+ # a simple query with no spaces
+ ast = query2ast.ParseUserQuery(
+ 'hamfancy', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ fulltext_cond = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['hamfancy'], []), fulltext_cond)
+
+ # negative word
+ ast = query2ast.ParseUserQuery(
+ '-hamfancy', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ fulltext_cond = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ # note: not NOT_TEXT_HAS.
+ MakeCond(NOT_TEXT_HAS, [ANY_FIELD], ['hamfancy'], []),
+ fulltext_cond)
+
+ # invalid fulltext term
+ ast = query2ast.ParseUserQuery(
+ 'ham=fancy\\', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ self.assertEqual([], ast.conjunctions[0].conds)
+
+ # an explicit "AND" query in the "featured" context
+ warnings = []
+ query2ast.ParseUserQuery(
+ 'ham AND fancy', 'label:featured', BUILTIN_ISSUE_FIELDS,
+ self.default_config, warnings=warnings)
+ self.assertEqual(
+ ['The only supported boolean operator is OR (all capitals).'],
+ warnings)
+
+ # an implicit "AND" query
+ ast = query2ast.ParseUserQuery(
+ 'ham fancy', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, ft_cond1, ft_cond2 = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['ham'], []), ft_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['fancy'], []), ft_cond2)
+
+ # Use word with non-operator prefix.
+ word_with_non_op_prefix = '%stest' % query2ast.NON_OP_PREFIXES[0]
+ ast = query2ast.ParseUserQuery(
+ word_with_non_op_prefix, '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ fulltext_cond = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['"%s"' % word_with_non_op_prefix], []),
+ fulltext_cond)
+
+ # mix positive and negative words
+ ast = query2ast.ParseUserQuery(
+ 'ham -fancy', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, ft_cond1, ft_cond2 = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['ham'], []), ft_cond1)
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [ANY_FIELD], ['fancy'], []), ft_cond2)
+
+ # converts terms to lower case
+ ast = query2ast.ParseUserQuery(
+ 'AmDude', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, fulltext_cond = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['amdude'], []), fulltext_cond)
+
+ def testParseUserQuery_Phrases(self):
+ # positive phrases
+ ast = query2ast.ParseUserQuery(
+ '"one two"', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, fulltext_cond = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['"one two"'], []), fulltext_cond)
+
+ # negative phrases
+ ast = query2ast.ParseUserQuery(
+ '-"one two"', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, fulltext_cond = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [ANY_FIELD], ['"one two"'], []), fulltext_cond)
+
+ # multiple phrases
+ ast = query2ast.ParseUserQuery(
+ '-"a b" "x y"', '-label:deprecated', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ scope_cond1, ft_cond1, ft_cond2 = ast.conjunctions[0].conds
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['deprecated'], []),
+ scope_cond1)
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [ANY_FIELD], ['"a b"'], []), ft_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['"x y"'], []), ft_cond2)
+
+ def testParseUserQuery_CodeSyntaxThatWeNeedToCopeWith(self):
+ # positive phrases
+ ast = query2ast.ParseUserQuery(
+ 'Base::Tuple', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD],
+ ['"base::tuple"'], []),
+ cond)
+
+ # stuff we just ignore
+ ast = query2ast.ParseUserQuery(
+ ':: - -- .', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ self.assertEqual([], ast.conjunctions[0].conds)
+
+ def testParseUserQuery_IsOperator(self):
+ """Test is:open, is:spam, and is:blocked."""
+ for keyword in ['open', 'spam', 'blocked']:
+ ast = query2ast.ParseUserQuery(
+ 'is:' + keyword, '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS[keyword]], [], []),
+ cond1)
+ ast = query2ast.ParseUserQuery(
+ '-is:' + keyword, '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NE, [BUILTIN_ISSUE_FIELDS[keyword]], [], []),
+ cond1)
+
+ def testParseUserQuery_HasOperator(self):
+ # Search for issues with at least one attachment
+ ast = query2ast.ParseUserQuery(
+ 'has:attachment', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_DEFINED, [BUILTIN_ISSUE_FIELDS['attachment']], [], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-has:attachment', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_NOT_DEFINED, [BUILTIN_ISSUE_FIELDS['attachment']], [], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'has=attachment', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_DEFINED, [BUILTIN_ISSUE_FIELDS['attachment']], [], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-has=attachment', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_NOT_DEFINED, [BUILTIN_ISSUE_FIELDS['attachment']], [], []),
+ cond1)
+
+ # Search for numeric fields for searches with 'has' prefix
+ ast = query2ast.ParseUserQuery(
+ 'has:attachments', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_DEFINED, [BUILTIN_ISSUE_FIELDS['attachments']], [], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-has:attachments', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_NOT_DEFINED, [BUILTIN_ISSUE_FIELDS['attachments']],
+ [], []),
+ cond1)
+
+ # If it is not a field, look for any key-value label.
+ ast = query2ast.ParseUserQuery(
+ 'has:Size', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(IS_DEFINED, [BUILTIN_ISSUE_FIELDS['label']], ['size'], []),
+ cond1)
+
+ def testParseUserQuery_Phase(self):
+ ast = query2ast.ParseUserQuery(
+ 'gate:Canary,Stable', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['gate']],
+ ['canary', 'stable'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-gate:Canary,Stable', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['gate']],
+ ['canary', 'stable'], []),
+ cond1)
+
+ def testParseUserQuery_Components(self):
+ """Parse user queries for components"""
+ ast = query2ast.ParseUserQuery(
+ 'component:UI', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['component']],
+ ['ui'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'Component:UI>AboutBox', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['component']],
+ ['ui>aboutbox'], []),
+ cond1)
+
+ def testParseUserQuery_OwnersReportersAndCc(self):
+ """Parse user queries for owner:, reporter: and cc:."""
+ ast = query2ast.ParseUserQuery(
+ 'owner:user', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['owner']],
+ ['user'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'owner:user@example.com', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['owner']],
+ ['user@example.com'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'owner=user@example.com', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['owner']],
+ ['user@example.com'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-reporter=user@example.com', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NE, [BUILTIN_ISSUE_FIELDS['reporter']],
+ ['user@example.com'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'cc=user@example.com,user2@example.com', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['cc']],
+ ['user@example.com', 'user2@example.com'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'cc:user,user2', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['cc']],
+ ['user', 'user2'], []),
+ cond1)
+
+ def testParseUserQuery_SearchWithinFields(self):
+ # Search for issues with certain filenames
+ ast = query2ast.ParseUserQuery(
+ 'attachment:filename', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['attachment']],
+ ['filename'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-attachment:filename', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NOT_TEXT_HAS, [BUILTIN_ISSUE_FIELDS['attachment']],
+ ['filename'], []),
+ cond1)
+
+ # Search for issues with a certain number of attachments
+ ast = query2ast.ParseUserQuery(
+ 'attachments:2', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['attachments']],
+ ['2'], [2]),
+ cond1)
+
+ # Searches with '=' syntax
+ ast = query2ast.ParseUserQuery(
+ 'attachment=filename', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['attachment']],
+ ['filename'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-attachment=filename', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NE, [BUILTIN_ISSUE_FIELDS['attachment']],
+ ['filename'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'milestone=2009', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['label']], ['milestone-2009'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-milestone=2009', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NE, [BUILTIN_ISSUE_FIELDS['label']], ['milestone-2009'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'milestone=2009-Q1', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2009-q1'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ '-milestone=2009-Q1', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(NE, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2009-q1'], []),
+ cond1)
+
+ # Searches with ':' syntax
+ ast = query2ast.ParseUserQuery(
+ 'summary:foo', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS,
+ [BUILTIN_ISSUE_FIELDS['summary']], ['foo'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'summary:"greetings programs"', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS,
+ [BUILTIN_ISSUE_FIELDS['summary']], ['greetings programs'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'summary:"Ӓ"', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS,
+ [BUILTIN_ISSUE_FIELDS['summary']], ['Ӓ'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'priority:high', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(KEY_HAS,
+ [BUILTIN_ISSUE_FIELDS['label']], ['priority-high'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'type:security', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(KEY_HAS,
+ [BUILTIN_ISSUE_FIELDS['label']], ['type-security'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'label:priority-high', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS,
+ [BUILTIN_ISSUE_FIELDS['label']], ['priority-high'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'blockedon:other:123', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['blockedon']],
+ ['other:123'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'cost=-2', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['label']],
+ ['cost--2'], []),
+ cond1)
+
+ # Searches with ':' and an email domain only.
+ ast = query2ast.ParseUserQuery(
+ 'reporter:@google.com', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS,
+ [BUILTIN_ISSUE_FIELDS['reporter']], ['@google.com'], []),
+ cond1)
+
+ # Search for issues in certain user hotlists.
+ ast = query2ast.ParseUserQuery(
+ 'hotlist=gatsby@chromium.org:Hotlist1', '',
+ BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(
+ EQ, [BUILTIN_ISSUE_FIELDS['hotlist']],
+ ['gatsby@chromium.org:hotlist1'], []),
+ cond1)
+
+ # Search for 'Hotlist' labels.
+ ast = query2ast.ParseUserQuery(
+ 'hotlist:sublabel', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(KEY_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['hotlist-sublabel'], []),
+ cond1)
+
+ def testParseUserQuery_SearchWithinCustomFields(self):
+ """Enums are treated as labels, other fields are kept as fields."""
+ fd1 = tracker_bizobj.MakeFieldDef(
+ 1, self.project_id, 'Size', tracker_pb2.FieldTypes.ENUM_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False)
+ fd2 = tracker_bizobj.MakeFieldDef(
+ 1, self.project_id, 'EstDays', tracker_pb2.FieldTypes.INT_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False)
+ self.default_config.field_defs.extend([fd1, fd2])
+ ast = query2ast.ParseUserQuery(
+ 'Size:Small EstDays>3', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ cond2 = ast.conjunctions[0].conds[1]
+ self.assertEqual(
+ MakeCond(KEY_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['size-small'], []),
+ cond1)
+ self.assertEqual(
+ MakeCond(GT, [fd2], ['3'], [3]),
+ cond2)
+
+ @mock.patch('time.time', return_value=NOW)
+ def testParseUserQuery_Approvals(self, _mock_time):
+ """Test approval queries are parsed correctly."""
+ fd1 = tracker_bizobj.MakeFieldDef(
+ 1, self.project_id, 'UIReview', tracker_pb2.FieldTypes.APPROVAL_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False)
+ fd2 = tracker_bizobj.MakeFieldDef(
+ 2, self.project_id, 'EstDays', tracker_pb2.FieldTypes.INT_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False)
+ fd3 = tracker_bizobj.MakeFieldDef(
+ 3, self.project_id, 'UXReview', tracker_pb2.FieldTypes.APPROVAL_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False)
+ self.default_config.field_defs.extend([fd1, fd2, fd3])
+ ast = query2ast.ParseUserQuery(
+ 'UXReview-approver:user1@mail.com,user2@mail.com UIReview:Approved '
+ 'UIReview-on>today-7', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ cond2 = ast.conjunctions[0].conds[1]
+ cond3 = ast.conjunctions[0].conds[2]
+ self.assertEqual(MakeCond(TEXT_HAS, [fd3],
+ ['user1@mail.com', 'user2@mail.com'], [],
+ key_suffix='-approver'), cond1)
+ self.assertEqual(MakeCond(TEXT_HAS, [fd1], ['approved'], []), cond2)
+ self.assertEqual(
+ cond3,
+ MakeCond(
+ GT, [fd1], [], [query2ast._CalculatePastDate(7, NOW)],
+ key_suffix='-on'))
+
+ def testParseUserQuery_PhaseFields(self):
+ fd = tracker_bizobj.MakeFieldDef(
+ 1, self.project_id, 'EstDays', tracker_pb2.FieldTypes.INT_TYPE,
+ 'applic', 'applic', False, False, False, None, None, None, False, None,
+ None, None, 'no_action', 'doc', False, is_phase_field=True)
+ self.default_config.field_defs.append(fd)
+ ast = query2ast.ParseUserQuery(
+ 'UXReview.EstDays>3', '', BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(GT, [fd], ['3'], [3], phase_name='uxreview'),
+ cond1)
+
+ def testParseUserQuery_QuickOr(self):
+ # quick-or searches
+ ast = query2ast.ParseUserQuery(
+ 'milestone:2008,2009,2010', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(KEY_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2008', 'milestone-2009', 'milestone-2010'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'label:milestone-2008,milestone-2009,milestone-2010', '',
+ BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2008', 'milestone-2009', 'milestone-2010'], []),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'milestone=2008,2009,2010', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2008', 'milestone-2009', 'milestone-2010'], []),
+ cond1)
+
+ # Duplicated and trailing commas are ignored.
+ ast = query2ast.ParseUserQuery(
+ 'milestone=2008,,2009,2010,', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['label']],
+ ['milestone-2008', 'milestone-2009', 'milestone-2010'], []),
+ cond1)
+
+ def testParseUserQuery_Dates(self):
+ # query with a daterange
+ ast = query2ast.ParseUserQuery(
+ 'modified>=2009-5-12', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ ts1 = int(time.mktime(datetime.datetime(2009, 5, 12).timetuple()))
+ self.assertEqual(
+ MakeCond(GE, [BUILTIN_ISSUE_FIELDS['modified']], [], [ts1]), cond1)
+
+ # query with quick-or
+ ast = query2ast.ParseUserQuery(
+ 'modified=2009-5-12,2009-5-13', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ cond1 = ast.conjunctions[0].conds[0]
+ ts1 = int(time.mktime(datetime.datetime(2009, 5, 12).timetuple()))
+ ts2 = int(time.mktime(datetime.datetime(2009, 5, 13).timetuple()))
+ self.assertEqual(
+ MakeCond(EQ, [BUILTIN_ISSUE_FIELDS['modified']], [], [ts1, ts2]), cond1)
+
+ # query with multiple dateranges
+ ast = query2ast.ParseUserQuery(
+ 'modified>=2009-5-12 opened<2008/1/1', '',
+ BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1, cond2 = ast.conjunctions[0].conds
+ ts1 = int(time.mktime(datetime.datetime(2009, 5, 12).timetuple()))
+ self.assertEqual(
+ MakeCond(GE, [BUILTIN_ISSUE_FIELDS['modified']], [], [ts1]), cond1)
+ ts2 = int(time.mktime(datetime.datetime(2008, 1, 1).timetuple()))
+ self.assertEqual(
+ MakeCond(LT, [BUILTIN_ISSUE_FIELDS['opened']], [], [ts2]), cond2)
+
+ # query with multiple dateranges plus a search term
+ ast = query2ast.ParseUserQuery(
+ 'one two modified>=2009-5-12 opened<2008/1/1', '',
+ BUILTIN_ISSUE_FIELDS, self.default_config)
+ ft_cond1, ft_cond2, cond1, cond2 = ast.conjunctions[0].conds
+ ts1 = int(time.mktime(datetime.datetime(2009, 5, 12).timetuple()))
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['one'], []), ft_cond1)
+ self.assertEqual(
+ MakeCond(TEXT_HAS, [ANY_FIELD], ['two'], []), ft_cond2)
+ self.assertEqual(
+ MakeCond(GE, [BUILTIN_ISSUE_FIELDS['modified']], [], [ts1]), cond1)
+ ts2 = int(time.mktime(datetime.datetime(2008, 1, 1).timetuple()))
+ self.assertEqual(
+ MakeCond(LT, [BUILTIN_ISSUE_FIELDS['opened']], [], [ts2]), cond2)
+
+ # query with a date field compared to "today"
+ ast = query2ast.ParseUserQuery(
+ 'modified<today', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config, now=NOW)
+ cond1 = ast.conjunctions[0].conds[0]
+ ts1 = query2ast._CalculatePastDate(0, now=NOW)
+ self.assertEqual(MakeCond(LT, [BUILTIN_ISSUE_FIELDS['modified']],
+ [], [ts1]),
+ cond1)
+
+ # query with a daterange using today-N alias
+ ast = query2ast.ParseUserQuery(
+ 'modified>=today-13', '', BUILTIN_ISSUE_FIELDS,
+ self.default_config, now=NOW)
+ cond1 = ast.conjunctions[0].conds[0]
+ ts1 = query2ast._CalculatePastDate(13, now=NOW)
+ self.assertEqual(MakeCond(GE, [BUILTIN_ISSUE_FIELDS['modified']],
+ [], [ts1]),
+ cond1)
+
+ ast = query2ast.ParseUserQuery(
+ 'modified>today-13', '', BUILTIN_ISSUE_FIELDS, self.default_config,
+ now=NOW)
+ cond1 = ast.conjunctions[0].conds[0]
+ ts1 = query2ast._CalculatePastDate(13, now=NOW)
+ self.assertEqual(MakeCond(GT, [BUILTIN_ISSUE_FIELDS['modified']],
+ [], [ts1]),
+ cond1)
+
+ # query with multiple old date query terms.
+ ast = query2ast.ParseUserQuery(
+ 'modified-after:2009-5-12 opened-before:2008/1/1 '
+ 'closed-after:2007-2-1', '',
+ BUILTIN_ISSUE_FIELDS, self.default_config)
+ cond1, cond2, cond3 = ast.conjunctions[0].conds
+ ts1 = int(time.mktime(datetime.datetime(2009, 5, 12).timetuple()))
+ self.assertEqual(
+ MakeCond(GT, [BUILTIN_ISSUE_FIELDS['modified']], [], [ts1]), cond1)
+ ts2 = int(time.mktime(datetime.datetime(2008, 1, 1).timetuple()))
+ self.assertEqual(
+ MakeCond(LT, [BUILTIN_ISSUE_FIELDS['opened']], [], [ts2]), cond2)
+ ts3 = int(time.mktime(datetime.datetime(2007, 2, 1).timetuple()))
+ self.assertEqual(
+ MakeCond(GT, [BUILTIN_ISSUE_FIELDS['closed']], [], [ts3]), cond3)
+
+ def testCalculatePastDate(self):
+ ts1 = query2ast._CalculatePastDate(0, now=NOW)
+ self.assertEqual(NOW, ts1)
+
+ ts2 = query2ast._CalculatePastDate(13, now=NOW)
+ self.assertEqual(ts2, NOW - 13 * 24 * 60 * 60)
+
+ # Try it once with time.time() instead of a known timestamp.
+ ts_system_clock = query2ast._CalculatePastDate(13)
+ self.assertTrue(ts_system_clock < int(time.time()))
+
+ def testParseUserQuery_BadDates(self):
+ bad_dates = ['today-13h', 'yesterday', '2/2', 'm/y/d',
+ '99/99/1999', '0-0-0']
+ for val in bad_dates:
+ with self.assertRaises(query2ast.InvalidQueryError) as cm:
+ query2ast.ParseUserQuery(
+ 'modified>=' + val, '', BUILTIN_ISSUE_FIELDS,
+ self.default_config)
+ self.assertEqual('Could not parse date: ' + val, cm.exception.message)
+
+ def testQueryToSubqueries_BasicQuery(self):
+ self.assertEqual(['owner:me'], query2ast.QueryToSubqueries('owner:me'))
+
+ def testQueryToSubqueries_EmptyQuery(self):
+ self.assertEqual([''], query2ast.QueryToSubqueries(''))
+
+ def testQueryToSubqueries_UnmatchedParenthesesThrowsError(self):
+ with self.assertRaises(query2ast.InvalidQueryError):
+ self.assertEqual(['Pri=1'], query2ast.QueryToSubqueries('Pri=1))'))
+ with self.assertRaises(query2ast.InvalidQueryError):
+ self.assertEqual(
+ ['label:Hello'], query2ast.QueryToSubqueries('((label:Hello'))
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ self.assertEqual(
+ ['owner:me'], query2ast.QueryToSubqueries('((((owner:me)))'))
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ self.assertEqual(
+ ['test=What'], query2ast.QueryToSubqueries('(((test=What))))'))
+
+ def testQueryToSubqueries_IgnoresEmptyGroups(self):
+ self.assertEqual([''], query2ast.QueryToSubqueries('()(()(()))()()'))
+
+ self.assertEqual(
+ ['owner:me'], query2ast.QueryToSubqueries('()(()owner:me)()()'))
+
+ def testQueryToSubqueries_BasicOr(self):
+ self.assertEqual(
+ ['owner:me', 'status:New', 'Pri=1'],
+ query2ast.QueryToSubqueries('owner:me OR status:New OR Pri=1'))
+
+ def testQueryToSubqueries_OrAtStartOrEnd(self):
+ self.assertEqual(
+ ['owner:me OR'], query2ast.QueryToSubqueries('owner:me OR'))
+
+ self.assertEqual(
+ ['OR owner:me'], query2ast.QueryToSubqueries('OR owner:me'))
+
+ def testQueryToSubqueries_BasicParentheses(self):
+ self.assertEqual(
+ ['owner:me status:New'],
+ query2ast.QueryToSubqueries('owner:me (status:New)'))
+
+ self.assertEqual(
+ ['owner:me status:New'],
+ query2ast.QueryToSubqueries('(owner:me) status:New'))
+
+ self.assertEqual(
+ ['owner:me status:New'],
+ query2ast.QueryToSubqueries('((owner:me) (status:New))'))
+
+ def testQueryToSubqueries_ParenthesesWithOr(self):
+ self.assertEqual(
+ ['Pri=1 owner:me', 'Pri=1 status:New'],
+ query2ast.QueryToSubqueries('Pri=1 (owner:me OR status:New)'))
+
+ self.assertEqual(
+ ['owner:me component:OhNo', 'status:New component:OhNo'],
+ query2ast.QueryToSubqueries('(owner:me OR status:New) component:OhNo'))
+
+ def testQueryToSubqueries_ParenthesesWithOr_Multiple(self):
+ self.assertEqual(
+ [
+ 'Pri=1 test owner:me', 'Pri=1 test status:New',
+ 'Pri=2 test owner:me', 'Pri=2 test status:New'
+ ],
+ query2ast.QueryToSubqueries(
+ '(Pri=1 OR Pri=2)(test (owner:me OR status:New))'))
+
+ def testQueryToSubqueries_OrNextToParentheses(self):
+ self.assertEqual(['A', 'B'], query2ast.QueryToSubqueries('(A) OR (B)'))
+
+ self.assertEqual(
+ ['A B', 'A C E', 'A D E'],
+ query2ast.QueryToSubqueries('A (B OR (C OR D) E)'))
+
+ self.assertEqual(
+ ['A B C', 'A B D', 'A E'],
+ query2ast.QueryToSubqueries('A (B (C OR D) OR E)'))
+
+ def testQueryToSubqueries_ExtraSpaces(self):
+ self.assertEqual(
+ ['A', 'B'], query2ast.QueryToSubqueries(' ( A ) OR ( B ) '))
+
+ self.assertEqual(
+ ['A B', 'A C E', 'A D E'],
+ query2ast.QueryToSubqueries(' A ( B OR ( C OR D ) E )'))
+
+ def testQueryToSubqueries_OrAtEndOfParentheses(self):
+ self.assertEqual(['A B'], query2ast.QueryToSubqueries('(A OR )(B)'))
+ self.assertEqual(
+ ['A B', 'A C'], query2ast.QueryToSubqueries('( OR A)(B OR C)'))
+ self.assertEqual(
+ ['A B', 'A C'], query2ast.QueryToSubqueries(' OR A (B OR C)'))
+
+ def testQueryToSubqueries_EmptyOrGroup(self):
+ self.assertEqual(
+ ['A C', 'C', 'B C'], query2ast.QueryToSubqueries('(A OR OR B)(C)'))
+
+ def testParseQuery_Basic(self):
+ self.assertEqual(
+ [
+ 'owner:me',
+ ],
+ query2ast._ParseQuery(
+ query2ast.PeekIterator(
+ [ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me')])))
+
+ def testParseQuery_Complex(self):
+ self.assertEqual(
+ [
+ 'owner:me',
+ 'Pri=1',
+ 'label=test',
+ ],
+ query2ast._ParseQuery(
+ query2ast.PeekIterator(
+ [
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=LEFT_PAREN),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='label=test'),
+ ])))
+
+ def testParseOrGroup_Basic(self):
+ self.assertEqual(
+ [
+ 'owner:me',
+ ],
+ query2ast._ParseOrGroup(
+ query2ast.PeekIterator(
+ [ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me')])))
+
+ def testParseOrGroup_TwoAdjacentAndGroups(self):
+ self.assertEqual(
+ [
+ 'owner:me Pri=1',
+ 'owner:me label=test',
+ ],
+ query2ast._ParseOrGroup(
+ query2ast.PeekIterator(
+ [
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me'),
+ ast_pb2.QueryToken(token_type=LEFT_PAREN),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='label=test'),
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ])))
+
+ def testParseAndGroup_Subquery(self):
+ self.assertEqual(
+ [
+ 'owner:me',
+ ],
+ query2ast._ParseAndGroup(
+ query2ast.PeekIterator(
+ [ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me')])))
+
+ def testParseAndGroup_ParenthesesGroup(self):
+ self.assertEqual(
+ [
+ 'owner:me',
+ 'Pri=1',
+ ],
+ query2ast._ParseAndGroup(
+ query2ast.PeekIterator(
+ [
+ ast_pb2.QueryToken(token_type=LEFT_PAREN),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ])))
+
+ def testParseAndGroup_Empty(self):
+ self.assertEqual([], query2ast._ParseAndGroup(query2ast.PeekIterator([])))
+
+ def testParseAndGroup_InvalidTokens(self):
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ParseAndGroup(
+ query2ast.PeekIterator(
+ [
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ]))
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ParseAndGroup(
+ query2ast.PeekIterator(
+ [
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ]))
+
+ def testValidateAndTokenizeQuery_Basic(self):
+ self.assertEqual(
+ [
+ ast_pb2.QueryToken(token_type=LEFT_PAREN),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='Pri=1'),
+ ast_pb2.QueryToken(token_type=RIGHT_PAREN),
+ ], query2ast._ValidateAndTokenizeQuery('(owner:me OR Pri=1)'))
+
+ def testValidateAndTokenizeQuery_UnmatchedParentheses(self):
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ValidateAndTokenizeQuery('(owner:me')
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ValidateAndTokenizeQuery('owner:me)')
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ValidateAndTokenizeQuery('(()owner:me))')
+
+ with self.assertRaises(query2ast.InvalidQueryError):
+ query2ast._ValidateAndTokenizeQuery('(()owner:me)())')
+
+ def testTokenizeSubqueryOnOr_NoOrOperator(self):
+ self.assertEqual(
+ [ast_pb2.QueryToken(token_type=SUBQUERY, value='owner:me')],
+ query2ast._TokenizeSubqueryOnOr('owner:me'))
+
+ def testTokenizeSubqueryOnOr_BasicOrOperator(self):
+ self.assertEqual(
+ [
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='A'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='B'),
+ ast_pb2.QueryToken(token_type=OR),
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='C'),
+ ], query2ast._TokenizeSubqueryOnOr('A OR B OR C'))
+
+ def testTokenizeSubqueryOnOr_EmptyOrOperator(self):
+ self.assertEqual(
+ [ast_pb2.QueryToken(token_type=OR)],
+ query2ast._TokenizeSubqueryOnOr(' OR '))
+
+ self.assertEqual(
+ [
+ ast_pb2.QueryToken(token_type=SUBQUERY, value='A'),
+ ast_pb2.QueryToken(token_type=OR),
+ ], query2ast._TokenizeSubqueryOnOr('A OR '))
+
+ def testMultiplySubqueries_Basic(self):
+ self.assertEqual(
+ ['owner:me Pri=1', 'owner:me Pri=2', 'test Pri=1', 'test Pri=2'],
+ query2ast._MultiplySubqueries(['owner:me', 'test'], ['Pri=1', 'Pri=2']))
+
+ def testMultiplySubqueries_OneEmpty(self):
+ self.assertEqual(
+ ['Pri=1', 'Pri=2'],
+ query2ast._MultiplySubqueries([], ['Pri=1', 'Pri=2']))
+ self.assertEqual(
+ ['Pri=1', 'Pri=2'],
+ query2ast._MultiplySubqueries([''], ['Pri=1', 'Pri=2']))
+
+ self.assertEqual(
+ ['Pri=1', 'Pri=2'],
+ query2ast._MultiplySubqueries(['Pri=1', 'Pri=2'], []))
+ self.assertEqual(
+ ['Pri=1', 'Pri=2'],
+ query2ast._MultiplySubqueries(['Pri=1', 'Pri=2'], ['']))
+
+ def testPeekIterator_Basic(self):
+ iterator = query2ast.PeekIterator([1, 2, 3])
+
+ self.assertEqual(1, iterator.peek())
+ self.assertEqual(1, iterator.next())
+
+ self.assertEqual(2, iterator.next())
+
+ self.assertEqual(3, iterator.peek())
+ self.assertEqual(3, iterator.next())
+
+ with self.assertRaises(StopIteration):
+ iterator.peek()
+
+ with self.assertRaises(StopIteration):
+ iterator.next()