# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

#

# This file is part of CubicWeb.

#

# CubicWeb is free software: you can redistribute it and/or modify it under the

# terms of the GNU Lesser General Public License as published by the Free

# Software Foundation, either version 2.1 of the License, or (at your option)

# any later version.

#

# CubicWeb is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

# details.

#

# You should have received a copy of the GNU Lesser General Public License along

# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

from datetime import datetime

from cubicweb.devtools.testlib import CubicWebTC

from cubicweb.sobjects.cwxmlparser import CWEntityXMLParser

orig_parse = CWEntityXMLParser.parse

def parse(self, url):

    try:

        url = RELATEDXML[url.split('?')[0]]

    except KeyError:

        pass

    return orig_parse(self, url)

def setUpModule():

    CWEntityXMLParser.parse = parse

def tearDownModule():

    CWEntityXMLParser.parse = orig_parse

BASEXML = ''.join(u'''

<rset size="1">

 <CWUser eid="5" cwuri="http://pouet.org/5" cwsource="system">

  <login>sthenault</login>

  <upassword>toto</upassword>

  <last_login_time>2011-01-25 14:14:06</last_login_time>

  <creation_date>2010-01-22 10:27:59</creation_date>

  <modification_date>2011-01-25 14:14:06</modification_date>

  <use_email role="subject">

    <EmailAddress cwuri="http://pouet.org/6" eid="6"/>

  </use_email>

  <in_group role="subject">

    <CWGroup cwuri="http://pouet.org/7" eid="7"/>

    <CWGroup cwuri="http://pouet.org/8" eid="8"/>

  </in_group>

  <tags role="object">

    <Tag cwuri="http://pouet.org/9" eid="9"/>

    <Tag cwuri="http://pouet.org/10" eid="10"/>

  </tags>

  <in_state role="subject">

    <State cwuri="http://pouet.org/11" eid="11" name="activated"/>

  </in_state>

 </CWUser>

</rset>

'''.splitlines())

RELATEDXML = {

    'http://pouet.org/6': u'''

<rset size="1">

 <EmailAddress eid="6" cwuri="http://pouet.org/6">

  <address>syt@logilab.fr</address>

  <modification_date>2010-04-13 14:35:56</modification_date>

  <creation_date>2010-04-13 14:35:56</creation_date>

  <tags role="object">

    <Tag cwuri="http://pouet.org/9" eid="9"/>

  </tags>

 </EmailAddress>

</rset>

''',

    'http://pouet.org/7': u'''

<rset size="1">

 <CWGroup eid="7" cwuri="http://pouet.org/7">

  <name>users</name>

  <tags role="object">

    <Tag cwuri="http://pouet.org/9" eid="9"/>

  </tags>

 </CWGroup>

</rset>

''',

    'http://pouet.org/8': u'''

<rset size="1">

 <CWGroup eid="8" cwuri="http://pouet.org/8">

  <name>unknown</name>

 </CWGroup>

</rset>

''',

    'http://pouet.org/9': u'''

<rset size="1">

 <Tag eid="9" cwuri="http://pouet.org/9">

  <name>hop</name>

 </Tag>

</rset>

''',

    'http://pouet.org/10': u'''

<rset size="1">

 <Tag eid="10" cwuri="http://pouet.org/10">

  <name>unknown</name>

 </Tag>

</rset>

''',

    }

OTHERXML = ''.join(u'''

<rset size="1">

 <CWUser eid="5" cwuri="http://pouet.org/5" cwsource="myfeed">

  <login>sthenault</login>

  <upassword>toto</upassword>

  <last_login_time>2011-01-25 14:14:06</last_login_time>

  <creation_date>2010-01-22 10:27:59</creation_date>

  <modification_date>2011-01-25 14:14:06</modification_date>

  <in_group role="subject">

    <CWGroup cwuri="http://pouet.org/7" eid="7"/>

  </in_group>

 </CWUser>

</rset>

'''.splitlines()

)

class CWEntityXMLParserTC(CubicWebTC):

    """/!\ this test use a pre-setup database /!\, if you modify above xml,

    REMOVE THE DATABASE TEMPLATE else it won't be considered

    """

    test_db_id = 'xmlparser'

    @classmethod

    def pre_setup_database(cls, session, config):

        myfeed = session.create_entity('CWSource', name=u'myfeed', type=u'datafeed',

                                   parser=u'cw.entityxml', url=BASEXML)

        myotherfeed = session.create_entity('CWSource', name=u'myotherfeed', type=u'datafeed',

                                            parser=u'cw.entityxml', url=OTHERXML)

        session.commit()

        myfeed.init_mapping([(('CWUser', 'use_email', '*'),

                              u'role=subject\naction=copy'),

                             (('CWUser', 'in_group', '*'),

                              u'role=subject\naction=link\nlinkattr=name'),

                             (('CWUser', 'in_state', '*'),

                              u'role=subject\naction=link\nlinkattr=name'),

                             (('*', 'tags', '*'),

                              u'role=object\naction=link-or-create\nlinkattr=name'),

                            ])

        myotherfeed.init_mapping([(('CWUser', 'in_group', '*'),

                                   u'role=subject\naction=link\nlinkattr=name'),

                                  (('CWUser', 'in_state', '*'),

                                   u'role=subject\naction=link\nlinkattr=name'),

                                  ])

        session.create_entity('Tag', name=u'hop')

    def test_complete_url(self):

        dfsource = self.repo.sources_by_uri['myfeed']

    def test_actions(self):

        dfsource = self.repo.sources_by_uri['myfeed']

        self.assertEqual(dfsource.mapping,

                         {u'CWUser': {

                             (u'in_group', u'subject', u'link'): [

                                 (u'CWGroup', {u'linkattr': u'name'})],

                             (u'in_state', u'subject', u'link'): [

                                 (u'State', {u'linkattr': u'name'})],

                             (u'tags', u'object', u'link-or-create'): [

                                 (u'Tag', {u'linkattr': u'name'})],

                             (u'use_email', u'subject', u'copy'): [

                                 (u'EmailAddress', {})]

                             },

                          u'CWGroup': {

                             (u'tags', u'object', u'link-or-create'): [

                                 (u'Tag', {u'linkattr': u'name'})],

                             },

                          u'EmailAddress': {

                             (u'tags', u'object', u'link-or-create'): [

                                 (u'Tag', {u'linkattr': u'name'})],

                             },

                          })

    def test_external_entity(self):

        dfsource = self.repo.sources_by_uri['myotherfeed']

    def test_noerror_missing_fti_attribute(self):

        dfsource = self.repo.sources_by_uri['myfeed']

<rset size="1">

 <Card eid="50" cwuri="http://pouet.org/50" cwsource="system">

  <title>how-to</title>

 </Card>

</rset>

'''], raise_on_error=True)

    def test_noerror_unspecified_date(self):

        dfsource = self.repo.sources_by_uri['myfeed']

<rset size="1">

 <Card eid="50" cwuri="http://pouet.org/50" cwsource="system">

  <title>how-to</title>

  <content>how-to</content>

  <synopsis>how-to</synopsis>

  <creation_date/>

 </Card>

</rset>

'''], raise_on_error=True)

if __name__ == '__main__':

    from logilab.common.testlib import unittest_main

    unittest_main()