from pyramid_cubicweb.tests import PyramidCWTest


class LoginTest(PyramidCWTest):
    def test_login_form(self):
        res = self.webapp.get('/login')
        self.assertIn('__login', res.text)

    def test_login_password_login(self):
        res = self.webapp.post('/login', {
            '__login': self.admlogin, '__password': self.admpassword})
        self.assertEqual(res.status_int, 303)

        res = self.webapp.get('/login')
        self.assertEqual(res.status_int, 303)

    def test_login_password_login_cookie_expires(self):
        res = self.webapp.post('/login', {
            '__login': self.admlogin, '__password': self.admpassword})
        self.assertEqual(res.status_int, 303)
        cookie = self.webapp.cookiejar._cookies[
            'localhost.local']['/']['auth_tkt']
        self.assertIsNone(cookie.expires)

        res = self.webapp.get('/logout')
        self.assertEqual(res.status_int, 303)

        res = self.webapp.post('/login', {
            '__login': self.admlogin, '__password': self.admpassword,
            '__setauthcookie': 1})
        self.assertEqual(res.status_int, 303)
        cookie = self.webapp.cookiejar._cookies[
            'localhost.local']['/']['auth_tkt']
        self.assertIsNotNone(cookie.expires)