"""functional tests for server'security

:license: GNU Lesser General Public License, v2.1 - http://www.gnu.org/licenses

"""

import sys

from logilab.common.testlib import unittest_main, TestCase

from cubicweb.devtools.apptest import RepositoryBasedTC

from cubicweb import Unauthorized, ValidationError

from cubicweb.server.querier import check_read_access

class BaseSecurityTC(RepositoryBasedTC):

    def setUp(self):

        RepositoryBasedTC.setUp(self)

        self.create_user('iaminusersgrouponly')

        self.readoriggroups = self.schema['Personne'].get_groups('read')

        self.addoriggroups = self.schema['Personne'].get_groups('add')

    def tearDown(self):

        RepositoryBasedTC.tearDown(self)

        self.schema['Personne'].set_groups('read', self.readoriggroups)

        self.schema['Personne'].set_groups('add', self.addoriggroups)

class LowLevelSecurityFunctionTC(BaseSecurityTC):

    def test_check_read_access(self):

        rql = u'Personne U where U nom "managers"'

        rqlst = self.repo.querier._rqlhelper.parse(rql).children[0]

        origgroups = self.schema['Personne'].get_groups('read')

        self.schema['Personne'].set_groups('read', ('users', 'managers'))

        self.repo.querier._rqlhelper.compute_solutions(rqlst)

        solution = rqlst.solutions[0]

        check_read_access(self.schema, self.session.user, rqlst, solution)

        cnx = self.login('anon')

        cu = cnx.cursor()

        self.assertRaises(Unauthorized,

                          check_read_access,

                          self.schema, cnx.user(self.current_session()), rqlst, solution)

        self.assertRaises(Unauthorized, cu.execute, rql)

    def test_upassword_not_selectable(self):

        self.assertRaises(Unauthorized,

                          self.execute, 'Any X,P WHERE X is CWUser, X upassword P')

        self.rollback()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        self.assertRaises(Unauthorized,

                          cu.execute, 'Any X,P WHERE X is CWUser, X upassword P')

