from .testlib 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'])