Advertisement
Guest User

Untitled

a guest
Aug 10th, 2016
108
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.68 KB | None | 0 0
  1. # -*- coding: utf-8 -*-
  2. from django.contrib.auth.models import User
  3. from django.core.urlresolvers import reverse
  4. from django.test import TestCase
  5. from rest_framework import status
  6. from rest_framework.test import APITestCase
  7. from .models import IosDeviceToken
  8. from .tasks import send_async_ios_notify
  9. from events.models import Event, Comment
  10. from profiles.models import Profile
  11. from datetime import datetime
  12. from messaging.message import send_message_to_users
  13. from django.test.utils import override_settings
  14.  
  15.  
  16. class AccountTests(APITestCase):
  17. EMAIL = 'a71d97l747@gmail.com'
  18. PASSWORD = 'g315mpmp'
  19. TOKEN = 'a5db23392b7149b9e0a71c475748f275157a470498af3be1fb8de15213b89766'
  20.  
  21. def test_push_to_device(self):
  22. user = User.objects.create_user(email=self.EMAIL, username=self.EMAIL, password=self.PASSWORD)
  23. profile = Profile.objects.create(user=user)
  24. IosDeviceToken.objects.create(profile=profile, token=self.TOKEN)
  25. self.client.login(password=self.PASSWORD, username=self.EMAIL)
  26. response = send_async_ios_notify(u'Проверочка для Вовы', self.TOKEN)
  27.  
  28. def test_push_to_device_through_signal(self):
  29. user1 = User.objects.create_user(email=self.EMAIL, username=self.EMAIL, password=self.PASSWORD)
  30. user2 = User.objects.create_user(email='1{}'.format(self.EMAIL), username='1{}'.format(self.EMAIL),
  31. password='1{}'.format(self.PASSWORD))
  32. profile = Profile.objects.create(user=user1)
  33. IosDeviceToken.objects.create(profile=profile, token=self.TOKEN)
  34. event = Event.objects.create(organizer=user1, date=datetime.now())
  35. Comment.objects.create(user=user2, event=event, text='asdasd')
  36. self.assertEquals(Comment.objects.all().count(), 1)
  37.  
  38. def test_api_add_device(self):
  39. user = User.objects.create_user(email=self.EMAIL, username=self.EMAIL, password=self.PASSWORD)
  40. profile = Profile.objects.create(user=user)
  41. self.client.login(password=self.PASSWORD, username=self.EMAIL)
  42. url = reverse('api_ios_token_create')
  43. data = {'profile': profile.id, 'token': self.TOKEN}
  44. response = self.client.post(url, data, format='json')
  45. self.assertEquals(response.status_code, 201)
  46. self.assertEquals(IosDeviceToken.objects.all().count(), 1)
  47.  
  48. def test_push_message_from_chat(self):
  49. user = User.objects.create_user(email=self.EMAIL, username=self.EMAIL, password=self.PASSWORD)
  50. profile = Profile.objects.create(user=user)
  51. IosDeviceToken.objects.create(profile=profile, token=self.TOKEN)
  52. send_message_to_users('test', [0, 1, 2], dict(text=u'Проверочка'))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement