import unittest
try:  # python2
    # Must try this first as io also exists in python2
    # but in the wrong one!
    import StringIO as io
except ImportError:  # python3
    import io
import re

from resync.resource import Resource
from resync.resource_dump_manifest import ResourceDumpManifest
from resync.sitemap import SitemapParseError


class TestResourceDumpManifest(unittest.TestCase):

    def test01_as_xml(self):
        rdm = ResourceDumpManifest()
        rdm.add(Resource('a.zip', timestamp=1))
        rdm.add(Resource('b.zip', timestamp=2))
        xml = rdm.as_xml()
        self.assertTrue(re.search(
            r'<rs:md .*capability="resourcedump-manifest"', xml), 'XML has capability')
        self.assertTrue(re.search(
            r'<url><loc>a.zip</loc><lastmod>1970-01-01T00:00:01Z</lastmod></url>', xml), 'XML has resource a')

    def test10_parse(self):
        xml = '<?xml version=\'1.0\' encoding=\'UTF-8\'?>\n\
<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9" xmlns:rs="http://www.openarchives.org/rs/terms/">\
<rs:md at="2013-08-08" capability="resourcedump-manifest"/>\
<url><loc>http://example.com/res1</loc><lastmod>2012-03-14T18:37:36Z</lastmod><rs:md length="12" path="/res1" /></url>\
<url><loc>http://example.com/res2</loc><lastmod>2012-03-14T18:37:36Z</lastmod><rs:md length="32" path="/res2"/></url>\
</urlset>'
        rdm = ResourceDumpManifest()
        rdm.parse(fh=io.StringIO(xml))
        self.assertEqual(len(rdm.resources), 2, 'got 2 resource dumps')
        self.assertEqual(
            rdm.capability, 'resourcedump-manifest', 'capability set')
        self.assertEqual(rdm.md_at, '2013-08-08')
        self.assertTrue('http://example.com/res1' in rdm.resources)
        self.assertTrue(rdm.resources['http://example.com/res1'].length, 12)
        self.assertTrue(rdm.resources['http://example.com/res1'].path, '/res1')
        self.assertTrue('http://example.com/res2' in rdm.resources)
        self.assertTrue(rdm.resources['http://example.com/res2'].path, '/res2')

    def test11_parse_no_capability(self):
        # For a resource dump this should be an error
        xml = '<?xml version=\'1.0\' encoding=\'UTF-8\'?>\n\
<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9" xmlns:rs="http://www.openarchives.org/rs/terms/">\
<rs:md from="2013-01-01"/>\
<url><loc>http://example.com/res1</loc><lastmod>2012-03-14T18:37:36Z</lastmod><rs:md length="12" path="/res1" /></url>\
<url><loc>http://example.com/res2</loc><lastmod>2012-03-14T18:37:36Z</lastmod><rs:md length="32" path="/res2"/></url>\
</urlset>'
        rdm = ResourceDumpManifest()
        self.assertRaises(SitemapParseError, rdm.parse, fh=io.StringIO(xml))

    def test12_parse_bad_capability(self):
        # the <rs:md capability="bad_capability".. should give error
        xml = '<?xml version=\'1.0\' encoding=\'UTF-8\'?>\n\
<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9" xmlns:rs="http://www.openarchives.org/rs/terms/">\
<rs:md capability="bad_capability" from="2013-01-01"/>\
<url><loc>http://example.com/a.zip</loc><lastmod>2012-03-14T18:37:36Z</lastmod><rs:md length="12" /></url>\
</urlset>'
        rdm = ResourceDumpManifest()
        self.assertRaises(SitemapParseError, rdm.parse, fh=io.StringIO(xml))