Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from ast import literal_eval as str_to_dict
- from django.core.urlresolvers import reverse
- def test_department_import(self):
- user = User.objects.create(username='test', email='admin@admin.com', password='Tedt123123', is_staff=True)
- url_path = reverse('department_import')
- response = self.client.get(url_path)
- self.assertEqual(response.status_code, 422)
- response_error = str_to_dict(response.content.decode("utf-8"))
- self.assertEqual(response_error['errors'], "Empty POST")
- response = self.client.post(url_path)
- self.assertEqual(response.status_code, 422)
- name = "Vasa"
- response = self.client.post(url_path, data={'name': name})
- self.assertEqual(response.status_code, 200)
- response_data = str_to_dict(response.content.decode("utf-8"))
- self.assertEqual(response_data['name'], name)
- self.assertEqual(response_data['id'], 1)
- self.assertTrue(Department.objects.filter(name=response_data['name']).exists())
- def test_user_import(self):
- user = User.objects.create(username='test', email='admin@admin.com', password='Tedt123123', is_staff=True)
- url_path = reverse('user_import')
- response = self.client.get(url_path)
- self.assertEqual(response.status_code, 422)
- response_data = str_to_dict(response.content.decode("utf-8"))
- self.assertEqual(response_data['errors'], "Empty POST")
- response = self.client.post(url_path)
- self.assertEqual(response.status_code, 422)
- new_department = Department.objects.create(name='new_department')
- new_user = User(username='Petya', email='Petya@koze.ru', password='123Petya')
- response = self.client.post(url_path, {
- 'department': new_department.id,
- 'username': new_user.username,
- 'email': new_user.email,
- })
- self.assertEqual(response.status_code, 200)
- response_data = str_to_dict(response.content.decode("utf-8"))
- self.assertEqual(response_data['username'], new_user.username)
- self.assertEqual(response_data['email'], new_user.email)
- new_user = User.objects.get(username=new_user.username)
- self.assertTrue(new_user.check_password(response_data['password']))
- # TODO не забудь
- def user_import(request):
- # TODO там поправить
- # user.departments.add(Department.objects.get(id=form.cleaned_data['department']))
- user.departments.add(form.cleaned_data['department'])
- # TODO urls
- url(r'^import/department/$', api.department_import, name='department_import'),
- url(r'^import/user/$', api.user_import, name='user_import'),
- # TODO в мидлвару except_urls = ('/login/', '/import/department', '/import/user')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement