|
3 | 3 |
|
4 | 4 | from django.test import TestCase |
5 | 5 | from django.dispatch import receiver |
6 | | -from django.db import models |
| 6 | +from django.contrib.auth.models import User |
7 | 7 | from django.contrib.contenttypes.models import ContentType |
8 | 8 |
|
9 | 9 | from .models import EncodeJob |
|
14 | 14 | ) |
15 | 15 |
|
16 | 16 |
|
17 | | - |
18 | 17 | PROJECT_ROOT = os.path.dirname(os.path.realpath(__file__)) |
19 | 18 | FIXTURE_DIRS = os.path.join(PROJECT_ROOT, 'fixtures') |
20 | 19 |
|
21 | 20 |
|
22 | | -class Item(models.Model): |
23 | | - name = models.CharField(max_length=100) |
24 | | - |
25 | 21 | # ====================== |
26 | 22 | # define signal receiver |
27 | 23 | # ====================== |
28 | 24 |
|
| 25 | + |
29 | 26 | @receiver(transcode_onprogress) |
30 | 27 | def encode_onprogress(sender, message, **kwargs): |
31 | 28 | job = EncodeJob.objects.get(pk=message['jobId']) |
@@ -54,8 +51,8 @@ class SNSNotificationTest(TestCase): |
54 | 51 | urls = 'dj_elastictranscoder.urls' |
55 | 52 |
|
56 | 53 | def setUp(self): |
57 | | - item = Item.objects.create(name='Hello') |
58 | | - content_type = ContentType.objects.get_for_model(Item) |
| 54 | + item = User.objects.create(username='Hello') |
| 55 | + content_type = ContentType.objects.get_for_model(User) |
59 | 56 | self.job_id = '1396802241671-jkmme8' |
60 | 57 |
|
61 | 58 | self.job = EncodeJob.objects.create(id=self.job_id, content_type=content_type, object_id=item.id) |
@@ -105,9 +102,8 @@ def test_transcode_onprogress(self): |
105 | 102 | """ |
106 | 103 |
|
107 | 104 | # assume an encode job was submitted |
108 | | - item = Item.objects.create(name='Hello') |
109 | | - |
110 | | - ctype = ContentType.objects.get_for_model(item) |
| 105 | + item = User.objects.create(username='Hello') |
| 106 | + ctype = ContentType.objects.get_for_model(User) |
111 | 107 |
|
112 | 108 | job = EncodeJob() |
113 | 109 | job.id = '1396802241671-jkmme8' |
@@ -139,9 +135,8 @@ def test_transcode_onerror(self): |
139 | 135 | """ |
140 | 136 |
|
141 | 137 | # assume an encode job was submitted |
142 | | - item = Item.objects.create(name='Hello') |
143 | | - |
144 | | - ctype = ContentType.objects.get_for_model(item) |
| 138 | + item = User.objects.create(username='Hello') |
| 139 | + ctype = ContentType.objects.get_for_model(User) |
145 | 140 |
|
146 | 141 | job = EncodeJob() |
147 | 142 | job.id = '1396802241671-jkmme8' |
@@ -173,9 +168,8 @@ def test_transcode_oncomplete(self): |
173 | 168 | """ |
174 | 169 |
|
175 | 170 | # assume an encode job was submitted |
176 | | - item = Item.objects.create(name='Hello') |
177 | | - |
178 | | - ctype = ContentType.objects.get_for_model(item) |
| 171 | + item = User.objects.create(username='Hello') |
| 172 | + ctype = ContentType.objects.get_for_model(User) |
179 | 173 |
|
180 | 174 | job = EncodeJob() |
181 | 175 | job.id = '1396802241671-jkmme8' |
|
0 commit comments