diff options
author | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-05-16 19:56:44 +0200 |
---|---|---|
committer | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-05-17 12:26:15 +0200 |
commit | a2dd618e3b4a7472fab852da450ca5eef92a922f (patch) | |
tree | 029b85facc31b4c7d4ee19afe99dfc05e7a00bd1 /tests/custom_lookups | |
parent | 11932e978f980846d2c3326ff070ece7e65bf75c (diff) | |
download | django-a2dd618e3b4a7472fab852da450ca5eef92a922f.tar.gz |
Fixed #22648 -- Transform.output_type should respect overridden custom_lookup and custom_transform.
Previously, class lookups from the output_type would be used, but any
changes to custom_lookup or custom_transform would be ignored.
Diffstat (limited to 'tests/custom_lookups')
-rw-r--r-- | tests/custom_lookups/tests.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/tests/custom_lookups/tests.py b/tests/custom_lookups/tests.py index 396974b4b1..c7af60b54d 100644 --- a/tests/custom_lookups/tests.py +++ b/tests/custom_lookups/tests.py @@ -89,6 +89,47 @@ class YearLte(models.lookups.LessThanOrEqual): YearTransform.register_lookup(YearLte) +class SQLFunc(models.Lookup): + def __init__(self, name, *args, **kwargs): + super(SQLFunc, self).__init__(*args, **kwargs) + self.name = name + + def as_sql(self, qn, connection): + return '%s()', [self.name] + + @property + def output_type(self): + return CustomField() + + +class SQLFuncFactory(object): + + def __init__(self, name): + self.name = name + + def __call__(self, *args, **kwargs): + return SQLFunc(self.name, *args, **kwargs) + + +class CustomField(models.Field): + + def get_lookup(self, lookup_name): + if lookup_name.startswith('lookupfunc_'): + key, name = lookup_name.split('_', 1) + return SQLFuncFactory(name) + return super(CustomField, self).get_lookup(lookup_name) + + def get_transform(self, lookup_name): + if lookup_name.startswith('transformfunc_'): + key, name = lookup_name.split('_', 1) + return SQLFuncFactory(name) + return super(CustomField, self).get_transform(lookup_name) + + +class CustomModel(models.Model): + field = CustomField() + + # We will register this class temporarily in the test method. @@ -341,3 +382,22 @@ class LookupTransformCallOrderTests(TestCase): finally: models.DateField._unregister_lookup(TrackCallsYearTransform) + + +class CustomisedMethodsTests(TestCase): + + def test_overridden_get_lookup(self): + q = CustomModel.objects.filter(field__lookupfunc_monkeys=3) + self.assertIn('monkeys()', str(q.query)) + + def test_overridden_get_transform(self): + q = CustomModel.objects.filter(field__transformfunc_banana=3) + self.assertIn('banana()', str(q.query)) + + def test_overridden_get_lookup_chain(self): + q = CustomModel.objects.filter(field__transformfunc_banana__lookupfunc_elephants=3) + self.assertIn('elephants()', str(q.query)) + + def test_overridden_get_transform_chain(self): + q = CustomModel.objects.filter(field__transformfunc_banana__transformfunc_pear=3) + self.assertIn('pear()', str(q.query)) |