from logilab.common.testlib import TestCase, unittest_main

from cubicweb.devtools import TestServerConfiguration

from cubicweb.xy import xy

from cubicweb.spa2rql import Sparql2rqlTranslator

xy.add_equivalence('Project', 'doap:Project')

xy.add_equivalence('Project.creation_date', 'doap:Project.doap:created')

config = TestServerConfiguration('data')

config.bootstrap_cubes()

schema = config.load_schema()

class XYTC(TestCase):

    def setUp(self):

        self.tr = Sparql2rqlTranslator(schema)

    def _test(self, sparql, rql):

        qi = self.tr.translate(sparql)

        self.assertEquals(qi.finalize(), rql)

    def XXX_test_base_01(self):

        self._test('SELECT * WHERE { }', 'Any X')

    def test_base_is(self):

        self._test('''

    PREFIX doap: <http://usefulinc.com/ns/doap#>

    SELECT ?project

    WHERE  {

      ?project a doap:Project;

    }''', 'Any PROJECT WHERE PROJECT is Project')

    def test_base_attr_sel(self):

        self._test('''

    PREFIX doap: <http://usefulinc.com/ns/doap#>

    SELECT ?created

    WHERE  {

      ?project a doap:Project;

              doap:created ?created.

    }''', 'Any CREATED WHERE PROJECT creation_date CREATED, PROJECT is Project')

    def test_base_any_attr_sel(self):

        self._test('''

    PREFIX dc: <http://purl.org/dc/elements/1.1/>

    SELECT ?x ?cd

    WHERE  {

      ?x dc:date ?cd;

    }''', 'Any X, CD WHERE X creation_date CD')

    def test_base_any_attr_sel_amb(self):

        xy.add_equivalence('Version.publication_date', 'doap:Version.dc:date')

        try:

            self._test('''

    PREFIX dc: <http://purl.org/dc/elements/1.1/>

    SELECT ?x ?cd

    WHERE  {

      ?x dc:date ?cd;

    }''', '(Any X, CD WHERE , X creation_date CD) UNION (Any X, CD WHERE , X publication_date CD, X is Version)')

        finally:

            xy.remove_equivalence('Version.publication_date', 'doap:Version.dc:date')

# # Two elements in the group

# PREFIX :  <http://example.org/ns#>

# SELECT *

# { :p :q :r  OPTIONAL { :a :b :c }

#   :p :q :r  OPTIONAL { :a :b :c }

# }

# PREFIX : <http://example.org/ns#>

# SELECT *

# {

#   { ?s ?p ?o } UNION { ?a ?b ?c }

# }

# PREFIX dob: <http://placetime.com/interval/gregorian/1977-01-18T04:00:00Z/P>

# PREFIX time: <http://www.ai.sri.com/daml/ontologies/time/Time.daml#>

# PREFIX dc: <http://purl.org/dc/elements/1.1/>

# SELECT ?desc

# WHERE  {

#   dob:1D a time:ProperInterval;

#          dc:description ?desc.

# }

if __name__ == '__main__':

    unittest_main()