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")
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')