Commit 980eda01 authored by Anssi Kääriäinen's avatar Anssi Kääriäinen
Browse files

Fixed custom_lookups tests for Oracle 

Refs #21821
parent 1c360dbb

tests/custom_lookups/tests.py

+12 −0
Original line number Diff line number Diff line 
from __future__ import unicode_literals



from datetime import date

import unittest
@@ -16,6 +18,12 @@ class Div3Lookup(models.Lookup): 
        params.extend(rhs_params)

        return '%s %%%% 3 = %s' % (lhs, rhs), params



    def as_oracle(self, qn, connection):

        lhs, params = self.process_lhs(qn, connection)

        rhs, rhs_params = self.process_rhs(qn, connection)

        params.extend(rhs_params)

        return 'mod(%s, 3) = %s' % (lhs, rhs), params





class Div3Transform(models.Transform):

    lookup_name = 'div3'
@@ -24,6 +32,10 @@ class Div3Transform(models.Transform): 
        lhs, lhs_params = qn.compile(self.lhs)

        return '%s %%%% 3' % (lhs,), lhs_params



    def as_oracle(self, qn, connection):

        lhs, lhs_params = qn.compile(self.lhs)

        return 'mod(%s, 3)' % lhs, lhs_params





class YearTransform(models.Transform):

    lookup_name = 'year'