From 3c8bea0fc39bd1311334937e5a787e6a48b46ac4 Mon Sep 17 00:00:00 2001 From: Swen Kooij Date: Tue, 30 May 2017 13:47:48 +0300 Subject: [PATCH] Add test for LocalizedRef in combination with ArrayAgg --- tests/test_expressions.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_expressions.py b/tests/test_expressions.py index dd01441..91a2c98 100644 --- a/tests/test_expressions.py +++ b/tests/test_expressions.py @@ -2,6 +2,7 @@ from django.test import TestCase from django.db import models from django.utils import translation from django.conf import settings +from django.contrib.postgres.aggregates import ArrayAgg from localized_fields.fields import LocalizedField from localized_fields.value import LocalizedValue @@ -77,3 +78,9 @@ class LocalizedExpressionsTestCase(TestCase): for index, value in enumerate(queryset): assert 'ro' in value assert str(index) in value + + # ensures that using this in combination with ArrayAgg works properly + queryset = create_queryset(ArrayAgg(LocalizedRef('features__text', 'ro'))).first() + assert isinstance(queryset, list) + for value in queryset: + assert 'ro' in value