def test_return_also_raw(self):
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq._raw_prefix = 'r_'
xq.return_also({'myid':'count(util:expand(%(xq_var)s/@id))'}, raw=True)
self.assert_('<r_myid>{count(util:expand($n/@id))}</r_myid>' in xq._constructReturn())
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq._raw_prefix = 'r_'
xq.return_also({'myid':'@id'}, raw=True)
self.assert_('<r_myid>{@id}</r_myid>' in xq._constructReturn())
def test_return_only(self):
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq.return_only({'myid':'@id', 'some_name':'name', 'first_letter':'substring(@n,1,1)'})
xq_return = xq._constructReturn()
self.assert_('return <el>' in xq_return)
self.assert_('<field>{$n/@id}</field>' in xq_return)
self.assert_('<field>{$n/name}</field>' in xq_return)
self.assert_('<field>{substring($n/@n,1,1)}</field>' in xq_return)
self.assert_('</el>' in xq_return)
xq = Xquery(xpath='/some/el/notroot')
xq.return_only({'id':'@id'})
self.assert_('return <notroot>' in xq._constructReturn())
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_return_also(self):
xq = Xquery(xpath='/el')
xq.xq_var = '$n'
xq.return_also({'myid':'@id', 'some_name':'name'})
self.assert_('{$n}' in xq._constructReturn())
self.assert_('<field>{$n/@id}</field>' in xq._constructReturn())
