Пример #1
0
    def test_user_menu(self):
        response = testing.get_user_client().get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertNotContains(response, reverse('add_article'), html=False)
        self.assertNotContains(response, reverse('add_event'), html=False)
        self.assertNotContains(response, reverse('add_post'), html=False)
        self.assertNotContains(response, reverse('add_song'), html=False)

        user = testing.create_user(perms=[Article.permstring()])
        response = testing.get_user_client(user).get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertContains(response, reverse('add_article'), html=False)
        self.assertNotContains(response, reverse('add_event'), html=False)
        self.assertNotContains(response, reverse('add_post'), html=False)
        self.assertNotContains(response, reverse('add_song'), html=False)

        user = testing.create_user(perms=[Event.permstring()])
        response = testing.get_user_client(user).get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertNotContains(response, reverse('add_article'), html=False)
        self.assertContains(response, reverse('add_event'), html=False)
        self.assertNotContains(response, reverse('add_post'), html=False)
        self.assertNotContains(response, reverse('add_song'), html=False)

        user = testing.create_user(perms=[Post.permstring()])
        response = testing.get_user_client(user).get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertNotContains(response, reverse('add_article'), html=False)
        self.assertNotContains(response, reverse('add_event'), html=False)
        self.assertContains(response, reverse('add_post'), html=False)
        self.assertNotContains(response, reverse('add_song'), html=False)

        user = testing.create_user(perms=[Song.permstring()])
        response = testing.get_user_client(user).get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertNotContains(response, reverse('add_article'), html=False)
        self.assertNotContains(response, reverse('add_event'), html=False)
        self.assertNotContains(response, reverse('add_post'), html=False)
        self.assertContains(response, reverse('add_song'), html=False)

        user = testing.create_user(perms=[Article.permstring(),
                                          Event.permstring(),
                                          Post.permstring(),
                                          Song.permstring()])
        response = testing.get_user_client(user).get(reverse('index'))
        self.assertEqual(200, response.status_code)
        self.assertNotContains(response, reverse('hello'), html=False)
        self.assertContains(response, reverse('goodbye'), html=False)
        self.assertContains(response, reverse('add_article'), html=False)
        self.assertContains(response, reverse('add_event'), html=False)
        self.assertContains(response, reverse('add_post'), html=False)
        self.assertContains(response, reverse('add_song'), html=False)