76 lines
3.6 KiB
Python
76 lines
3.6 KiB
Python
from tests.testcase_with_tmpdir import TestCase
|
|
|
|
import logging
|
|
import os.path
|
|
import re
|
|
import unittest
|
|
from resync.client_utils import init_logging,count_true_args,parse_links,parse_link,parse_capabilities,parse_capability_lists
|
|
from resync.client import ClientFatalError
|
|
|
|
class TestClientUtils(TestCase):
|
|
|
|
def test01_init_logging(self):
|
|
#to_file=False, logfile=None, default_logfile='/tmp/resync.log',
|
|
# human=True, verbose=False, eval_mode=False,
|
|
# default_logger='client', extra_loggers=None):
|
|
tmplog = os.path.join(self.tmpdir,'tmp.log')
|
|
init_logging( to_file=True, default_logfile=tmplog, extra_loggers=['x1','x2'] )
|
|
# check x1 and x2 set, not x3 (can tell by level)
|
|
self.assertTrue( logging.getLogger('x1').level, logging.DEBUG )
|
|
self.assertTrue( logging.getLogger('x2').level, logging.DEBUG )
|
|
self.assertEqual( logging.getLogger('x3').level, 0 )
|
|
# write something, check goes to file
|
|
log = logging.getLogger('resync')
|
|
log.warning('PIGS MIGHT FLY')
|
|
logtxt = open(tmplog,'r').read()
|
|
self.assertTrue( re.search(r'WARNING \| PIGS MIGHT FLY', logtxt) )
|
|
|
|
def test02_count_true_args(self):
|
|
self.assertEqual( count_true_args(), 0 )
|
|
self.assertEqual( count_true_args(True), 1 )
|
|
self.assertEqual( count_true_args(False), 0 )
|
|
self.assertEqual( count_true_args(0,1,2,3), 3 )
|
|
|
|
def test03_parse_links(self):
|
|
self.assertEqual( parse_links( [] ), [] )
|
|
self.assertEqual( parse_links( ['u,h'] ), [{'href': 'h', 'rel': 'u'}] )
|
|
self.assertEqual( parse_links( ['u,h','v,i'] ), [{'href': 'h', 'rel': 'u'},{'href': 'i', 'rel': 'v'}] )
|
|
self.assertRaises( ClientFatalError, parse_links, 'xx' )
|
|
self.assertRaises( ClientFatalError, parse_links, ['u'] )
|
|
self.assertRaises( ClientFatalError, parse_links, ['u,h','u'] )
|
|
|
|
def test04_parse_link(self):
|
|
# Input string of the form: rel,href,att1=val1,att2=val2
|
|
self.assertEqual( parse_link('u,h'), {'href': 'h', 'rel': 'u'} )
|
|
self.assertEqual( parse_link('u,h,a=b'), {'a': 'b', 'href': 'h', 'rel': 'u'} )
|
|
self.assertEqual( parse_link('u,h,a=b,c=d'), {'a': 'b', 'c': 'd', 'href': 'h', 'rel': 'u'} )
|
|
self.assertEqual( parse_link('u,h,a=b,a=d'), {'a': 'd', 'href': 'h', 'rel': 'u'} ) # desired??
|
|
self.assertRaises( ClientFatalError, parse_link, '' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u,' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u,h,,' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u,h,a' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u,h,a=' )
|
|
self.assertRaises( ClientFatalError, parse_link, 'u,h,a=b,=c' )
|
|
|
|
def test05_parse_capabilities(self):
|
|
# Input string of the form: cap_name=uri,cap_name=uri
|
|
# good
|
|
c = parse_capabilities( 'a=' )
|
|
self.assertEqual( len(c), 1 )
|
|
self.assertEqual( c['a'], '' )
|
|
c = parse_capabilities( 'a=b,c=' )
|
|
self.assertEqual( len(c), 2 )
|
|
self.assertEqual( c['a'], 'b' )
|
|
# bad
|
|
self.assertRaises( ClientFatalError, parse_capabilities, 'a' )
|
|
self.assertRaises( ClientFatalError, parse_capabilities, 'a=b,' )
|
|
|
|
def test06_parse_capability_lists(self):
|
|
# Input string of the form: uri,uri
|
|
self.assertEqual( parse_capability_lists('a,b'), ['a','b'] )
|
|
|
|
if __name__ == '__main__':
|
|
suite = unittest.defaultTestLoader.loadTestsFromTestCase(TestClientUtils)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|
