44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
|
from django.db.models import CharField
|
||
|
|
from django.db.models.functions import Upper
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
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):
|
||
|
|
try:
|
||
|
|
CharField.register_lookup(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
|
||
|
|
)
|
||
|
|
finally:
|
||
|
|
CharField._unregister_lookup(Upper)
|