44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from django.db.models import CharField
|
|
from django.db.models.functions import Upper
|
|
from django.test import TestCase
|
|
from django.test.utils import register_lookup
|
|
|
|
from ..models import Author
|
|
|
|
|
|
class UpperTests(TestCase):
|
|
def test_basic(self):
|
|
Author.objects.create(name="John Smith", alias="smithj")
|
|
Author.objects.create(name="Rhonda")
|
|
authors = Author.objects.annotate(upper_name=Upper("name"))
|
|
self.assertQuerysetEqual(
|
|
authors.order_by("name"),
|
|
[
|
|
"JOHN SMITH",
|
|
"RHONDA",
|
|
],
|
|
lambda a: a.upper_name,
|
|
)
|
|
Author.objects.update(name=Upper("name"))
|
|
self.assertQuerysetEqual(
|
|
authors.order_by("name"),
|
|
[
|
|
("JOHN SMITH", "JOHN SMITH"),
|
|
("RHONDA", "RHONDA"),
|
|
],
|
|
lambda a: (a.upper_name, a.name),
|
|
)
|
|
|
|
def test_transform(self):
|
|
with register_lookup(CharField, Upper):
|
|
Author.objects.create(name="John Smith", alias="smithj")
|
|
Author.objects.create(name="Rhonda")
|
|
authors = Author.objects.filter(name__upper__exact="JOHN SMITH")
|
|
self.assertQuerysetEqual(
|
|
authors.order_by("name"),
|
|
[
|
|
"John Smith",
|
|
],
|
|
lambda a: a.name,
|
|
)
|