def setUp(self): from django_pypuppetdb.tastypie_authentication import \ PuppetDBAuthentication self.user = User.objects.create( username='******', email='*****@*****.**') self.auth = PuppetDBAuthentication()
def setUp(self): self.user = User.objects.create( username='******', email='*****@*****.**') self.auth = PuppetDBAuthentication()
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)