Django-测试case

import pymysql
from django.contrib.auth.models import User
from django.test import TestCase, Client

# Create your tests here.
from django.utils.datetime_safe import datetime

from loginApp.models import Guest, Event


class ModelTest(TestCase):
    '''
    在 setUp()初始化方法中，创建一条发布会和嘉宾数据。
    '''
    def setUp(self):
        Event.objects.create(id=1, name="oneplus 3 event", status=True, limit=2000, address='shenzhen', start_time='2016-08-31 02:18:22')
        Guest.objects.create(id=1,event_id=1,realname='alen', phone='13711001101',email='alen@mail.com', sign=False)

    def test_event_models(self):
        result = Event.objects.get(name="oneplus 3 event")
        self.assertEqual(result.address, "shenzhen")
        self.assertTrue(result.status)

    def test_guest_models(self):
        result = Guest.objects.get(phone='13711001101')
        self.assertEqual(result.realname, "alen")
        self.assertFalse(result.sign)

class LoginActionTest(TestCase):
    ''' 测试登录函数'''
    def setUp(self):
        User.objects.create_user('admin', 'admin@mail.com', 'admin123456')
        self.c = Client()

    def test_login_action_username_password_null(self):
        ''' 用户名密码为空 '''
        test_data = {'username':'','password':''}
        response = self.c.post('/login_action/', data=test_data)
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"username or password error!", response.content)

    def test_login_action_username_password_error(self):
        ''' 用户名密码错误 '''
        test_data = {'username': 'abc', 'password': '123'}
        response = self.c.post('/login_action/', data=test_data)
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"username or password error!", response.content)

    def test_login_action_success(self):
        ''' 登录成功 '''
        test_data = {'username': 'admin', 'password': 'admin123456'}
        response = self.c.post('/login_action/', data=test_data)
        self.assertEqual(response.status_code, 302)

class EventManageTest(TestCase):
    ''' 发布会管理 '''
    def setUp(self):
        Event.objects.create(id=2,name='xiaomi5',limit=2000,status=True, address='beijing',start_time=datetime(2016,8,10,14,0,0))
        self.c = Client()

    def test_event_manage_success(self):
        ''' 测试发布会:xiaomi5 '''
        response = self.c.post('/event_manage/')
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"xiaomi5", response.content)
        self.assertIn(b"beijing", response.content)
    def test_event_manage_search_success(self):
        ''' 测试发布会搜索 '''
        response = self.c.post('/search_name/',{"name":"xiaomi5"})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"xiaomi5", response.content)
        self.assertIn(b"beijing", response.content)

class GuestManageTest(TestCase):
    ''' 嘉宾管理 '''
    def setUp(self):
        Event.objects.create(id=1,name="xiaomi5",limit=2000, address='beijing',status=1,start_time=datetime(2016,8,10,14,0,0))
        Guest.objects.create(realname="alen",phone=18611001100, email='alen@mail.com',sign=0,event_id=1)
        self.c = Client()

    def test_event_manage_success(self):
        ''' 测试嘉宾信息: alen '''
        response = self.c.post('/guest_manage/')
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"alen", response.content)
        self.assertIn(b"18611001100", response.content)

    def test_guest_manage_search_success(self):
        ''' 测试嘉宾搜索 '''
        response = self.c.post('/search_phone/',{"phone":"18611001100"})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"alen", response.content)
        self.assertIn(b"18611001100", response.content)

class SignIndexActionTest(TestCase):
    ''' 发布会签到 '''
    def setUp(self):
        Event.objects.create(id=1,name="xiaomi5",limit=2000,address='beijing', status=1,start_time='2017-8-10 12:30:00')
        Event.objects.create(id=2,name="oneplus4",limit=2000,address='shenzhen', status=1,start_time='2017-6-10 12:30:00')
        Guest.objects.create(realname="alen",phone=18611001100, email='alen@mail.com',sign=0,event_id=1)
        Guest.objects.create(realname="una",phone=18611001101, email='una@mail.com',sign=1,event_id=2)
        self.c = Client()

    def test_sign_index_action_phone_null(self):
        ''' 手机号为空 '''
        response = self.c.post('/sign_index_action/1/',{"phone":""})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"phone error.", response.content)

    def test_sign_index_action_phone_or_event_id_error(self):
        ''' 手机号或发布会 id 错误 '''
        response = self.c.post('/sign_index_action/2/',{"phone":"18611001100"})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"event id or phone error.", response.content)
    def test_sign_index_action_user_sign_has(self):
        ''' 用户已签到 '''
        response = self.c.post('/sign_index_action/2/',{"phone":"18611001101"})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"user has sign in.", response.content)
    def test_sign_index_action_sign_success(self):
        ''' 签到成功 '''
        response = self.c.post('/sign_index_action/1/',{"phone":"18611001100"})
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"sign in success!", response.content)

　　运行：python3 manage.py test