|
| 1 | +frommodel_bakeryimportbaker |
| 2 | + |
| 3 | +fromdjango.urlsimportreverse |
1 | 4 | fromdjango.confimportsettings
|
2 | 5 | frompydotorgimportcontext_processors
|
3 |
| -fromdjango.testimportTestCase |
4 |
| - |
5 |
| - |
6 |
| -classMockRequest: |
7 |
| -def__init__(self,path): |
8 |
| -self.path=path |
9 |
| -super().__init__() |
| 6 | +fromdjango.testimportTestCase,RequestFactory |
| 7 | +fromdjango.contrib.auth.modelsimportAnonymousUser |
10 | 8 |
|
11 | 9 |
|
12 | 10 | classTemplateProcessorsTestCase(TestCase):
|
| 11 | +defsetUp(self): |
| 12 | +self.factory=RequestFactory() |
| 13 | + |
13 | 14 | deftest_url_name(self):
|
14 |
| -mock_request=MockRequest(path='/inner/') |
15 |
| -self.assertEqual({'URL_NAMESPACE':'','URL_NAME':'inner'},context_processors.url_name(mock_request)) |
| 15 | +request=self.factory.get('/inner/') |
| 16 | +self.assertEqual({'URL_NAMESPACE':'','URL_NAME':'inner'},context_processors.url_name(request)) |
16 | 17 |
|
17 |
| -mock_request=MockRequest(path='/events/calendars/') |
18 |
| -self.assertEqual({'URL_NAMESPACE':'events','URL_NAME':'events:calendar_list'},context_processors.url_name(mock_request)) |
| 18 | +request=self.factory.get('/events/calendars/') |
| 19 | +self.assertEqual({'URL_NAMESPACE':'events','URL_NAME':'events:calendar_list'},context_processors.url_name(request)) |
19 | 20 |
|
20 |
| -mock_request=MockRequest(path='/getit-404/releases/3.3.3/not-an-actual-thing/') |
21 |
| -self.assertEqual({},context_processors.url_name(mock_request)) |
| 21 | +request=self.factory.get('/getit-404/releases/3.3.3/not-an-actual-thing/') |
| 22 | +self.assertEqual({},context_processors.url_name(request)) |
22 | 23 |
|
23 |
| -mock_request=MockRequest(path='/getit-404/releases/3.3.3/\r\n/') |
24 |
| -self.assertEqual({},context_processors.url_name(mock_request)) |
| 24 | +request=self.factory.get('/getit-404/releases/3.3.3/\r\n/') |
| 25 | +self.assertEqual({},context_processors.url_name(request)) |
25 | 26 |
|
26 |
| -mock_request=MockRequest(path='/nothing/here/') |
27 |
| -self.assertEqual({},context_processors.url_name(mock_request)) |
| 27 | +request=self.factory.get('/nothing/here/') |
| 28 | +self.assertEqual({},context_processors.url_name(request)) |
28 | 29 |
|
29 | 30 | deftest_blog_url(self):
|
30 |
| -mock_request=MockRequest(path='/about/') |
31 |
| -self.assertEqual({'BLOG_URL':settings.PYTHON_BLOG_URL},context_processors.blog_url(mock_request)) |
| 31 | +request=self.factory.get('/about/') |
| 32 | +self.assertEqual({'BLOG_URL':settings.PYTHON_BLOG_URL},context_processors.blog_url(request)) |
| 33 | + |
| 34 | +deftest_user_nav_bar_links_for_non_psf_members(self): |
| 35 | +request=self.factory.get('/about/') |
| 36 | +request.user=baker.make(settings.AUTH_USER_MODEL,username='foo') |
| 37 | + |
| 38 | +expected_nav= { |
| 39 | +"account": { |
| 40 | +"label":"Your Account", |
| 41 | +"urls": [ |
| 42 | + {"url":reverse("users:user_detail",args=['foo']),"label":"View profile"}, |
| 43 | + {"url":reverse("users:user_profile_edit"),"label":"Edit profile"}, |
| 44 | + {"url":reverse("account_change_password"),"label":"Change password"}, |
| 45 | + ], |
| 46 | + }, |
| 47 | +"psf_membership": { |
| 48 | +"label":"Membership", |
| 49 | +"urls": [ |
| 50 | + {"url":reverse("users:user_nominations_view"),"label":"Nominations"}, |
| 51 | + {"url":reverse("users:user_membership_create"),"label":"Become a PSF member"}, |
| 52 | + ], |
| 53 | + } |
| 54 | + } |
| 55 | + |
| 56 | +self.assertEqual( |
| 57 | + {"USER_NAV_BAR":expected_nav}, |
| 58 | +context_processors.user_nav_bar_links(request) |
| 59 | + ) |
| 60 | + |
| 61 | +deftest_user_nav_bar_links_for_psf_members(self): |
| 62 | +request=self.factory.get('/about/') |
| 63 | +request.user=baker.make(settings.AUTH_USER_MODEL,username='foo') |
| 64 | +baker.make('users.Membership',creator=request.user) |
| 65 | + |
| 66 | +expected_nav= { |
| 67 | +"account": { |
| 68 | +"label":"Your Account", |
| 69 | +"urls": [ |
| 70 | + {"url":reverse("users:user_detail",args=['foo']),"label":"View profile"}, |
| 71 | + {"url":reverse("users:user_profile_edit"),"label":"Edit profile"}, |
| 72 | + {"url":reverse("account_change_password"),"label":"Change password"}, |
| 73 | + ], |
| 74 | + }, |
| 75 | +"psf_membership": { |
| 76 | +"label":"Membership", |
| 77 | +"urls": [ |
| 78 | + {"url":reverse("users:user_nominations_view"),"label":"Nominations"}, |
| 79 | + {"url":reverse("users:user_membership_edit"),"label":"Edit PSF membership"}, |
| 80 | + ], |
| 81 | + } |
| 82 | + } |
| 83 | + |
| 84 | +self.assertEqual( |
| 85 | + {"USER_NAV_BAR":expected_nav}, |
| 86 | +context_processors.user_nav_bar_links(request) |
| 87 | + ) |
| 88 | + |
| 89 | +deftest_user_nav_bar_links_for_anonymous_user(self): |
| 90 | +request=self.factory.get('/about/') |
| 91 | +request.user=AnonymousUser() |
32 | 92 |
|
| 93 | +self.assertEqual({"USER_NAV_BAR": {}},context_processors.user_nav_bar_links(request)) |