DM and notification views

bookwyrm/tests/test_views.py

@@ -75,26 +75,6 @@ class Views(TestCase):
         self.assertFalse(views.is_api_request(request))
 
 
-    def test_home_tab(self):
-        ''' there are so many views, this just makes sure it LOADS '''
-        request = self.factory.get('')
-        request.user = self.local_user
-        result = views.home_tab(request, 'local')
-        self.assertIsInstance(result, TemplateResponse)
-        self.assertEqual(result.template_name, 'feed.html')
-        self.assertEqual(result.status_code, 200)
-
-
-    def test_direct_messages_page(self):
-        ''' there are so many views, this just makes sure it LOADS '''
-        request = self.factory.get('')
-        request.user = self.local_user
-        result = views.direct_messages_page(request)
-        self.assertIsInstance(result, TemplateResponse)
-        self.assertEqual(result.template_name, 'direct_messages.html')
-        self.assertEqual(result.status_code, 200)
-
-
     def test_get_activity_feed(self):
         ''' loads statuses '''
         rat = models.User.objects.create_user(
@@ -264,26 +244,6 @@ class Views(TestCase):
         self.assertEqual(result.status_code, 200)
 
 
-    def test_about_page(self):
-        ''' there are so many views, this just makes sure it LOADS '''
-        request = self.factory.get('')
-        request.user = self.local_user
-        result = views.about_page(request)
-        self.assertIsInstance(result, TemplateResponse)
-        self.assertEqual(result.template_name, 'about.html')
-        self.assertEqual(result.status_code, 200)
-
-
-    def test_notifications_page(self):
-        ''' there are so many views, this just makes sure it LOADS '''
-        request = self.factory.get('')
-        request.user = self.local_user
-        result = views.notifications_page(request)
-        self.assertIsInstance(result, TemplateResponse)
-        self.assertEqual(result.template_name, 'notifications.html')
-        self.assertEqual(result.status_code, 200)
-
-
     def test_user_page(self):
         ''' there are so many views, this just makes sure it LOADS '''
         request = self.factory.get('')

bookwyrm/tests/views/test_direct_message.py

@@ -0,0 +1,28 @@
+''' test for app action functionality '''
+from django.template.response import TemplateResponse
+from django.test import TestCase
+from django.test.client import RequestFactory
+
+from bookwyrm import models
+from bookwyrm import views
+
+
+class DirectMessageViews(TestCase):
+    ''' dms '''
+    def setUp(self):
+        ''' we need basic test data and mocks '''
+        self.factory = RequestFactory()
+        self.local_user = models.User.objects.create_user(
+            'mouse@local.com', 'mouse@mouse.mouse', 'password',
+            local=True, localname='mouse')
+
+
+    def test_direct_messages_page(self):
+        ''' there are so many views, this just makes sure it LOADS '''
+        view = views.DirectMessages.as_view()
+        request = self.factory.get('')
+        request.user = self.local_user
+        result = view(request)
+        self.assertIsInstance(result, TemplateResponse)
+        self.assertEqual(result.template_name, 'direct_messages.html')
+        self.assertEqual(result.status_code, 200)

bookwyrm/tests/views/test_notifications.py

@@ -0,0 +1,41 @@
+''' test for app action functionality '''
+from django.template.response import TemplateResponse
+from django.test import TestCase
+from django.test.client import RequestFactory
+
+from bookwyrm import models
+from bookwyrm import views
+
+
+class NotificationViews(TestCase):
+    ''' notifications '''
+    def setUp(self):
+        ''' we need basic test data and mocks '''
+        self.factory = RequestFactory()
+        self.local_user = models.User.objects.create_user(
+            'mouse@local.com', 'mouse@mouse.mouse', 'password',
+            local=True, localname='mouse')
+
+    def test_notifications_page(self):
+        ''' there are so many views, this just makes sure it LOADS '''
+        view = views.Notifications.as_view()
+        request = self.factory.get('')
+        request.user = self.local_user
+        result = view(request)
+        self.assertIsInstance(result, TemplateResponse)
+        self.assertEqual(result.template_name, 'notifications.html')
+        self.assertEqual(result.status_code, 200)
+
+    def test_clear_notifications(self):
+        ''' erase notifications '''
+        models.Notification.objects.create(
+            user=self.local_user, notification_type='MENTION')
+        models.Notification.objects.create(
+            user=self.local_user, notification_type='MENTION', read=True)
+        self.assertEqual(models.Notification.objects.count(), 2)
+        view = views.Notifications.as_view()
+        request = self.factory.post('')
+        request.user = self.local_user
+        result = view(request)
+        self.assertEqual(result.status_code, 302)
+        self.assertEqual(models.Notification.objects.count(), 1)