117 lines
5.6 KiB
Python
117 lines
5.6 KiB
Python
import unittest
|
|
import re
|
|
import logging
|
|
import sys, contextlib
|
|
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
|
|
|
|
from resync.client import Client, ClientFatalError
|
|
|
|
# From http://stackoverflow.com/questions/2654834/capturing-stdout-within-the-same-process-in-python
|
|
class Data(object):
|
|
pass
|
|
|
|
@contextlib.contextmanager
|
|
def capture_stdout():
|
|
old = sys.stdout
|
|
capturer = io.StringIO()
|
|
sys.stdout = capturer
|
|
data = Data()
|
|
yield data
|
|
sys.stdout = old
|
|
data.result = capturer.getvalue()
|
|
|
|
|
|
class TestClient(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
def test01_make_resource_list_empty(self):
|
|
c = Client()
|
|
# No mapping is an error
|
|
self.assertRaises( ClientFatalError, c.build_resource_list )
|
|
|
|
def test02_bad_source_uri(self):
|
|
c = Client()
|
|
self.assertRaises( ClientFatalError, c.baseline_or_audit )
|
|
c.set_mappings( ['http://example.org/bbb','/tmp/this_does_not_exist'] )
|
|
self.assertRaises( ClientFatalError, c.baseline_or_audit )
|
|
|
|
def test06_write_capability_list(self):
|
|
c = Client()
|
|
with capture_stdout() as capturer:
|
|
c.write_capability_list( { 'a':'uri_a', 'b':'uri_b' } )
|
|
self.assertTrue( re.search(r'<urlset ',capturer.result) )
|
|
self.assertTrue( re.search(r'<rs:md capability="capabilitylist" />',capturer.result) )
|
|
self.assertTrue( re.search(r'<url><loc>uri_a</loc><rs:md capability="a"',capturer.result) )
|
|
self.assertTrue( re.search(r'<url><loc>uri_b</loc><rs:md capability="b"',capturer.result) )
|
|
|
|
def test07_write_source_description(self):
|
|
c = Client()
|
|
with capture_stdout() as capturer:
|
|
c.write_source_description( [ 'a','b','c' ] )
|
|
#print capturer.result
|
|
self.assertTrue( re.search(r'<urlset ',capturer.result) )
|
|
self.assertTrue( re.search(r'<rs:md capability="description" />',capturer.result) )
|
|
self.assertTrue( re.search(r'<url><loc>a</loc><rs:md capability="capabilitylist" /></url>',capturer.result) )
|
|
self.assertTrue( re.search(r'<url><loc>b</loc><rs:md capability="capabilitylist" /></url>',capturer.result) )
|
|
|
|
def test20_parse_document(self):
|
|
# Key property of the parse_document() method is that it parses the
|
|
# document and identifies its type
|
|
c = Client()
|
|
with capture_stdout() as capturer:
|
|
c.sitemap_name='tests/testdata/examples_from_spec/resourcesync_ex_1.xml'
|
|
c.parse_document()
|
|
self.assertTrue( re.search(r'Parsed resourcelist document with 2 entries',capturer.result) )
|
|
with capture_stdout() as capturer:
|
|
c.sitemap_name='tests/testdata/examples_from_spec/resourcesync_ex_17.xml'
|
|
c.parse_document()
|
|
self.assertTrue( re.search(r'Parsed resourcedump document with 3 entries',capturer.result) )
|
|
with capture_stdout() as capturer:
|
|
c.sitemap_name='tests/testdata/examples_from_spec/resourcesync_ex_19.xml'
|
|
c.parse_document()
|
|
self.assertTrue( re.search(r'Parsed changelist document with 4 entries',capturer.result) )
|
|
with capture_stdout() as capturer:
|
|
c.sitemap_name='tests/testdata/examples_from_spec/resourcesync_ex_22.xml'
|
|
c.parse_document()
|
|
self.assertTrue( re.search(r'Parsed changedump document with 3 entries',capturer.result) )
|
|
|
|
def test40_write_resource_list_mappings(self):
|
|
c = Client()
|
|
c.set_mappings( ['http://example.org/','tests/testdata/'] )
|
|
# with no explicit paths seting the mapping will be used
|
|
with capture_stdout() as capturer:
|
|
c.write_resource_list()
|
|
#sys.stderr.write(capturer.result)
|
|
self.assertTrue( re.search(r'<rs:md at="\S+" capability="resourcelist"', capturer.result ) )
|
|
self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_a</loc>', capturer.result ) )
|
|
self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_b</loc>', capturer.result ) )
|
|
self.assertTrue( re.search(r'<url><loc>http://example.org/dir2/file_x</loc>', capturer.result ) )
|
|
|
|
def test41_write_resource_list_path(self):
|
|
c = Client()
|
|
c.set_mappings( ['http://example.org/','tests/testdata/'] )
|
|
# with an explicit paths setting only the specified paths will be included
|
|
with capture_stdout() as capturer:
|
|
c.write_resource_list(paths='tests/testdata/dir1')
|
|
sys.stderr.write(capturer.result)
|
|
self.assertTrue( re.search(r'<rs:md at="\S+" capability="resourcelist"', capturer.result ) )
|
|
self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_a</loc>', capturer.result ) )
|
|
self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_b</loc>', capturer.result ) )
|
|
self.assertFalse( re.search(r'<url><loc>http://example.org/dir2/file_x</loc>', capturer.result ) )
|
|
# Travis CI does not preserve timestamps from github so test here for the file
|
|
# size but not the datestamp
|
|
#self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_a</loc><lastmod>[\w\-:]+</lastmod><rs:md length="20" /></url>', capturer.result ) )
|
|
#self.assertTrue( re.search(r'<url><loc>http://example.org/dir1/file_b</loc><lastmod>[\w\-:]+</lastmod><rs:md length="45" /></url>', capturer.result ) )
|
|
|
|
if __name__ == '__main__':
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestClient)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|
