def test_get_device_model_mock(self, mock_get_device_model):
mock_get_device_model.return_value = Device
response = utils.get_device_model()
self.assertIs(response, Device)
from mock import patch
from django.contrib.auth import get_user_model
from django.core.urlresolvers import reverse
from django.test import TestCase, override_settings
from rest_framework import status
from rest_framework.test import APITestCase
from gcm import utils
User = get_user_model()
Device = utils.get_device_model()
class ApiDeviceTest(APITestCase):
def setUp(self):
self.user = User.objects.create(username='******', email='*****@*****.**')
self.user.set_password('test')
self.user.save()
self.client.login(username='******', password='******')
self.device = Device.objects.create(dev_id='123-1', dev_type='ANDROID', reg_id='123-1')
def test_resgister_device(self):
url = reverse('device-list')
data = {'dev_id': '123-2', 'dev_type': 'ANDROID', 'reg_id': '123-2'}
response = self.client.post(url, data, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(response.data, data)
def test_unresgister_device(self):
kwargs = {'pk': 1}
url = reverse('device-detail', kwargs=kwargs)
def test_get_device_model_real(self):
response = utils.get_device_model()
self.assertIs(response, Device)
from rest_framework.serializers import ModelSerializer
from gcm.utils import get_device_model
Device = get_device_model()
class DeviceSerializer(ModelSerializer):
class Meta:
model = Device
exclude = ("id", "creation_date", "modified_date", "is_active")
def test_get_device_model_mock(self, mock_get_device_model):
mock_get_device_model.return_value = Device
response = utils.get_device_model()
self.assertIs(response, Device)
def test_get_device_model_real(self):
response = utils.get_device_model()
self.assertIs(response, Device)
from mock import patch
from django.contrib.auth import get_user_model
from django.core.urlresolvers import reverse
from django.test import TestCase, override_settings
from rest_framework import status
from rest_framework.test import APITestCase
from gcm import utils
User = get_user_model()
Device = utils.get_device_model()
class ApiDeviceTest(APITestCase):
def setUp(self):
self.user = User.objects.create(username='******', email='*****@*****.**')
self.user.set_password('test')
self.user.save()
self.client.login(username='******', password='******')
self.device = Device.objects.create(dev_id='123-1',
dev_type='ANDROID',
reg_id='123-1')
def test_resgister_device(self):
url = reverse('device-list')
data = {'dev_id': '123-2', 'dev_type': 'ANDROID', 'reg_id': '123-2'}
response = self.client.post(url, data, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(response.data, data)
def test_unresgister_device(self):
from django.contrib import admin
from gcm.utils import get_device_model
Device = get_device_model()
@admin.register(Device)
class DeviceAdmin(admin.ModelAdmin):
list_display = ['dev_id', 'dev_type', 'modified_date', 'is_active']
search_fields = ('dev_id', 'dev_type')
list_filter = ['is_active']
date_hierarchy = 'modified_date'
readonly_fields = ('dev_id', 'reg_id')
