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_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_method(self):
head1("## 准备测试method里面的数据")
list1("* 测试如果SerializerMethodField不返回数据会怎么样")
data = {"text": 'text'}
s = serializers.TestMethodSerializer(data=data)
s.is_valid()
s.save()
out.write(style.HTTP_INFO(s.data))
out.write(style.ERROR(s.errors))
return
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_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_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_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_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)
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_manytomany_field(self):
head1("\n## 测试ManyToManyField")
list1("* 如果symmetrical=True默认")
student = Student.objects.create(name="George")
student2 = Student.objects.create(name="Li Lei")
student3 = Student.objects.create(name="Han Mei")
student2.friends.add(student)
student2.friends.add(student3)
student3.friends.add(student2)
print("student2的好友有:", end=" ")
print(student2.friends.all())
print("认为我是好友的有(只能是一样的):", end=" ")
# print(student2.student_set.all()) 这个会报错
print(student2.friends.all())
list1("* 如果symmetrical=False")
teacher = Teacher.objects.create(name="王主任")
teacher2 = Teacher.objects.create(name="小李主任")
teacher3 = Teacher.objects.create(name="小红主任")
teacher2.teachers.add(teacher)
teacher2.teachers.add(teacher3)
teacher3.teachers.add(teacher)
print("给teacher3指导过的有:", end=" ")
print(teacher3.teachers.all())
print("teacher3指导过的人有:", end=" ")
# import ipdb
# ipdb.set_trace()
print(teacher3.teacher_set.all())
list1("* 检查是否存在")
new_student = Student.objects.create(name="New Student")
print("现在new_student的friends里面有没有George", student
in new_student.friends.all())
result = new_student.friends.add(student)
print("add以后现在new_student的friends里面有没有George", student
in new_student.friends.all())
print("但是add的return: ", result)
head1("ManyToManyField测试完毕")
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_request_data(self):
head1("准备测试request.data")
list1("* 测试request.data类型,是否可变")
client = Client()
response = client.post("/testapp/basicmodel/", {"text": "text"})
# -*- coding: utf-8 -*-
# Xiang Wang @ 2018-06-07 17:49:05
import sys
from django.core.management.base import OutputWrapper
from django.core.management.color import color_style
from django.test import Client
from django.test import TestCase
from testapp import models, filters
from testapp import head, head1, list1, list2, info
out = OutputWrapper(sys.stdout)
style = color_style()
list1("list1")
list2("list2")
class FilterTestCase(TestCase):
def setUp(self):
out.write(style.MIGRATE_HEADING("准备测试过滤功能"))
datas = [[{
"_bool": _bool,
"_int": _int
} for _int in [0, 1, -1]] for _bool in [True, False, None]]
for data_1 in datas:
for data in data_1:
# print(data)
models.TestFilterModel2.objects.create(**data)
