Commit b47e862d authored by Alex Hill's avatar Alex Hill Committed by Tim Graham
Browse files

Added test for ForeignObject.get_extra_descriptor_filter()

parent 14c1fd07

tests/foreign_object/models.py

+1 −1
Original line number Diff line number Diff line
@@ -146,7 +146,7 @@ class ActiveTranslationField(models.ForeignObject): 
    def get_extra_restriction(self, where_class, alias, related_alias):

        return ColConstraint(alias, 'lang', get_language())



    def get_extra_descriptor_filter(self):

    def get_extra_descriptor_filter(self, instance):

        return {'lang': get_language()}



    def contribute_to_class(self, cls, name):

tests/foreign_object/tests.py

+3 −0
Original line number Diff line number Diff line
@@ -319,6 +319,9 @@ class MultiColumnFKTests(TestCase): 
        at1_fi.save()

        at2_en = ArticleTranslation(article=a1, lang='en', title='Title', body='Lalalalala')

        at2_en.save()



        self.assertEqual(Article.objects.get(pk=a1.pk).active_translation, at1_fi)



        with self.assertNumQueries(1):

            fetched = Article.objects.select_related('active_translation').get(

                active_translation__title='Otsikko')