class SecurityTC(BaseSecurityTC):

    def setUp(self):

        BaseSecurityTC.setUp(self)

        # implicitly test manager can add some entities

        self.execute("INSERT Affaire X: X sujet 'cool'")

        self.execute("INSERT Societe X: X nom 'logilab'")

        self.execute("INSERT Personne X: X nom 'bidule'")

        self.execute('INSERT CWGroup X: X name "staff"')

        self.commit()

    def test_insert_security(self):

        cnx = self.login('anon')

        cu = cnx.cursor()

        cu.execute("INSERT Personne X: X nom 'bidule'")

        self.assertRaises(Unauthorized, cnx.commit)

        self.assertEquals(cu.execute('Personne X').rowcount, 1)

    def test_insert_rql_permission(self):

        # test user can only add une affaire related to a societe he owns

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("INSERT Affaire X: X sujet 'cool'")

        self.assertRaises(Unauthorized, cnx.commit)

        # test nothing has actually been inserted

        self.restore_connection()

        self.assertEquals(self.execute('Affaire X').rowcount, 1)

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("INSERT Affaire X: X sujet 'cool'")

        cu.execute("INSERT Societe X: X nom 'chouette'")

        cu.execute("SET A concerne S WHERE A sujet 'cool', S nom 'chouette'")

        cnx.commit()

    def test_update_security_1(self):

        cnx = self.login('anon')

        cu = cnx.cursor()

        # local security check

        cu.execute( "SET X nom 'bidulechouette' WHERE X is Personne")

        self.assertRaises(Unauthorized, cnx.commit)

        self.restore_connection()

        self.assertEquals(self.execute('Personne X WHERE X nom "bidulechouette"').rowcount, 0)

    def test_update_security_2(self):

        cnx = self.login('anon')

        cu = cnx.cursor()

        self.repo.schema['Personne'].set_groups('read', ('users', 'managers'))

        self.repo.schema['Personne'].set_groups('add', ('guests', 'users', 'managers'))

        self.assertRaises(Unauthorized, cu.execute, "SET X nom 'bidulechouette' WHERE X is Personne")

        #self.assertRaises(Unauthorized, cnx.commit)

        # test nothing has actually been inserted

        self.restore_connection()

        self.assertEquals(self.execute('Personne X WHERE X nom "bidulechouette"').rowcount, 0)

    def test_update_security_3(self):

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("INSERT Personne X: X nom 'biduuule'")

        cu.execute("INSERT Societe X: X nom 'looogilab'")

        cu.execute("SET X travaille S WHERE X nom 'biduuule', S nom 'looogilab'")

    def test_update_rql_permission(self):

        self.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        self.commit()

        # test user can only update une affaire related to a societe he owns

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("SET X sujet 'pascool' WHERE X is Affaire")

        # this won't actually do anything since the selection query won't return anything

        cnx.commit()

        # to actually get Unauthorized exception, try to update an entity we can read

        cu.execute("SET X nom 'toto' WHERE X is Societe")

        self.assertRaises(Unauthorized, cnx.commit)

        cu.execute("INSERT Affaire X: X sujet 'pascool'")

        cu.execute("INSERT Societe X: X nom 'chouette'")

        cu.execute("SET A concerne S WHERE A sujet 'pascool', S nom 'chouette'")

        cu.execute("SET X sujet 'habahsicestcool' WHERE X sujet 'pascool'")

        cnx.commit()

    def test_delete_security(self):

        # FIXME: sample below fails because we don't detect "owner" can't delete

        # user anyway, and since no user with login == 'bidule' exists, no

        # exception is raised

        #user._groups = {'guests':1}

        #self.assertRaises(Unauthorized,

        #                  self.o.execute, user, "DELETE CWUser X WHERE X login 'bidule'")

        # check local security

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        self.assertRaises(Unauthorized, cu.execute, "DELETE CWGroup Y WHERE Y name 'staff'")

    def test_delete_rql_permission(self):

        self.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        self.commit()

        # test user can only dele une affaire related to a societe he owns

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        # this won't actually do anything since the selection query won't return anything

        cu.execute("DELETE Affaire X")

        cnx.commit()

        # to actually get Unauthorized exception, try to delete an entity we can read

        self.assertRaises(Unauthorized, cu.execute, "DELETE Societe S")

        cu.execute("INSERT Affaire X: X sujet 'pascool'")

        cu.execute("INSERT Societe X: X nom 'chouette'")

        cu.execute("SET A concerne S WHERE A sujet 'pascool', S nom 'chouette'")

        cnx.commit()

##         # this one should fail since it will try to delete two affaires, one authorized

##         # and the other not

