def test_signals(self): count = Log.objects.count() p = Person(name=self.username, date='2010-10-10') p.save() self.failUnlessEqual(count + 1, Log.objects.count()) p.delete() self.failUnlessEqual(count + 2, Log.objects.count())
class TestQueryList(APITestCase): def setUp(self): f = Department(is_superuser=False, name='调度车间') f.save() Department.objects.create(is_superuser=False, name='出发场车间') b = Department(is_superuser=True, name='站领导') b.save() self.custom_user = Person(name='test', department=f, user=User.objects.create(username='******', password='******')) self.custom_user.save() self.super_user = Person(name='super', department=b, user=User.objects.create(username='******', password='******')) self.super_user.save() def test_custom_person_get_data(self): ''' 测试普通用户的请求,有时间段 :return: ''' url = reverse('query-call-over') self.client.force_login(User.objects.get(username='******')) response = self.client.get(url, data={'start': '2016-7-20', 'end': '2016-8-31'}) assert isinstance(response, Response) assert self.assertEqual(response.status_code, 201, response.data) def test_custom_person_get_data_without_time(self): ''' 测试普通用户的请求,无时间段 :return: ''' url = reverse('query-call-over') self.client.force_login(User.objects.get(username='******')) response = self.client.get(url)
def show_tepmlate_persons(request): if request.method == 'POST': person_name = request.POST.get('person_name') person_name = Person(Name=person_name) person_name.save() page = 'persons' data = Person.objects.all() context = {'page': page, 'data': data} return render(request, 'persons.html', context)
def __update_person(self, user, user_infos): persons = Person.objects.filter(user=user) if not persons: person = Person(user=user, global_id=user_infos['USER_FGS'], first_name=user_infos['USER_FIRST_NAME'], last_name=user_infos['USER_LAST_NAME'],email=user_infos['USER_EMAIL']) else: person = persons[0] person.first_name = user.first_name person.last_name = user.last_name person.email = user.email person.save()
def create(self, validated_data): user = User.objects.create_user(**self.initial_data) user.save() # person = Person(user=user, **person_data) person = Person(user=user) person.save() is_user = self.initial_data['is_user'] if is_user == 'customer': customer = Customer(person_info=person) customer.save() elif is_user == 'vendor': vendor = Vendor(person_info=person) vendor.save() else: '''Here goes Partner''' return user