def test_set_limits(self):
# subsequence with xpath
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq.set_limits(low=0, high=4)
self.assertEqual('subsequence(/el, 1, 4)', xq.getQuery())
# subsequence with FLWR query
xq.return_only({'name': 'name'})
self.assert_('subsequence(for $n in' in xq.getQuery())
# additive limits
xq = Xquery(xpath='/el')
xq.set_limits(low=2, high=10)
xq.set_limits(low=1, high=5)
self.assertEqual('subsequence(/el, 4, 4)', xq.getQuery())
# no high specified
xq = Xquery(xpath='/el')
xq.set_limits(low=10)
self.assertEqual('subsequence(/el, 11, )', xq.getQuery())
# no low
xq = Xquery(xpath='/el')
xq.set_limits(high=15)
self.assertEqual('subsequence(/el, 1, 15)', xq.getQuery())
def test_filters(self):
xq = Xquery(xpath='/el')
xq.add_filter('.', 'contains', 'dog')
self.assertEquals('/el[contains(., "dog")]', xq.getQuery())
# filters are additive
xq.add_filter('.', 'startswith', 'S')
self.assertEquals('/el[contains(., "dog")][starts-with(., "S")]', xq.getQuery())
def test_filter_exists(self):
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'exists', True)
self.assertEquals('/el[@id]', xq.getQuery())
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'exists', False)
self.assertEquals('/el[not(@id)]', xq.getQuery())
def test_namespaces(self):
xq = Xquery(xpath='/foo:el', namespaces={'foo': 'urn:foo#'})
ns_declaration = '''declare namespace foo='urn:foo#';'''
# xpath-only xquery should have namespace declaration
self.assert_(ns_declaration in xq.getQuery())
# full FLOWR xquery should also have declaration
xq.return_only({'id': '@id'})
self.assert_(ns_declaration in xq.getQuery())
def test_not_filters(self):
xq = Xquery(xpath='/el')
xq.add_filter('.', 'contains', 'dog', mode='NOT')
self.assertEquals('/el[not(contains(., "dog"))]', xq.getQuery())
xq = Xquery(xpath='/el')
xq.add_filter('.', 'contains', 'dog', mode='NOT')
xq.add_filter('.', 'startswith', 'S', mode='NOT')
self.assertEquals('/el[not(contains(., "dog") or starts-with(., "S"))]',
xq.getQuery())
def test_filter_in(self):
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'in', ['a', 'b', 'c'])
self.assertEquals('/el[contains(("a","b","c"), @id)]', xq.getQuery())
# filter on a 'special' field - requires let & where statements
xq = Xquery(xpath='/el')
xq.add_filter('document_name', 'in', ['a.xml', 'b.xml'])
self.assert_('let $document_name' in xq.getQuery())
self.assert_('where contains(("a.xml","b.xml"), $document_name)' in xq.getQuery())
def test_filter_in(self):
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'in', ['a', 'b', 'c'])
self.assertEquals('/el[@id="a" or @id="b" or @id="c"]', xq.getQuery())
# filter on a 'special' field - requires let & where statements
xq = Xquery(xpath='/el')
xq.add_filter('document_name', 'in', ['a.xml', 'b.xml'])
self.assert_('let $document_name' in xq.getQuery())
self.assert_('where $document_name="a.xml" or $document_name="b.xml"'
in xq.getQuery())
def test_filter_gtlt(self):
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'gt', 5)
self.assert_('where $n/@id > 5' in xq.getQuery())
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'gte', 5)
self.assert_('where $n/@id >= 5' in xq.getQuery())
xq.add_filter('@id', 'lt', '5')
self.assert_('where $n/@id < 5' in xq.getQuery())
xq.add_filter('@id', 'lte', 3)
self.assert_('where $n/@id <= 3' in xq.getQuery())
def test_filter_gtlt(self):
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'gt', 5)
self.assert_('el[@id > 5]' in xq.getQuery())
xq = Xquery(xpath='/el')
xq.add_filter('@id', 'gte', 5)
self.assert_('/el[@id >= 5]' in xq.getQuery())
xq.add_filter('@id', 'lt', '10')
self.assert_('el[@id >= 5]' in xq.getQuery())
self.assert_('[@id < "10"]' in xq.getQuery())
xq.add_filter('@id', 'lte', 3)
self.assert_('[@id <= 3]' in xq.getQuery())
def test_set_collection(self):
# initialize with no collection
xq = Xquery(xpath='/el')
xq.set_collection('coll')
self.assertEquals('collection("/db/coll")/el', xq.getQuery())
# initialize with one collection, then switch
xq = Xquery(collection='coll1')
xq.set_collection('coll2')
self.assertEquals('collection("/db/coll2")/node()', xq.getQuery())
# leading slash is ok too
xq.set_collection('/coll3')
self.assertEquals('collection("/db/coll3")/node()', xq.getQuery())
# set to None
xq.set_collection(None)
self.assertEquals('/node()', xq.getQuery())
def test_sort(self):
xq = Xquery(collection="mycoll")
xq.xq_var = '$n'
xq.sort('@id')
self.assert_('order by $n/@id ascending' in xq.getQuery())
self.assert_('collection("/db/mycoll")' in xq.getQuery())
# prep_xpath function should clean up more complicated xpaths
xq.sort('name|@id')
self.assert_('order by $n/name|$n/@id' in xq.getQuery())
# sort descending
xq.sort('@id', ascending=False)
self.assert_('order by $n/@id descending' in xq.getQuery())
# sort case-insensitive
xq.sort('@id', case_insensitive=True)
self.assert_('order by fn:lower-case($n/@id) ascending' in xq.getQuery())
# case-insensitive and descending
xq.sort('@id', case_insensitive=True, ascending=False)
self.assert_('order by fn:lower-case($n/@id) descending' in xq.getQuery())
def test_fulltext_options(self):
# pass in options for a full-text query
xq = Xquery(xpath='/el', fulltext_options={'default-operator': 'and'})
xq.add_filter('.', 'fulltext_terms', 'dog')
self.assert_('<default-operator>and</default-operator>' in xq.getQuery())
self.assert_('/el[ft:query(., "dog", $ft_options)]', xq.getQuery())
def test_distinct(self):
# distinct-values
xq = Xquery(xpath='/el')
xq.distinct()
self.assertEqual('distinct-values(/el)', xq.getQuery())
def test_clear_limits(self):
xq = Xquery(xpath='/el')
xq.set_limits(low=2, high=5)
xq.clear_limits()
self.assertEqual('/el', xq.getQuery())
def test_defaults(self):
xq = Xquery()
self.assertEquals('/node()', xq.getQuery())
def test_return_also__highlight(self):
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq.return_also({'fulltext_score': ''})
xq.add_filter('.', 'highlight', 'dog star')
self.assert_('(/el[ft:query(., "dog star")]|/el)' in xq.getQuery())
def test_return_also__fulltext_score(self):
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq.return_also({'fulltext_score': ''})
self.assert_('let $fulltext_score := ft:score($n)' in xq.getQuery())
self.assert_('<fulltext_score>{$fulltext_score}</fulltext_score>' in xq._constructReturn())
def test_xpath(self):
xq = Xquery(xpath='/path/to/el')
self.assertEquals('/path/to/el', xq.getQuery())
def test_coll(self):
xq = Xquery(collection='myExistColl')
self.assertEquals('collection("/db/myExistColl")/node()', xq.getQuery())
xq = Xquery(xpath='/root/el', collection='/coll/sub')
self.assertEquals('collection("/db/coll/sub")/root/el', xq.getQuery())
def test_filter_escaping(self):
xq = Xquery(xpath='/el')
xq.add_filter('.', 'contains', '"&')
self.assertEquals('/el[contains(., """&")]', xq.getQuery())
def test_filters_highlight(self):
xq = Xquery(xpath='/el')
xq.add_filter('.', 'highlight', 'dog star')
self.assertEquals('(/el[ft:query(., "dog star")]|/el)', xq.getQuery())
def test_filters_fulltext(self):
xq = Xquery(xpath='/el')
xq.add_filter('.', 'fulltext_terms', 'dog')
self.assertEquals('/el[ft:query(., "dog")]', xq.getQuery())
def test_document(self):
xq = Xquery(xpath='/el', document="/db/coll/file.xml")
self.assertEquals('doc("/db/coll/file.xml")/el', xq.getQuery())
# document takes precedence over collection
xq.set_collection('coll') # should be ignored
self.assertEquals('doc("/db/coll/file.xml")/el', xq.getQuery())