##         self.assertRaises(Unauthorized, cu.execute, "DELETE Affaire X")

        cu.execute("DELETE Affaire X WHERE X sujet 'pascool'")

        cnx.commit()

    def test_insert_relation_rql_permission(self):

        cnx = self.login('iaminusersgrouponly')

        session = self.current_session()

        cu = cnx.cursor(session)

        cu.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        # should raise Unauthorized since user don't own S

        # though this won't actually do anything since the selection query won't return anything

        cnx.commit()

        # to actually get Unauthorized exception, try to insert a relation were we can read both entities

        rset = cu.execute('Personne P')

        self.assertEquals(len(rset), 1)

        ent = rset.get_entity(0, 0)

        session.set_pool() # necessary

        self.assertRaises(Unauthorized,

                          ent.e_schema.check_perm, session, 'update', ent.eid)

        self.assertRaises(Unauthorized,

                          cu.execute, "SET P travaille S WHERE P is Personne, S is Societe")

        # test nothing has actually been inserted:

        self.assertEquals(cu.execute('Any P,S WHERE P travaille S,P is Personne, S is Societe').rowcount, 0)

        cu.execute("INSERT Societe X: X nom 'chouette'")

        cu.execute("SET A concerne S WHERE A is Affaire, S nom 'chouette'")

        cnx.commit()

    def test_delete_relation_rql_permission(self):

        self.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        # this won't actually do anything since the selection query won't return anything

        cu.execute("DELETE A concerne S")

        cnx.commit()

        # to actually get Unauthorized exception, try to delete a relation we can read

        self.restore_connection()

        eid = self.execute("INSERT Affaire X: X sujet 'pascool'")[0][0]

        self.execute('SET X owned_by U WHERE X eid %(x)s, U login "iaminusersgrouponly"', {'x': eid}, 'x')

        self.execute("SET A concerne S WHERE A sujet 'pascool', S is Societe")

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        self.assertRaises(Unauthorized, cu.execute, "DELETE A concerne S")

        cu.execute("INSERT Societe X: X nom 'chouette'")

        cu.execute("SET A concerne S WHERE A is Affaire, S nom 'chouette'")

        cnx.commit()

        cu.execute("DELETE A concerne S WHERE S nom 'chouette'")

    def test_user_can_change_its_upassword(self):

        ueid = self.create_user('user')

        cnx = self.login('user')

        cu = cnx.cursor()

        cu.execute('SET X upassword %(passwd)s WHERE X eid %(x)s',

                   {'x': ueid, 'passwd': 'newpwd'}, 'x')

        cnx.commit()

        cnx.close()

        cnx = self.login('user', 'newpwd')

    def test_user_cant_change_other_upassword(self):

        ueid = self.create_user('otheruser')

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute('SET X upassword %(passwd)s WHERE X eid %(x)s',

                   {'x': ueid, 'passwd': 'newpwd'}, 'x')

        self.assertRaises(Unauthorized, cnx.commit)

    # read security test

    def test_read_base(self):

        self.schema['Personne'].set_groups('read', ('users', 'managers'))

        cnx = self.login('anon')

        cu = cnx.cursor()

        self.assertRaises(Unauthorized,

                          cu.execute, 'Personne U where U nom "managers"')

    def test_read_erqlexpr_base(self):

        eid = self.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        rset = cu.execute('Affaire X')

        self.assertEquals(rset.rows, [])

        self.assertRaises(Unauthorized, cu.execute, 'Any X WHERE X eid %(x)s', {'x': eid}, 'x')

        # cache test

        self.assertRaises(Unauthorized, cu.execute, 'Any X WHERE X eid %(x)s', {'x': eid}, 'x')

        aff2 = cu.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        soc1 = cu.execute("INSERT Societe X: X nom 'chouette'")[0][0]

        cu.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        cnx.commit()

        rset = cu.execute('Any X WHERE X eid %(x)s', {'x': aff2}, 'x')

        self.assertEquals(rset.rows, [[aff2]])

        # more cache test w/ NOT eid

        rset = cu.execute('Affaire X WHERE NOT X eid %(x)s', {'x': eid}, 'x')

        self.assertEquals(rset.rows, [[aff2]])

        rset = cu.execute('Affaire X WHERE NOT X eid %(x)s', {'x': aff2}, 'x')

        self.assertEquals(rset.rows, [])

    def test_read_erqlexpr_has_text1(self):

        aff1 = self.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        card1 = self.execute("INSERT Card X: X title 'cool'")[0][0]

        self.execute('SET X owned_by U WHERE X eid %(x)s, U login "iaminusersgrouponly"', {'x': card1}, 'x')

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        aff2 = cu.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        soc1 = cu.execute("INSERT Societe X: X nom 'chouette'")[0][0]

        cu.execute("SET A concerne S WHERE A eid %(a)s, S eid %(s)s", {'a': aff2, 's': soc1},

                   ('a', 's'))

        cnx.commit()

        self.assertRaises(Unauthorized, cu.execute, 'Any X WHERE X eid %(x)s', {'x':aff1}, 'x')

        self.failUnless(cu.execute('Any X WHERE X eid %(x)s', {'x':aff2}, 'x'))

        self.failUnless(cu.execute('Any X WHERE X eid %(x)s', {'x':card1}, 'x'))

        rset = cu.execute("Any X WHERE X has_text 'cool'")

        self.assertEquals(sorted(eid for eid, in rset.rows),

                          [card1, aff2])

    def test_read_erqlexpr_has_text2(self):

        self.execute("INSERT Personne X: X nom 'bidule'")

        self.execute("INSERT Societe X: X nom 'bidule'")

        self.commit()

        self.schema['Personne'].set_groups('read', ('managers',))

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        rset = cu.execute('Any N WHERE N has_text "bidule"')

        self.assertEquals(len(rset.rows), 1, rset.rows)

        rset = cu.execute('Any N WITH N BEING (Any N WHERE N has_text "bidule")')

        self.assertEquals(len(rset.rows), 1, rset.rows)

    def test_read_erqlexpr_optional_rel(self):

        self.execute("INSERT Personne X: X nom 'bidule'")

        self.execute("INSERT Societe X: X nom 'bidule'")

        self.commit()

        self.schema['Personne'].set_groups('read', ('managers',))

        cnx = self.login('anon')

        cu = cnx.cursor()

        rset = cu.execute('Any N,U WHERE N has_text "bidule", N owned_by U?')

        self.assertEquals(len(rset.rows), 1, rset.rows)

    def test_read_erqlexpr_aggregat(self):

        self.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        rset = cu.execute('Any COUNT(X) WHERE X is Affaire')

        self.assertEquals(rset.rows, [[0]])

        aff2 = cu.execute("INSERT Affaire X: X sujet 'cool'")[0][0]

        soc1 = cu.execute("INSERT Societe X: X nom 'chouette'")[0][0]

        cu.execute("SET A concerne S WHERE A is Affaire, S is Societe")

        cnx.commit()

        rset = cu.execute('Any COUNT(X) WHERE X is Affaire')

        self.assertEquals(rset.rows, [[1]])

        rset = cu.execute('Any ETN, COUNT(X) GROUPBY ETN WHERE X is ET, ET name ETN')

        values = dict(rset)

        self.assertEquals(values['Affaire'], 1)

        self.assertEquals(values['Societe'], 2)

        rset = cu.execute('Any ETN, COUNT(X) GROUPBY ETN WHERE X is ET, ET name ETN WITH X BEING ((Affaire X) UNION (Societe X))')

        self.assertEquals(len(rset), 2)

        values = dict(rset)

        self.assertEquals(values['Affaire'], 1)

        self.assertEquals(values['Societe'], 2)

    def test_attribute_security(self):

        # only managers should be able to edit the 'test' attribute of Personne entities

        eid = self.execute("INSERT Personne X: X nom 'bidule', X web 'http://www.debian.org', X test TRUE")[0][0]

        self.commit()

        self.execute('SET X test FALSE WHERE X eid %(x)s', {'x': eid}, 'x')

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("INSERT Personne X: X nom 'bidule', X web 'http://www.debian.org', X test TRUE")

        self.assertRaises(Unauthorized, cnx.commit)

        cu.execute("INSERT Personne X: X nom 'bidule', X web 'http://www.debian.org', X test FALSE")

        self.assertRaises(Unauthorized, cnx.commit)

        eid = cu.execute("INSERT Personne X: X nom 'bidule', X web 'http://www.debian.org'")[0][0]

        cnx.commit()

        cu.execute('SET X test FALSE WHERE X eid %(x)s', {'x': eid}, 'x')

        self.assertRaises(Unauthorized, cnx.commit)

        cu.execute('SET X test TRUE WHERE X eid %(x)s', {'x': eid}, 'x')

        self.assertRaises(Unauthorized, cnx.commit)

        cu.execute('SET X web "http://www.logilab.org" WHERE X eid %(x)s', {'x': eid}, 'x')

        cnx.commit()

        cnx.close()

    def test_attribute_security_rqlexpr(self):

        # Note.para attribute editable by managers or if the note is in "todo" state

        note = self.execute("INSERT Note X: X para 'bidule'").get_entity(0, 0)

        self.commit()

        note.fire_transition('markasdone')

        self.execute('SET X para "truc" WHERE X eid %(x)s', {'x': note.eid}, 'x')

        self.commit()

        cnx = self.login('iaminusersgrouponly')

        cu = cnx.cursor()

        cu.execute("SET X para 'chouette' WHERE X eid %(x)s", {'x': note.eid}, 'x')

        self.assertRaises(Unauthorized, cnx.commit)

        note2 = cu.execute("INSERT Note X: X para 'bidule'").get_entity(0, 0)

        cnx.commit()

        note2.fire_transition('markasdone')