test.vieuws.py update

CrzyHAX91 · Nov 26, 2024 · dc5a5fd · dc5a5fd
1 parent 26462d2
commit dc5a5fd
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/dropship_project/test_views.py b/dropship_project/test_views.py
@@ -24,14 +24,22 @@ def test_home_view(self):
         self.assertContains(response, 'Welcome to Dropship V2')
 
     def test_dashboard_view(self):
-        self.client.login(username='testuser', password='testpassword')
+        self.client.force_login(self.user)
         response = self.client.get(reverse('user_dashboard'))
         self.assertEqual(response.status_code, 200)
         self.assertContains(response, 'User Dashboard')
 
     def test_profile_view(self):
-        self.client.login(username='testuser', password='testpassword')
+        self.client.force_login(self.user)
         response = self.client.get(reverse('user_profile'))
         self.assertEqual(response.status_code, 200)
         self.assertContains(response, 'User Profile')
 
+    def test_login_required_dashboard(self):
+        response = self.client.get(reverse('user_dashboard'))
+        self.assertRedirects(response, f'{reverse("account_login")}?next={reverse("user_dashboard")}')
+
+    def test_login_required_profile(self):
+        response = self.client.get(reverse('user_profile'))
+        self.assertRedirects(response, f'{reverse("account_login")}?next={reverse("user_profile")}')
+