mirror of
https://github.com/django/django.git
synced 2025-09-26 12:09:19 +00:00
Refs #28643 -- Added Repeat database function.
Thanks Tim Graham and Nick Pope for reviews.
This commit is contained in:
parent
6141c752fe
commit
55cc26941a
6 changed files with 64 additions and 4 deletions
24
tests/db_functions/test_repeat.py
Normal file
24
tests/db_functions/test_repeat.py
Normal file
|
@ -0,0 +1,24 @@
|
|||
from django.db.models import CharField, Value
|
||||
from django.db.models.functions import Length, Repeat
|
||||
from django.test import TestCase
|
||||
|
||||
from .models import Author
|
||||
|
||||
|
||||
class RepeatTests(TestCase):
|
||||
def test_basic(self):
|
||||
Author.objects.create(name='John', alias='xyz')
|
||||
tests = (
|
||||
(Repeat('name', 0), ''),
|
||||
(Repeat('name', 2), 'JohnJohn'),
|
||||
(Repeat('name', Length('alias'), output_field=CharField()), 'JohnJohnJohn'),
|
||||
(Repeat(Value('x'), 3, output_field=CharField()), 'xxx'),
|
||||
)
|
||||
for function, repeated_text in tests:
|
||||
with self.subTest(function=function):
|
||||
authors = Author.objects.annotate(repeated_text=function)
|
||||
self.assertQuerysetEqual(authors, [repeated_text], lambda a: a.repeated_text, ordered=False)
|
||||
|
||||
def test_negative_number(self):
|
||||
with self.assertRaisesMessage(ValueError, "'number' must be greater or equal to 0."):
|
||||
Repeat('name', -1)
|
Loading…
Add table
Add a link
Reference in a new issue