def setUp(self):
            from django_pypuppetdb.tastypie_authentication import \
                PuppetDBAuthentication

            self.user = User.objects.create(
                username='******', email='*****@*****.**')
            self.auth = PuppetDBAuthentication()
Пример #2
0
 def setUp(self):
     self.user = User.objects.create(
         username='******', email='*****@*****.**')
     self.auth = PuppetDBAuthentication()
Пример #3
0
    class TestTastypieAuthentication(TestCase):
        def setUp(self):
            self.user = User.objects.create(
                username='******', email='*****@*****.**')
            self.auth = PuppetDBAuthentication()

        def test_authenticate_without_connection(self):
            self.assertIsNone(self.auth.authenticate())

        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'check_user', Mock(return_value=None))
        def test_authenticate_incorrect_username(self):
            self.assertIsNone(self.auth.authenticate('test'))

        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'check_user', Mock(return_value=None))
        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'verify_password', Mock(return_value=True))
        def test_authenticate_incorrect_password(self):
            self.assertIsNone(self.auth.authenticate('test'))

        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'check_user', Mock(
                   return_value=Mock(parameters={'groups': 'a'})
               ))
        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'verify_password', Mock(return_value=True))
        def test_authenticate_with_existing_user(self):
            user = self.auth.authenticate('test', 'password')
            self.assertIsInstance(user, User)
            self.assertEqual(user.username, 'test')
            self.assertFalse(user.is_staff)
            self.assertFalse(user.is_superuser)

        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'check_user', Mock(
                   return_value=Mock(parameters={'groups': 'a'})
               ))
        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'verify_password', Mock(return_value=True))
        def test_authenticate_with_new_user(self):
            user = self.auth.authenticate('new user', 'password')
            self.assertIsInstance(user, User)
            self.assertEqual(user.username, 'new user')
            self.assertIsNotNone(user.api_key)
            self.assertFalse(user.is_staff)
            self.assertFalse(user.is_superuser)

        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'check_user', Mock(
                   return_value=Mock(parameters={'groups': 'admins'})
               ))
        @patch('django_pypuppetdb.user_authentication.UserAuthentication.'
               'verify_password', Mock(return_value=True))
        def test_user_has_admin_rights(self):
            user = self.auth.authenticate('new user', 'password')
            self.assertIsInstance(user, User)
            self.assertEqual(user.username, 'new user')
            self.assertIsNotNone(user.api_key)
            self.assertTrue(user.is_staff)
            self.assertTrue(user.is_superuser)

        def test_get_user_with_incorrect_id(self):
            self.assertIsNone(self.auth.get_user(2))

        def test_get_user(self):
            user = self.auth.get_user(self.user.id)
            self.assertEqual(user, self.user)