def test_extra2(self):
head1("## 测试如果meta里面多了不是model的字段怎么办")
info("当前的text数量: {}".format(BasicModel.objects.count()))
instance = BasicModel.objects.create()
serializer = serializers.TestExtraFieldSerializer(instance,
data={
"text": "text",
"num": 123
})
serializer.is_valid(raise_exception=True)
serializer.save()
info(serializer.data) # 会直接返回num
info("现在的text数量: {}".format(BasicModel.objects.count()))
head1("## 那么如果覆盖了save函数怎么办")
info("会报错，因为save已经被覆盖了，所以之后调用save()会报错")
serializer = serializers.TestOverWriteSaveSerializer(
instance, data={
"text": "text_oversave",
"num": 123,
"save": 343
})
serializer.is_valid(raise_exception=True)
# import ipdb
# ipdb.set_trace()
# serializer.save()
info(serializer.data)
def test_property(self):
head1("准备测试property的序列化")
info("测试用pro")
data = {}
serializer1 = serializers.TestPropertySerializer(data=data)
serializer1.is_valid()
serializer1.save()
def test_user_modal(self):
head1("## 准备测试用户model")
list1("* 测试model的fields username")
user = User.objects.create()
info("直接创建用户成功")
info("用户的username为空字符串")
self.assertEqual(user.username, "")
def test_path(self):
head1("\n## 测试使用path")
list1("* 测试如果是int的converter遇到")
client = Client()
info("使用负数传入就不支持了")
response = client.get("/testapp/testint/1/")
self.assertEqual(response.resolver_match.kwargs, {"pk": 1})
response = client.get("/testapp/testint/-1/")
self.assertEqual(response.status_code, 200)
def test_template(self):
head1("\n## 测试模板")
list1("* 测试不同的url,的参数的传递")
client = Client()
info("通过request.resolver_match.kwargs可以获取到url里面额外的参数")
response = client.get("/testapp/url/")
self.assertEqual(response.template_name, ["testapp/测试模板.html"])
info("根据这个参数来判断要使用admin的template还是普通的template")
response2 = client.get("/testapp/url2/")
self.assertEqual(response2.template_name, ["testapp/测试模板_admin.html"])
def test_unique(self):
info("准备测试unique")
# TestUniqueModel.objects.create(text1='')
# TestUniqueModel.objects.create(text1='') # 报错，重复
print(TestUniqueModel.objects.create(
text1='',
text2='',
).text3)
print(TestUniqueModel.objects.create(text1='er',
text2='text2').text4) # 报错，重复
def setUp(self):
head1("## 准备测试时间的过滤和排序")
self.test_date_list = [
"2018-05-20T23:59:59+08:00", # 2018-05-20T15:59:59+00:00"
"2018-05-21T00:00:00+08:00", # 2018-05-20T16:00:00+00:00"
"2018-05-21T23:59:59+08:00", # 2018-05-21T15:59:59+00:00
"2018-05-22T00:00:00+08:00", # 2018-05-21T16:00:00+00:00
"2018-05-22T00:00:00+00:00", # 2018-05-22T00:00:00+00:00
"2018-05-22T23:00:00+08:00", # 2018-05-22T15:00:00+00:00
]
for date in self.test_date_list:
datetime_obj = DateTimeModel.objects.create(time=date)
info(datetime_obj)
def test_limit_nested_serializer(self):
head("# 准备测试nest的字段,能否限制数量")
# python3 manage.py test
# testapp.testserializer.MySerializerTestCase.test_limit_nested_serializer
text1 = BasicModel.objects.create(text="text1")
text2 = BasicModel.objects.create(text="text2")
text3 = BasicModel.objects.create(text="text3")
filt1 = ManyModel.objects.create()
filt1.texts.add(text1)
filt1.texts.add(text2)
filt1.texts.add(text3)
info(serializers.TestLimitSerializer(filt1).data)
info("实现不了, 直接用SerializerMethodField吧")
def test_multichoice_filter(self):
head("# 准备测试multichoice filter")
head1("## 过滤CharField")
models.TestFilter.objects.create(_type="类型1")
fenlei1 = models.BasicModel.objects.create(text="分类1")
fenlei2 = models.BasicModel.objects.create(text="分类2")
fenlei3 = models.BasicModel.objects.create(text="分类3")
instance1 = models.TestFilter.objects.create(_type="类型1")
instance2 = models.TestFilter.objects.create(_type="类型2")
instance3 = models.TestFilter.objects.create(_type="类型3")
instance1.basic_model.add(fenlei1)
instance2.basic_model.add(fenlei2)
instance3.basic_model.add(fenlei3)
client = Client()
info("找到类型1和类型2的")
response = client.get("/testapp/testfilter/?_type=类型1&_type=类型2",
headers={"accept": "application/json"})
info(response.data)
head1("## 过滤ManyToManyField")
response = client.get(
"/testapp/testfilter/?basic_model=1&basic_model=2",
headers={"accept": "application/json"})
info(response.data)
response = client.get(
"/testapp/testfilter/?basic_model=1&basic_model=3",
headers={"accept": "application/json"})
info(response.data)
def test_validation(self):
out.write(style.HTTP_INFO("准备测试自己的validation"))
data_list = [
{
"status": "0"
},
{
"status": "1"
},
{
"status": "2"
},
]
for data in data_list:
serializer = serializers.TestValidateSerializer(data=data)
print(serializer.is_valid())
# out.write(style.SUCCESS())
info("测试自己的validation完毕")
def test_many(self):
head1("## 准备测试多对多")
list1("* 有数据后批量创建")
BasicModel.objects.create(text='1')
BasicModel.objects.create(text='2')
a = serializers.ManySerializer(data={'texts': ['1', '3']})
a.is_valid(raise_exception=True)
# a.save(texts=[])
a.save()
success("保存成功")
info(a.instance.texts.all())
success("保存后,如果这个text对应id的basicmodel存在,就会添加")
BasicModel.objects.create(text='text')
a.instance.texts.add(*BasicModel.objects.all())
# print(serializers.ManyDetailSerializer(a.instance).data)
list1("* 准备测试空数组的多对多")
b = serializers.ManySerializer(data={'texts': []})
b.is_valid(raise_exception=True)
b.save()
info("保存成功")
def test_decimal(self):
head1("# 准备测试decimal这个field")
list1("* 测试基础的maxj_digits, decimal_palces超出会怎么样")
data = {"deci": "0.333"}
s = TestDecimalSerializer(data=data)
s.is_valid()
info(s.errors)
info("如果数字数量超出了,会报错, 就算小数位数只有1分,整数部分仍然不能多一位, 反之整数部分小了,小数部分也不能变多")
info("默认不能传入空字符串和None")
def test_cache_control(self):
head1("## 准备测试cache_control")
info("不可以使用cache_control来缓存函数")
start = time.time()
service.slow_function(3)
end = time.time()
info("第一次执行`slow_function`的时间: {}".format(end - start))
start = time.time()
service.slow_function(3)
end = time.time()
info("第二次执行`slow_function`的时间: {}".format(end - start))
def test_viewset_post(self):
head("# 准备测试ViewSet的post请求")
head1("## 看看post调用的是create还是post")
info("如果是post请求，调用的是create函数")
response = client.post(
"/testapp/testfilter/?format=json",
headers={"accept": "application/json"},
json={
"_type": "类型1"
}
)
info(f"服务器返回的状态码: {response.status_code}")
info(f"服务器返回的状态码: {response.content.decode('UTF-8')}")
def test_extra(self):
info("准备测试如果Meta.fields里面少了字段怎么办")
info("会报错")
return
from rest_framework import serializers
class ExtraSerializer(serializers.ModelSerializer):
text = serializers.CharField() # 不行，如果定义了，Meta里面必须有
class Meta:
model = BasicModel
fields = ["id"]
basic_model = BasicModel.objects.first()
info(ExtraSerializer(basic_model).data)
def test_source(self):
head("准备测试source这个参数")
list1("* 测试如果外键为None, 这个field必须设置default, 不然会报错")
basicmodel = BasicModel.objects.create(text='text')
fkm = ForeignKeyModel2.objects.create()
info("数据创建成功")
info(serializers.TestSourceSerializer(fkm).data)
info("可以看到显示的是None")
list1("* 测试如果save, 会发生什么, 他的数据竟然是 {'text': 'text'}, 不是简单的text")
list2(" 1. 如果instance的外键为None")
testsource_ser = serializers.TestSourceSerializer(
instance=fkm, data={"text": "text"})
testsource_ser.is_valid(raise_exception=True)
testsource_ser.save()
list2(" 2. 如果instance的外键不是None, 仍然是 {'text': 'text'}")
fkm.text = basicmodel
fkm.save()
testsource_ser = serializers.TestSourceSerializer(
instance=fkm, data={"text": "text"})
testsource_ser.is_valid(raise_exception=True)
testsource_ser.save()
def test_default(self):
head1("## 测试如果有default会得到什么结果")
for i in [
{},
{
"text": ""
},
{
"text": " "
},
{
"text": None
},
{
"text": "123"
},
]:
serializer = serializers.TestDefaultSerializer(data=i)
info("i= {}".format(i))
if serializer.is_valid():
info("serializer.validated_data: {}".format(
serializer.validated_data))
else:
info("不合法")
def test_datetimefield_filter_date(self):
head1("### 准备测试通过`_date`来过滤时间")
list1("* 普通时间直接过滤2018-05-21")
queryset = DateTimeModel.objects.filter(time__date="2018-05-21")
for instance in queryset:
info(instance)
info("返回了当前时区下时间为2018-05-21的时间")
# new_date = timezone.now() - timezone.timedelta(0, 12*3600)
new_date = timezone.datetime(2018, 5, 21, 23, 0, 0, 0, timezone.utc)
info(new_date)
list2("* utc时间下2018-05-21 23:00:00.date过滤")
date = new_date.date()
info(date)
queryset2 = DateTimeModel.objects.filter(time__date=new_date.date())
for instance in queryset2:
info(instance)
list2("* utc时间下2018-05-21 23:00:00用时间直接过滤")
queryset3 = DateTimeModel.objects.filter(time__date=new_date)
for instance in queryset3:
info(instance)
info("直接输入时间，会先把先把时间转化成本地时区，然后再过滤")
list2("* 当地时区下过滤")
def test_datetime_field(self):
head1("\n## 测试和时间有关的Field")
list1("* 测试DurationField")
datetime_obj = models.DateTimeModel.objects.create(time=timezone.now())
info(isinstance(datetime_obj.duration, timedelta))
def test_local(self):
return
# 别测了. 有bug. 如果你设置user_tz=False, 你就别用timezone参数啊
head1("\n## 测试使用local是的timezone")
data = {'time': '2020-06-10T03:45:13.026Z'} # 当作本地时间的
serializer = serializers.LocalModelSerializer(data=data)
import ipdb
ipdb.set_trace()
serializer.is_valid(raise_exception=True)
info(serializer.validated_data)
return
for data in [
{
'time': '2020-06-10T03:45:13.026Z'
}, # 正常
{
'time': '2020-06-10 03:45:13'
}, # 没有时区,就是正常的
{
'time': '2020-06-10T03:45:13.026+08:00'
}, # 只有这个会转化
{
'time': '2020-06-10T03:45:13.026Z'
} # 当作本地时间的
]:
info("输入: ")
info(data)
serializer = serializers.LocalModelSerializer(data=data)
serializer.is_valid(raise_exception=True)
info(serializer.validated_data)
instance = serializer.save()
info("转化")
info(serializers.LocalModelSerializer(instance).data)
def test_id(self):
head("准备测试id这个field")
list1("* 测试create的时候带id")
data = {"id": 2, "text": "文字"}
info("data: {}".format(data))
serializer = serializers.TestIdSerializer(data=data)
serializer.is_valid(raise_exception=True)
serializer.save()
info("serializer.data: {}".format(serializer.data))
info("如果create的时候带了id参数, validated_data会是不存在id, 并且save的时候id会自动修改")
list1("* 测试patch的时候带id")
instance = serializer.instance
data2 = {"id": 4, "text": "文字2"}
info("data: {}".format(data))
serializer = serializers.TestIdSerializer(instance, data=data)
serializer.is_valid(raise_exception=True)
info("serializer.validated_data: {}".format(serializer.validated_data))
serializer.save()
info("serializer.data: {}".format(serializer.data))
info("如果put的时候带了id参数, validated_data也会过滤掉这个id")
def test_datetime(self):
head1("\n## 测试DateTimeField")
list1("* 准备测试datetime field")
data = {'time': timezone.now(), 'duration': 3600.1}
serializer = serializers.DateTimeModelSerializer(data=data)
serializer.is_valid(raise_exception=True)
info("datetime的对象可以直接当作data传进去")
data = {'time': timezone.now().isoformat(), 'duration': 3600.1}
serializer = serializers.DateTimeModelSerializer(data=data)
serializer.is_valid(raise_exception=True)
info("datetime的国际化时间可以直接传进去")
list1("* 准备测试input_formats")
data = {"time": "2019-01-01", "duration": 3600.1}
serializer = serializers.DateTimeModelSerializer(data=data)
serializer.is_valid(raise_exception=True)
info(serializer.validated_data)
data = {"time": "2019年1月1日", "duration": 3600.1}
serializer = serializers.DateTimeModelSerializer(data=data)
serializer.is_valid(raise_exception=True)
info(serializer.validated_data)
info("可以输入日期格式, 会自动变成local的00点00分00秒")
list1("* 准备测试duration field")
info("duration可以直接传入一个小数，代表多少秒")
serializer.save()
info(serializer.data)
data = {'time': timezone.now(), 'duration': "11 001:2:3.5"}
serializer2 = serializers.DateTimeModelSerializer(data=data)
serializer2.is_valid(raise_exception=True)
serializer2.save()
info("也可以传入符合格式的字符串(前面多写或者少写几个0没关系")
info(serializer2.data)