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 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())
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 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 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
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 test_person(self):
person = Person()
print "Vanilla - test_person. Person Count", Person.objects.all()
