def test_get_max_db_param_values_user_values(db_name, db_version, user_limit):
os.environ['USER_DB_MAX_PARAMS'] = str(user_limit)
with patch.object(connection, 'vendor',
db_name), patch.object(sqlite3, 'sqlite_version',
db_version):
assert get_max_params_for_db() == user_limit
def test_get_max_db_param_values_user_values_invalid(caplog):
os.environ['USER_DB_MAX_PARAMS'] = 'not int invalid'
with patch.object(connection, 'vendor', 'sqlite'),\
patch.object(sqlite3, 'sqlite_version', '3.32.0'),\
caplog.at_level(logging.ERROR):
assert get_max_params_for_db() == 32766
assert F'Invalid Environment Variable "USER_DB_MAX_PARAMS", int expected but got "not int invalid".' in caplog.text
def test_ordering_max_limit_exceeded(fixture_property_number_filter, monkeypatch, max_params, expected_result_list):
monkeypatch.setenv('USER_DB_MAX_PARAMS', str(max_params), prepend=False)
assert get_max_params_for_db() == max_params
class PropertyOrderingFilterSet(PropertyFilterSet):
prop_age = PropertyOrderingFilter(fields=('prop_age', 'prop_age'))
class Meta:
model = OrderingFilterModel
exclude = ['first_name', 'last_name', 'username', 'age']
prop_filter_fs = PropertyOrderingFilterSet({'prop_age': 'prop_age'}, queryset=OrderingFilterModel.objects.all())
with patch.object(QuerySet, 'count') as mock_method:
mock_method.side_effect = (OperationalError(), None)
assert list(prop_filter_fs.qs.values_list('id', flat=True)) == expected_result_list
def test_get_max_db_param_values(db_name, db_version, max_params):
with patch.object(connection, 'vendor',
db_name), patch.object(sqlite3, 'sqlite_version',
db_version):
assert get_max_params_for_db() == max_params