Advertisement
Guest User

Untitled

a guest
Mar 28th, 2016
132
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.56 KB | None | 0 0
  1. from ast import literal_eval as str_to_dict
  2. from django.core.urlresolvers import reverse
  3.  
  4. def test_department_import(self):
  5. user = User.objects.create(username='test', email='admin@admin.com', password='Tedt123123', is_staff=True)
  6.  
  7. url_path = reverse('department_import')
  8. response = self.client.get(url_path)
  9. self.assertEqual(response.status_code, 422)
  10. response_error = str_to_dict(response.content.decode("utf-8"))
  11. self.assertEqual(response_error['errors'], "Empty POST")
  12.  
  13. response = self.client.post(url_path)
  14. self.assertEqual(response.status_code, 422)
  15.  
  16. name = "Vasa"
  17. response = self.client.post(url_path, data={'name': name})
  18. self.assertEqual(response.status_code, 200)
  19. response_data = str_to_dict(response.content.decode("utf-8"))
  20. self.assertEqual(response_data['name'], name)
  21. self.assertEqual(response_data['id'], 1)
  22.  
  23. self.assertTrue(Department.objects.filter(name=response_data['name']).exists())
  24.  
  25. def test_user_import(self):
  26. user = User.objects.create(username='test', email='admin@admin.com', password='Tedt123123', is_staff=True)
  27.  
  28. url_path = reverse('user_import')
  29. response = self.client.get(url_path)
  30.  
  31. self.assertEqual(response.status_code, 422)
  32. response_data = str_to_dict(response.content.decode("utf-8"))
  33. self.assertEqual(response_data['errors'], "Empty POST")
  34.  
  35. response = self.client.post(url_path)
  36. self.assertEqual(response.status_code, 422)
  37.  
  38. new_department = Department.objects.create(name='new_department')
  39. new_user = User(username='Petya', email='Petya@koze.ru', password='123Petya')
  40. response = self.client.post(url_path, {
  41. 'department': new_department.id,
  42. 'username': new_user.username,
  43. 'email': new_user.email,
  44. })
  45. self.assertEqual(response.status_code, 200)
  46. response_data = str_to_dict(response.content.decode("utf-8"))
  47. self.assertEqual(response_data['username'], new_user.username)
  48. self.assertEqual(response_data['email'], new_user.email)
  49.  
  50. new_user = User.objects.get(username=new_user.username)
  51. self.assertTrue(new_user.check_password(response_data['password']))
  52.  
  53. # TODO не забудь
  54. def user_import(request):
  55. # TODO там поправить
  56. # user.departments.add(Department.objects.get(id=form.cleaned_data['department']))
  57. user.departments.add(form.cleaned_data['department'])
  58.  
  59. # TODO urls
  60. url(r'^import/department/$', api.department_import, name='department_import'),
  61. url(r'^import/user/$', api.user_import, name='user_import'),
  62.  
  63. # TODO в мидлвару except_urls = ('/login/', '/import/department', '/import/user')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement