|
bgneal@938
|
1 """Tests for the downloads app signal handlers."""
|
|
bgneal@938
|
2
|
|
bgneal@938
|
3 from django.contrib.auth.models import User
|
|
bgneal@938
|
4 from django.test import TestCase
|
|
bgneal@938
|
5
|
|
bgneal@938
|
6 from mock import Mock
|
|
bgneal@938
|
7
|
|
bgneal@938
|
8 import custom_search.receivers
|
|
bgneal@938
|
9
|
|
bgneal@938
|
10 from downloads.models import Category
|
|
bgneal@938
|
11 from downloads.models import Download
|
|
bgneal@938
|
12 from downloads.models import PendingDownload
|
|
bgneal@938
|
13
|
|
bgneal@938
|
14
|
|
bgneal@938
|
15 class ReceiverTestCase(TestCase):
|
|
bgneal@938
|
16
|
|
bgneal@938
|
17 fixtures = ['downloads_categories.json']
|
|
bgneal@938
|
18
|
|
bgneal@938
|
19 def setUp(self):
|
|
bgneal@938
|
20 self.user = User.objects.create_user('user', 'user@example.com', 'pw')
|
|
bgneal@938
|
21
|
|
bgneal@938
|
22 # Don't let our custom search signal handler class catch any of the
|
|
bgneal@938
|
23 # signals we are throwing here.
|
|
bgneal@938
|
24 custom_search.receivers.signal_processor.teardown()
|
|
bgneal@938
|
25
|
|
bgneal@938
|
26 def tearDown(self):
|
|
bgneal@938
|
27 custom_search.receivers.signal_processor.setup()
|
|
bgneal@938
|
28
|
|
bgneal@938
|
29 def test_download_signal_handlers(self):
|
|
bgneal@938
|
30
|
|
bgneal@938
|
31 category = Category.objects.get(pk=1)
|
|
bgneal@938
|
32 dl = Download(category=category,
|
|
bgneal@938
|
33 title='Title',
|
|
bgneal@938
|
34 description='Cool stuff',
|
|
bgneal@938
|
35 is_public=True,
|
|
bgneal@938
|
36 user=self.user,
|
|
bgneal@938
|
37 ip_address='127.0.0.1')
|
|
bgneal@938
|
38 dl.save()
|
|
bgneal@938
|
39
|
|
bgneal@938
|
40 category = Category.objects.get(pk=1)
|
|
bgneal@938
|
41 self.assertEqual(1, category.count)
|
|
bgneal@938
|
42
|
|
bgneal@938
|
43 category2 = Category.objects.get(pk=4)
|
|
bgneal@938
|
44 dl.category = category2
|
|
bgneal@938
|
45 dl.save()
|
|
bgneal@938
|
46
|
|
bgneal@938
|
47 category = Category.objects.get(pk=1)
|
|
bgneal@938
|
48 self.assertEqual(0, category.count)
|
|
bgneal@938
|
49 category2 = Category.objects.get(pk=4)
|
|
bgneal@938
|
50 self.assertEqual(1, category2.count)
|
|
bgneal@938
|
51
|
|
bgneal@938
|
52 dl.delete()
|
|
bgneal@938
|
53 category2 = Category.objects.get(pk=4)
|
|
bgneal@938
|
54 self.assertEqual(0, category2.count)
|
|
bgneal@938
|
55
|
|
bgneal@938
|
56 def test_pending_download_signal_handlers(self):
|
|
bgneal@938
|
57
|
|
bgneal@938
|
58 category = Category.objects.get(pk=1)
|
|
bgneal@938
|
59 dl = PendingDownload(
|
|
bgneal@938
|
60 category=category,
|
|
bgneal@938
|
61 title='Title',
|
|
bgneal@938
|
62 description='Cool stuff',
|
|
bgneal@938
|
63 user=self.user,
|
|
bgneal@938
|
64 ip_address='127.0.0.1')
|
|
bgneal@938
|
65 dl.save()
|
|
bgneal@938
|
66
|
|
bgneal@938
|
67 dl.file = Mock()
|
|
bgneal@938
|
68 dl.delete()
|
|
bgneal@938
|
69
|
|
bgneal@938
|
70 dl.file.delete.assert_called_with(save=False)
|
|
bgneal@938
|
71
|
