############################################################################## # # Copyright (c) 2006-2009 Nexedi SA and Contributors. All Rights Reserved. # Jerome Perrin <jerome@nexedi.com> # Vincent Pelletier <vincent@nexedi.com> # # WARNING: This program as such is intended to be used by professional # programmers who take the whole responsability of assessing all potential # consequences resulting from its eventual inadequacies and bugs # End users who are looking for a ready-to-use solution with commercial # garantees and support are strongly adviced to contract a Free Software # Service Company # # This program is Free Software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # ############################################################################## import unittest import sys from DateTime import DateTime from Products.ZSQLMethods.SQL import SQL as ZSQLMethod from Products.CMFCore.Expression import Expression from Products.ZSQLCatalog.SQLCatalog import Catalog as SQLCatalog from Products.ZSQLCatalog.ZSQLCatalog import ZCatalog as ZSQLCatalog from Products.ZSQLCatalog.SQLCatalog import Query from Products.ZSQLCatalog.SQLCatalog import ComplexQuery from Products.ZSQLCatalog.SQLCatalog import SimpleQuery from Products.ZSQLCatalog.SQLCatalog import NegatedQuery from Products.ZSQLCatalog.Query.Query import Query as _Query from Products.ZSQLCatalog.Query.EntireQuery import EntireQuery from Products.ZSQLCatalog.Query.RelatedQuery import RelatedQuery from DateTime import DateTime class ReferenceQuery: operator = None column = None value = None def __init__(self, *args, **kw): self.operator = kw.pop('operator', None) assert len(args) == 0 or len(kw) == 0 self.args = [] for arg in args: if isinstance(arg, (tuple, list)): self.args.extend(arg) else: self.args.append(arg) if len(kw) == 1: self.column, self.value = kw.items()[0] elif len(kw) > 1: raise ValueError, 'kw must not have more than one item: %r' % (kw, ) def __eq__(self, other): if isinstance(other, SimpleQuery): return self.column is not None and \ other.getColumn() == self.column and \ other.getValue() == self.value and \ other.operator == self.operator elif isinstance(other, ComplexQuery): if not (len(other.query_list) == len(self.args) and \ other.logical_operator == self.operator): return False other_query_list = other.query_list[:] for subquery in self.args: for other_query_id in xrange(len(other_query_list)): other_query = other_query_list[other_query_id] if subquery == other_query: other_query_list.pop(other_query_id) break else: return False return len(other_query_list) == 0 elif isinstance(other, EntireQuery): return len(self.args) == 1 and \ self.args[0] == other.query elif isinstance(other, RelatedQuery): return self == other.join_condition elif isinstance(other, Query): return self == other.wrapped_query else: raise TypeError, 'Compared value is not a (known) Query instance: (%s) %r' % (other.__class__.__name__, other) def __repr__(self): return '<%s column=%r operator=%r value=%r args=%r>' % \ (self.__class__.__name__, self.column, self.operator, self.value, self.args) class RelatedReferenceQuery: def __init__(self, reference_subquery): self.subquery = reference_subquery def __eq__(self, other): return isinstance(other, RelatedQuery) and \ self.subquery == other.join_condition def __repr__(self): return '<%s %r>' % (self.__class__.__name__, self.subquery) class DummyCatalog(SQLCatalog): """ Mimic a table stucture. """ sql_catalog_keyword_search_keys = ('keyword', ) sql_catalog_datetime_search_keys = ('date', ) sql_catalog_full_text_search_keys = ('fulltext', ) sql_catalog_scriptable_keys = ('scriptable_keyword | scriptableKeyScript', ) def getColumnMap(self): return { 'uid': ['foo', 'bar'], 'default': ['foo', ], 'keyword': ['foo', ], 'date': ['foo', ], 'fulltext': ['foo', ], 'other_uid': ['bar', ] } def getSQLCatalogRelatedKeyList(self, key_list): return [ 'related_default | bar,foo/default/z_related_table', 'related_keyword | bar,foo/keyword/z_related_table', 'related_date | bar,foo/date/z_related_table' ] def z_related_table(self, *args, **kw): assert kw.get('src__', False) assert 'query_table' in kw assert 'table_0' in kw assert 'table_1' in kw assert len(kw) == 4 return '%(table_0)s.uid = %(query_table)s.uid AND %(table_0)s.other_uid = %(table_1)s' % kw def scriptableKeyScript(self, value): return SimpleQuery(operator='=', keyword=value) class TestSQLCatalog(unittest.TestCase): def setUp(self): self._catalog = DummyCatalog('dummy_catalog') def assertCatalogRaises(self, exception, kw): self.assertRaises(exception, self._catalog, src__=1, query_table='foo', **kw) def catalog(self, reference_tree, kw, check_search_text=True): reference_param_dict = self._catalog._queryResults(query_table='foo', **kw) query = self._catalog.buildQuery(kw) self.assertEqual(reference_tree, query) search_text = query.asSearchTextExpression(self._catalog) if check_search_text: # XXX: sould "keyword" be always used for search text searches ? search_text_param_dict = self._catalog._queryResults(query_table='foo', keyword=search_text) self.assertEqual(reference_param_dict, search_text_param_dict, 'Query: %r\nSearchText: %r\nReference: %r\nSecond rendering: %r' % \ (query, search_text, reference_param_dict, search_text_param_dict)) def test_001_DefaultKey(self): for column in ('default', 'related_default'): self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='a'), operator='and'), {column: 'a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', default='%a'), operator='and'), {column: '%a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<', default='a'), operator='and'), {column: '<a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<=', default='a'), operator='and'), {column: '<=a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>=', default='a'), operator='and'), {column: '>=a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>', default='a'), operator='and'), {column: '>a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='!=', default='a'), operator='and'), {column: '!=a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='a b'), operator='and'), {column: 'a b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='a >b'), operator='and'), {column: 'a >b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>', default='a >b'), operator='and'), {column: '>a >b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='>a >b'), operator='and'), {column: '">a >b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>', default='>a >b'), operator='and'), {column: '>">a >b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='a OR b'), operator='and'), {column: 'a OR b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<', default='path'), operator='and'), {column: {'query': 'path', 'range': 'max'}}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', default=['a', 'b']), operator='and'), {column: ['a', 'b']}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', default=['a', 'b']), operator='and'), {column: ['=a', '=b']}) def test_002_keyOverride(self): self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='%a'), operator='and'), {'default': {'query': '%a', 'key': 'ExactMatch'}}, check_search_text=False) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='<a'), operator='and'), {'default': {'query': '<a', 'key': 'ExactMatch'}}, check_search_text=False) def test_003_DateTimeKey(self): for column in ('date', 'related_date'): self.catalog(ReferenceQuery(ReferenceQuery(operator='>=', date=DateTime('2008/10/01 12:10:21')), operator='and'), {column: {'query': '>2008/10/01 12:10:20', 'format': '%y/%m/%d'}}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>=', date=DateTime('2008/10/01 12:10:21 CEST')), operator='and'), {column: {'query': '>2008/10/01 12:10:20 CEST', 'format': '%y/%m/%d'}}) self.catalog(ReferenceQuery(ReferenceQuery(operator='>=', date=DateTime('2008/10/01 12:10:21 CET')), operator='and'), {column: {'query': '>2008/10/01 12:10:20 CET', 'format': '%y/%m/%d'}}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/10/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/10/02 UTC')) , operator='and'), operator='and'), {column: '2008/10/01 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/01/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2009/01/01 UTC')) , operator='and'), operator='and'), {column: '2008 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/01/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/02/01 UTC')) , operator='and'), operator='and'), {column: '2008/01 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/10/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/10/02 UTC')) , operator='and'), operator='and'), {column: {'type': 'date', 'query': '10/01/2008 UTC', 'format': '%m/%d/%Y'}}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/10/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/10/02 UTC')) , operator='and'), operator='and'), {column: {'type': 'date', 'query': '01/10/2008 UTC', 'format': '%d/%m/%Y'}}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', date=[DateTime('2008/01/10 UTC'), DateTime('2008/01/09 UTC')]), operator='and'), {column: {'query': ['2008/01/10 UTC', '2008/01/09 UTC'], 'operator': 'in'}}, check_search_text=False) self.catalog(ReferenceQuery(ReferenceQuery(operator='>', date=DateTime('2008/01/10 UTC')), operator='and'), {column: {'query': '2008/01/10 UTC', 'range': 'nlt'}}, check_search_text=False) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/01/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2009/01/01 UTC')) , operator='and'), operator='and'), {column: '2008 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/02/01 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/03/01 UTC')) , operator='and'), operator='and'), {column: '2008/02 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/02/02 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/02/03 UTC')) , operator='and'), operator='and'), {column: '2008/02/02 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/02/02 10:00:00 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/02/02 11:00:00 UTC')) , operator='and'), operator='and'), {column: '2008/02/02 10 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/02/02 10:10:00 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/02/02 10:11:00 UTC')) , operator='and'), operator='and'), {column: '2008/02/02 10:10 UTC'}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='>=', date=DateTime('2008/02/02 10:10:10 UTC')), ReferenceQuery(operator='<', date=DateTime('2008/02/02 10:10:11 UTC')) , operator='and'), operator='and'), {column: '2008/02/02 10:10:10 UTC'}) def test_004_KeywordKey(self): for column in ('keyword', 'related_keyword'): self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a%'), operator='and'), {column: 'a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a b%'), operator='and'), {column: 'a b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a b%'), operator='and'), {column: '"a b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='!=', keyword='a'), operator='and'), {column: '!=a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='not like', keyword='%a'), operator='and'), {column: '!=%a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a'), operator='and'), {column: '%a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<', keyword='a'), operator='and'), {column: '<a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<', keyword='path'), operator='and'), {column: {'query': 'path', 'range': 'max'}}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='like', keyword='%a%'), ReferenceQuery(operator='like', keyword='%b%') , operator='or'), operator='and'), {column: ['a', 'b']}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', keyword=['a', 'b']), operator='and'), {column: ['=a', '=b']}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='like', keyword='%a%'), ReferenceQuery(operator='<', keyword='b') , operator='or'), operator='and'), {column: ['a', '<b']}) self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(operator='like', keyword='%a%'), ReferenceQuery(operator='like', keyword='%b') , operator='or'), operator='and'), {column: ['a', '%b']}) def test_005_SearchText(self): self.catalog(ReferenceQuery(ReferenceQuery(ReferenceQuery(operator='like', keyword='%=a%'), ReferenceQuery(operator='like', keyword='%=b%'), operator='or'), operator='and'), {'keyword': '"=a" OR "=b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', keyword=['a', 'b']), operator='and'), {'keyword': '="a" OR ="b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', keyword=['a', 'b']), operator='and'), {'keyword': '=a OR =b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='in', keyword=['a', 'b', 'c']), operator='and'), {'keyword': '=a OR =b OR =c'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a%'), operator='and'), {'keyword': 'keyword:a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='a'), operator='and'), {'keyword': 'default:a'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%a b%'), operator='and'), {'keyword': 'a b'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%=a OR =b%'), operator='and'), {'keyword': '"=a OR =b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', keyword='=a OR =b'), operator='and'), {'keyword': '="=a OR =b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='<', keyword='=a OR =b'), operator='and'), {'keyword': '<"=a OR =b"'}) self.catalog(ReferenceQuery(ReferenceQuery(operator='like', keyword='%"a" OR "b"%'), operator='and'), {'keyword': '"\\"a\\" OR \\"b\\""'}) def test_006_testRelatedKey_with_multiple_join(self): # The name of catalog parameter does not matter at all # ComplexQuery(ComplexQuery(AutoQuery(RelatedQuery(SimpleQuery())), AutoQuery(RelatedQuery(SimpleQuery())))) # 'AutoQuery' doesn't need any ReferenceQuery equivalent. self.catalog(ReferenceQuery(ReferenceQuery( ReferenceQuery(RelatedReferenceQuery(ReferenceQuery(operator='=', default='a')), operator='and'), ReferenceQuery(RelatedReferenceQuery(ReferenceQuery(operator='=', default='b')), operator='and') , operator='and'), operator='and'), {'query': ComplexQuery(Query(related_default='a'), Query(related_default='b'))}) def test_007_testScriptableKey(self): self.catalog(ReferenceQuery(ReferenceQuery(operator='=', keyword='%a%'), operator='and'), {'scriptable_keyword': '%a%'}) def test_008_testRawKey(self): self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='%a%'), operator='and'), {'default': {'query': '%a%', 'key': 'RawKey'}}, check_search_text=False) self.catalog(ReferenceQuery(ReferenceQuery(operator='=', default='>a'), operator='and'), {'default': {'query': '>a', 'key': 'RawKey'}}, check_search_text=False) def test_009_testFullTextKey(self): self.catalog(ReferenceQuery(ReferenceQuery(operator='match', fulltext='a'), operator='and'), {'fulltext': 'a'}) ##return catalog(title=Query(title='a', operator='not')) #return catalog(title={'query': 'a', 'operator': 'not'}) #return catalog(title={'query': ['a', 'b'], 'operator': 'not'}) #return context.portal_catalog(source_title="toto", source_description="tutu", src__=1) #print catalog(query=ComplexQuery(Query(title='1'), ComplexQuery(Query(portal_type='Foo') ,Query(portal_type='Bar'), operator='or'), operator='and')) #print catalog(title={'query': ('path', 2), 'operator': 'and'}, exception=TypeError) #print catalog(sort_on=[('source_title', )], check_search_text=False) #print catalog(query=ComplexQuery(Query(source_title='foo'), Query(source_title='bar')), sort_on=[('source_title', ), ('source_title_1', )], check_search_text=False) def test_suite(): suite = unittest.TestSuite() suite.addTest(unittest.makeSuite(TestSQLCatalog)) return suite