57 lines
3.7 KiB
Python
57 lines
3.7 KiB
Python
import unittest
|
|
import re
|
|
from resync.resource import Resource
|
|
from resync.source_description import SourceDescription
|
|
|
|
|
|
class TestSourceDescription(unittest.TestCase):
|
|
|
|
def test01_empty(self):
|
|
rsd = SourceDescription()
|
|
rsd.describedby = "http://example.org/about"
|
|
self.assertEqual(len(rsd), 0)
|
|
rsd.md_at = None
|
|
self.assertEqual(rsd.as_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:ln href="http://example.org/about" rel="describedby" /><rs:md capability="description" /></urlset>')
|
|
|
|
def test02_add(self):
|
|
rsd = SourceDescription()
|
|
self.assertEqual(len(rsd), 0)
|
|
rsd.add(Resource("http://example.org/u1"))
|
|
rsd.add([Resource("http://example.org/u2"),
|
|
Resource("http://example.org/u3")])
|
|
self.assertEqual(rsd.uris(),
|
|
['http://example.org/u1',
|
|
'http://example.org/u2',
|
|
'http://example.org/u3'])
|
|
|
|
def test03_one_caplist(self):
|
|
rsd = SourceDescription()
|
|
rsd.describedby = "http://example.org/about"
|
|
self.assertEqual(len(rsd), 0)
|
|
rsd.md_at = None
|
|
rsd.add_capability_list("http://example.org/ds1/cl.xml")
|
|
self.assertEqual(len(rsd), 1)
|
|
self.assertEqual(rsd.as_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:ln href="http://example.org/about" rel="describedby" /><rs:md capability="description" /><url><loc>http://example.org/ds1/cl.xml</loc><rs:md capability="capabilitylist" /></url></urlset>')
|
|
|
|
def test04_a_bunch(self):
|
|
rsd = SourceDescription()
|
|
rsd.describedby = "http://example.org/about"
|
|
self.assertEqual(len(rsd), 0)
|
|
rsd.add_capability_list("http://example.org/ds1/cl.xml")
|
|
rsd.add_capability_list("http://example.org/ds2/cl.xml")
|
|
rsd.add_capability_list("http://example.org/ds3/cl.xml")
|
|
self.assertEqual(len(rsd), 3)
|
|
self.assertEqual(rsd.as_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:ln href="http://example.org/about" rel="describedby" /><rs:md capability="description" /><url><loc>http://example.org/ds1/cl.xml</loc><rs:md capability="capabilitylist" /></url><url><loc>http://example.org/ds2/cl.xml</loc><rs:md capability="capabilitylist" /></url><url><loc>http://example.org/ds3/cl.xml</loc><rs:md capability="capabilitylist" /></url></urlset>')
|
|
|
|
def test05_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:ln href="http://example.org/about" rel="describedby" /><rs:md capability="description" /><url><loc>http://example.org/ds1/cl.xml</loc><rs:md capability="capabilitylist" /></url><url><loc>http://example.org/ds2/cl.xml</loc><rs:md capability="capabilitylist" /></url><url><loc>http://example.org/ds3/cl.xml</loc><rs:md capability="capabilitylist" /></url></urlset>'
|
|
sd = SourceDescription()
|
|
sd.parse(str_data=xml)
|
|
self.assertEqual(sd.link_href('describedby'), 'http://example.org/about',
|
|
'describedby link')
|
|
self.assertEqual(sd.capability, 'description')
|
|
self.assertEqual(len(sd.resources), 3, 'got 3 capacility lists')
|
|
[r1, r2, r3] = sd.resources
|
|
self.assertEqual(r1.uri, 'http://example.org/ds1/cl.xml')
|
|
self.assertEqual(r1.capability, 'capabilitylist')
|
