gpt4 book ai didi

django - 如何重写测试设置中的一个值

转载 作者:行者123 更新时间:2023-12-05 00:48:20 24 4
gpt4 key购买 nike

如何在测试设置中重写 'user': '5/minute' on 'user': '2/day'

REST_FRAMEWORK = {
'DEFAULT_THROTTLE_RATES': {
'user': '5/minute'
}
}

如何使用 @override_settings 或带有上下文管理的 mb 执行此操作?我只需要在一个测试中应用这个,我不能重写整个字典

def test_mytest(self):
value = settings.REST_FRAMEWORK
value['DEFAULT_THROTTLE_RATES'] = {'user':'2/day'}

data1 = {}
data2 = {}
data3 = {}

with self.settings(REST_FRAMEWORK=value):
resp1 = self.client.post(self.url, data1, format='json')
resp2 = self.client.post(self.url, data1, format='json')
resp3 = self.client.post(self.url, data1, format='json')

assert resp3.status_code == 429, resp3.data

但是没有429错误,虽然值已经改变了

最佳答案

    from unittest import mock

@mock.patch('rest_framework.throttling.SimpleRateThrottle.get_rate')
def test_api(self, get_rate):
get_rate.return_value = '1/minute'

from rest_framework.settings import api_settings
print(api_settings.DEFAULT_THROTTLE_RATES)
print(api_settings.user_settings)


url = 'api'

response = self.client.get(url)
self.assertEqual(response.status_code, HTTP_200_OK)

response = self.client.get(url)
self.assertEqual(response.status_code, HTTP_429_TOO_MANY_REQUESTS)

关于django - 如何重写测试设置中的一个值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52037157/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com