1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
# Copyright (c) Microsoft Corporation.
# Licensed under the BSD license.
from unittest import skipUnless
from django import VERSION
from django.core.exceptions import ValidationError
from django.db import OperationalError
"""
Import BaseDatabaseOperations before sqlite3.operations.DatabaseOperations because:
1. DatabaseOperations in sqlite3.operations inherits from BaseDatabaseOperations.
2. Importing the base class first avoids potential circular import issues.
3. This order ensures the base class is available for subclassing.
4. This behavior was introduced in Django 5.1 and is backward compatible with earlier versions.
"""
from django.db.backends.base.operations import BaseDatabaseOperations
from django.test import TestCase, skipUnlessDBFeature
from ..models import BinaryData, Pizza, Topping
if VERSION >= (3, 2):
from ..models import TestCheckConstraintWithUnicode
@skipUnless(
VERSION >= (3, 1),
"Django 3.0 and below doesn't support different databases in unit tests",
)
class TestMultpleDatabases(TestCase):
databases = ['default', 'sqlite']
def test_in_split_parameter_list_as_sql(self):
# Issue: https://github.com/microsoft/mssql-django/issues/92
# Mimic databases that have a limit on parameters (e.g. Oracle DB)
old_max_in_list_size = BaseDatabaseOperations.max_in_list_size
BaseDatabaseOperations.max_in_list_size = lambda self: 100
mssql_iterations = 3000
Pizza.objects.bulk_create([Pizza() for _ in range(mssql_iterations)])
Topping.objects.bulk_create([Topping() for _ in range(mssql_iterations)])
prefetch_result = Pizza.objects.prefetch_related('toppings')
self.assertEqual(len(prefetch_result), mssql_iterations)
# Different iterations since SQLite has max host parameters of 999 for versions prior to 3.32.0
# Info about limit: https://www.sqlite.org/limits.html
sqlite_iterations = 999
Pizza.objects.using('sqlite').bulk_create([Pizza() for _ in range(sqlite_iterations)])
Topping.objects.using('sqlite').bulk_create([Topping() for _ in range(sqlite_iterations)])
prefetch_result_sqlite = Pizza.objects.using('sqlite').prefetch_related('toppings')
self.assertEqual(len(prefetch_result_sqlite), sqlite_iterations)
BaseDatabaseOperations.max_in_list_size = old_max_in_list_size
def test_binaryfield_init(self):
binary_data = b'\x00\x46\xFE'
binary = BinaryData(binary=binary_data)
binary.save()
binary.save(using='sqlite')
try:
binary.full_clean()
except ValidationError:
self.fail()
b1 = BinaryData.objects.filter(binary=binary_data)
self.assertSequenceEqual(
b1,
[binary],
)
b2 = BinaryData.objects.using('sqlite').filter(binary=binary_data)
self.assertSequenceEqual(
b2,
[binary],
)
@skipUnlessDBFeature('supports_table_check_constraints')
@skipUnless(
VERSION >= (3, 2),
"Django 3.1 and below has errors from running migrations for this test",
)
def test_checkconstraint_get_check_sql(self):
TestCheckConstraintWithUnicode.objects.create(name='abc')
try:
TestCheckConstraintWithUnicode.objects.using('sqlite').create(name='abc')
except OperationalError:
self.fail()
def test_queryset_bulk_update(self):
objs = [
BinaryData.objects.create(binary=b'\x00') for _ in range(5)
]
for obj in objs:
obj.binary = None
BinaryData.objects.bulk_update(objs, ["binary"])
self.assertCountEqual(BinaryData.objects.filter(binary__isnull=True), objs)
objs = [
BinaryData.objects.using('sqlite').create(binary=b'\x00') for _ in range(5)
]
for obj in objs:
obj.binary = None
BinaryData.objects.using('sqlite').bulk_update(objs, ["binary"])
self.assertCountEqual(BinaryData.objects.using('sqlite').filter(binary__isnull=True), objs)
|
