#!/usr/bin/python
# -*- coding: utf-8 -*-

"""Test the classic PyGreSQL interface.

Sub-tests for the module functions and constants.

Contributed by Christoph Zwerschke.

These tests do not need a database to test against.
"""

try:
    import unittest2 as unittest  # for Python < 2.7
except ImportError:
    import unittest

import json
import re

import pg  # the module under test

from datetime import timedelta

try:  # noinspection PyUnresolvedReferences
    long
except NameError:  # Python >= 3.0
    long = int

try:  # noinspection PyUnresolvedReferences
    unicode
except NameError:  # Python >= 3.0
    unicode = str


class TestHasConnect(unittest.TestCase):
    """Test existence of basic pg module functions."""

    def testhasPgError(self):
        self.assertTrue(issubclass(pg.Error, Exception))

    def testhasPgWarning(self):
        self.assertTrue(issubclass(pg.Warning, Exception))

    def testhasPgInterfaceError(self):
        self.assertTrue(issubclass(pg.InterfaceError, pg.Error))

    def testhasPgDatabaseError(self):
        self.assertTrue(issubclass(pg.DatabaseError, pg.Error))

    def testhasPgInternalError(self):
        self.assertTrue(issubclass(pg.InternalError, pg.DatabaseError))

    def testhasPgOperationalError(self):
        self.assertTrue(issubclass(pg.OperationalError, pg.DatabaseError))

    def testhasPgProgrammingError(self):
        self.assertTrue(issubclass(pg.ProgrammingError, pg.DatabaseError))

    def testhasPgIntegrityError(self):
        self.assertTrue(issubclass(pg.IntegrityError, pg.DatabaseError))

    def testhasPgDataError(self):
        self.assertTrue(issubclass(pg.DataError, pg.DatabaseError))

    def testhasPgNotSupportedError(self):
        self.assertTrue(issubclass(pg.NotSupportedError, pg.DatabaseError))

    def testhasPgInvalidResultError(self):
        self.assertTrue(issubclass(pg.InvalidResultError, pg.DataError))

    def testhasPgNoResultError(self):
        self.assertTrue(issubclass(pg.NoResultError, pg.InvalidResultError))

    def testhasPgMultipleResultsError(self):
        self.assertTrue(
            issubclass(pg.MultipleResultsError, pg.InvalidResultError))

    def testhasConnect(self):
        self.assertTrue(callable(pg.connect))

    def testhasEscapeString(self):
        self.assertTrue(callable(pg.escape_string))

    def testhasEscapeBytea(self):
        self.assertTrue(callable(pg.escape_bytea))

    def testhasUnescapeBytea(self):
        self.assertTrue(callable(pg.unescape_bytea))

    def testDefHost(self):
        d0 = pg.get_defhost()
        d1 = 'pgtesthost'
        pg.set_defhost(d1)
        self.assertEqual(pg.get_defhost(), d1)
        pg.set_defhost(d0)
        self.assertEqual(pg.get_defhost(), d0)

    def testDefPort(self):
        d0 = pg.get_defport()
        d1 = 1234
        pg.set_defport(d1)
        self.assertEqual(pg.get_defport(), d1)
        if d0 is None:
            d0 = -1
        pg.set_defport(d0)
        if d0 == -1:
            d0 = None
        self.assertEqual(pg.get_defport(), d0)

    def testDefOpt(self):
        d0 = pg.get_defopt()
        d1 = '-h pgtesthost -p 1234'
        pg.set_defopt(d1)
        self.assertEqual(pg.get_defopt(), d1)
        pg.set_defopt(d0)
        self.assertEqual(pg.get_defopt(), d0)

    def testDefBase(self):
        d0 = pg.get_defbase()
        d1 = 'pgtestdb'
        pg.set_defbase(d1)
        self.assertEqual(pg.get_defbase(), d1)
        pg.set_defbase(d0)
        self.assertEqual(pg.get_defbase(), d0)


class TestParseArray(unittest.TestCase):
    """Test the array parser."""

    test_strings = [
        ('', str, ValueError),
        ('{}', None, []),
        ('{}', str, []),
        ('   {   }   ', None, []),
        ('{', str, ValueError),
        ('{{}', str, ValueError),
        ('{}{', str, ValueError),
        ('[]', str, ValueError),
        ('()', str, ValueError),
        ('{[]}', str, ['[]']),
        ('{hello}', int, ValueError),
        ('{42}', int, [42]),
        ('{ 42 }', int, [42]),
        ('{42', int, ValueError),
        ('{ 42 ', int, ValueError),
        ('{hello}', str, ['hello']),
        ('{ hello }', str, ['hello']),
        ('{hi}   ', str, ['hi']),
        ('{hi}   ?', str, ValueError),
        ('{null}', str, [None]),
        (' { NULL } ', str, [None]),
        ('   {   NULL   }   ', str, [None]),
        (' { not null } ', str, ['not null']),
        (' { not NULL } ', str, ['not NULL']),
        (' {"null"} ', str, ['null']),
        (' {"NULL"} ', str, ['NULL']),
        ('{Hi!}', str, ['Hi!']),
        ('{"Hi!"}', str, ['Hi!']),
        ('{" Hi! "}', str, [' Hi! ']),
        ('{a"}', str, ValueError),
        ('{"b}', str, ValueError),
        ('{a"b}', str, ValueError),
        (r'{a\"b}', str, ['a"b']),
        (r'{a\,b}', str, ['a,b']),
        (r'{a\bc}', str, ['abc']),
        (r'{"a\bc"}', str, ['abc']),
        (r'{\a\b\c}', str, ['abc']),
        (r'{"\a\b\c"}', str, ['abc']),
        (r'{"a"b"}', str, ValueError),
        (r'{"a""b"}', str, ValueError),
        (r'{"a\"b"}', str, ['a"b']),
        ('{"{}"}', str, ['{}']),
        (r'{\{\}}', str, ['{}']),
        ('{"{a,b,c}"}', str, ['{a,b,c}']),
        ("{'abc'}", str, ["'abc'"]),
        ('{"abc"}', str, ['abc']),
        (r'{\"abc\"}', str, ['"abc"']),
        (r"{\'abc\'}", str, ["'abc'"]),
        (r"{abc,d,efg}", str, ['abc', 'd', 'efg']),
        ('{Hello World!}', str, ['Hello World!']),
        ('{Hello, World!}', str, ['Hello', 'World!']),
        ('{Hello,\ World!}', str, ['Hello', ' World!']),
        ('{Hello\, World!}', str, ['Hello, World!']),
        ('{"Hello World!"}', str, ['Hello World!']),
        ('{this, should, be, null}', str, ['this', 'should', 'be', None]),
        ('{This, should, be, NULL}', str, ['This', 'should', 'be', None]),
        ('{3, 2, 1, null}', int, [3, 2, 1, None]),
        ('{3, 2, 1, NULL}', int, [3, 2, 1, None]),
        ('{3,17,51}', int, [3, 17, 51]),
        (' { 3 , 17 , 51 } ', int, [3, 17, 51]),
        ('{3,17,51}', str, ['3', '17', '51']),
        (' { 3 , 17 , 51 } ', str, ['3', '17', '51']),
        ('{1,"2",abc,"def"}', str, ['1', '2', 'abc', 'def']),
        ('{{}}', int, [[]]),
        ('{{},{}}', int, [[], []]),
        ('{ {} , {} , {} }', int, [[], [], []]),
        ('{ {} , {} , {} , }', int, ValueError),
        ('{{{1,2,3},{4,5,6}}}', int, [[[1, 2, 3], [4, 5, 6]]]),
        ('{{1,2,3},{4,5,6},{7,8,9}}', int, [[1, 2, 3], [4, 5, 6], [7, 8, 9]]),
        ('{20000, 25000, 25000, 25000}', int, [20000, 25000, 25000, 25000]),
        ('{{{17,18,19},{14,15,16},{11,12,13}},'
         '{{27,28,29},{24,25,26},{21,22,23}},'
         '{{37,38,39},{34,35,36},{31,32,33}}}', int,
            [[[17, 18, 19], [14, 15, 16], [11, 12, 13]],
             [[27, 28, 29], [24, 25, 26], [21, 22, 23]],
             [[37, 38, 39], [34, 35, 36], [31, 32, 33]]]),
        ('{{"breakfast", "consulting"}, {"meeting", "lunch"}}', str,
            [['breakfast', 'consulting'], ['meeting', 'lunch']]),
        ('[1:3]={1,2,3}', int, [1, 2, 3]),
        ('[-1:1]={1,2,3}', int, [1, 2, 3]),
        ('[-1:+1]={1,2,3}', int, [1, 2, 3]),
        ('[-3:-1]={1,2,3}', int, [1, 2, 3]),
        ('[+1:+3]={1,2,3}', int, [1, 2, 3]),
        ('[0:2]={1,2,3}', int, [1, 2, 3]),
        ('[7:9]={1,2,3}', int, [1, 2, 3]),
        ('[]={1,2,3}', int, ValueError),
        ('[1:]={1,2,3}', int, ValueError),
        ('[:3]={1,2,3}', int, ValueError),
        ('[1:1][-2:-1][3:5]={{{1,2,3},{4,5,6}}}',
            int, [[[1, 2, 3], [4, 5, 6]]]),
        ('  [1:1]  [-2:-1]  [3:5]  =  { { { 1 , 2 , 3 }, {4 , 5 , 6 } } }',
            int, [[[1, 2, 3], [4, 5, 6]]]),
        ('[1:1][3:5]={{1,2,3},{4,5,6}}', int, [[1, 2, 3], [4, 5, 6]]),
        ('[3:5]={{1,2,3},{4,5,6}}', int, ValueError),
        ('[1:1][-2:-1][3:5]={{1,2,3},{4,5,6}}', int, ValueError)]

    def testParserParams(self):
        f = pg.cast_array
        self.assertRaises(TypeError, f)
        self.assertRaises(TypeError, f, None)
        self.assertRaises(TypeError, f, '{}', 1)
        self.assertRaises(TypeError, f, '{}', b',',)
        self.assertRaises(TypeError, f, '{}', None, None)
        self.assertRaises(TypeError, f, '{}', None, 1)
        self.assertRaises(TypeError, f, '{}', None, b'')
        self.assertRaises(ValueError, f, '{}', None, b'\\')
        self.assertRaises(ValueError, f, '{}', None, b'{')
        self.assertRaises(ValueError, f, '{}', None, b'}')
        self.assertRaises(TypeError, f, '{}', None, b',;')
        self.assertEqual(f('{}'), [])
        self.assertEqual(f('{}', None), [])
        self.assertEqual(f('{}', None, b';'), [])
        self.assertEqual(f('{}', str), [])
        self.assertEqual(f('{}', str, b';'), [])

    def testParserSimple(self):
        r = pg.cast_array('{a,b,c}')
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 3)
        self.assertEqual(r, ['a', 'b', 'c'])

    def testParserNested(self):
        f = pg.cast_array
        r = f('{{a,b,c}}')
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 1)
        r = r[0]
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 3)
        self.assertEqual(r, ['a', 'b', 'c'])
        self.assertRaises(ValueError, f, '{a,{b,c}}')
        r = f('{{a,b},{c,d}}')
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 2)
        r = r[1]
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 2)
        self.assertEqual(r, ['c', 'd'])
        r = f('{{a},{b},{c}}')
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 3)
        r = r[1]
        self.assertIsInstance(r, list)
        self.assertEqual(len(r), 1)
        self.assertEqual(r[0], 'b')
        r = f('{{{{{{{abc}}}}}}}')
        for i in range(7):
            self.assertIsInstance(r, list)
            self.assertEqual(len(r), 1)
            r = r[0]
        self.assertEqual(r, 'abc')

    def testParserTooDeeplyNested(self):
        f = pg.cast_array
        for n in 3, 5, 9, 12, 16, 32, 64, 256:
            r = '%sa,b,c%s' % ('{' * n, '}' * n)
            if n > 16:  # hard coded maximum depth
                self.assertRaises(ValueError, f, r)
            else:
                r = f(r)
                for i in range(n - 1):
                    self.assertIsInstance(r, list)
                    self.assertEqual(len(r), 1)
                    r = r[0]
                self.assertEqual(len(r), 3)
                self.assertEqual(r, ['a', 'b', 'c'])

    def testParserCast(self):
        f = pg.cast_array
        self.assertEqual(f('{1}'), ['1'])
        self.assertEqual(f('{1}', None), ['1'])
        self.assertEqual(f('{1}', int), [1])
        self.assertEqual(f('{1}', str), ['1'])
        self.assertEqual(f('{a}'), ['a'])
        self.assertEqual(f('{a}', None), ['a'])
        self.assertRaises(ValueError, f, '{a}', int)
        self.assertEqual(f('{a}', str), ['a'])
        cast = lambda s: '%s is ok' % s
        self.assertEqual(f('{a}', cast), ['a is ok'])

    def testParserDelim(self):
        f = pg.cast_array
        self.assertEqual(f('{1,2}'), ['1', '2'])
        self.assertEqual(f('{1,2}', delim=b','), ['1', '2'])
        self.assertEqual(f('{1;2}'), ['1;2'])
        self.assertEqual(f('{1;2}', delim=b';'), ['1', '2'])
        self.assertEqual(f('{1,2}', delim=b';'), ['1,2'])

    def testParserWithData(self):
        f = pg.cast_array
        for string, cast, expected in self.test_strings:
            if expected is ValueError:
                self.assertRaises(ValueError, f, string, cast)
            else:
                self.assertEqual(f(string, cast), expected)

    def testParserWithoutCast(self):
        f = pg.cast_array

        for string, cast, expected in self.test_strings:
            if cast is not str:
                continue
            if expected is ValueError:
                self.assertRaises(ValueError, f, string)
            else:
                self.assertEqual(f(string), expected)

    def testParserWithDifferentDelimiter(self):
        f = pg.cast_array

        def replace_comma(value):
            if isinstance(value, str):
                return value.replace(',', ';')
            elif isinstance(value, list):
                return [replace_comma(v) for v in value]
            else:
                return value

        for string, cast, expected in self.test_strings:
            string = replace_comma(string)
            if expected is ValueError:
                self.assertRaises(ValueError, f, string, cast)
            else:
                expected = replace_comma(expected)
                self.assertEqual(f(string, cast, b';'), expected)


class TestParseRecord(unittest.TestCase):
    """Test the record parser."""

    test_strings = [
        ('', None, ValueError),
        ('', str, ValueError),
        ('(', None, ValueError),
        ('(', str, ValueError),
        ('()', None, (None,)),
        ('()', str, (None,)),
        ('()', int, (None,)),
        ('(,)', str, (None, None)),
        ('( , )', str, (' ', ' ')),
        ('(")', None, ValueError),
        ('("")', None, ('',)),
        ('("")', str, ('',)),
        ('("")', int, ValueError),
        ('("" )', None, (' ',)),
        ('("" )', str, (' ',)),
        ('("" )', int, ValueError),
        ('    ()    ', None, (None,)),
        ('   (   )   ', None, ('   ',)),
        ('(', str, ValueError),
        ('(()', str, ('(',)),
        ('(())', str, ValueError),
        ('()(', str, ValueError),
        ('()()', str, ValueError),
        ('[]', str, ValueError),
        ('{}', str, ValueError),
        ('([])', str, ('[]',)),
        ('(hello)', int, ValueError),
        ('(42)', int, (42,)),
        ('( 42 )', int, (42,)),
        ('(  42)', int, (42,)),
        ('(42)', str, ('42',)),
        ('( 42 )', str, (' 42 ',)),
        ('(  42)', str, ('  42',)),
        ('(42', int, ValueError),
        ('( 42 ', int, ValueError),
        ('(hello)', str, ('hello',)),
        ('( hello )', str, (' hello ',)),
        ('(hello))', str, ValueError),
        ('   (hello)   ', str, ('hello',)),
        ('   (hello)   )', str, ValueError),
        ('(hello)?', str, ValueError),
        ('(null)', str, ('null',)),
        ('(null)', int, ValueError),
        (' ( NULL ) ', str, (' NULL ',)),
        ('   (   NULL   )   ', str, ('   NULL   ',)),
        (' ( null null ) ', str, (' null null ',)),
        (' ("null") ', str, ('null',)),
        (' ("NULL") ', str, ('NULL',)),
        ('(Hi!)', str, ('Hi!',)),
        ('("Hi!")', str, ('Hi!',)),
        ("('Hi!')", str, ("'Hi!'",)),
        ('(" Hi! ")', str, (' Hi! ',)),
        ('("Hi!" )', str, ('Hi! ',)),
        ('( "Hi!")', str, (' Hi!',)),
        ('( "Hi!" )', str, (' Hi! ',)),
        ('( ""Hi!"" )', str, (' Hi! ',)),
        ('( """Hi!""" )', str, (' "Hi!" ',)),
        ('(a")', str, ValueError),
        ('("b)', str, ValueError),
        ('("a" "b)', str, ValueError),
        ('("a" "b")', str, ('a b',)),
        ('( "a" "b" "c" )', str, (' a b c ',)),
        ('(  "a"  "b"  "c"  )', str, ('  a  b  c  ',)),
        ('(  "a,b"  "c,d"  )', str, ('  a,b  c,d  ',)),
        ('( "(a,b,c)" d, e, "f,g")', str, (' (a,b,c) d', ' e', ' f,g')),
        ('(a",b,c",d,"e,f")', str, ('a,b,c', 'd', 'e,f')),
        ('( """a,b""", ""c,d"", "e,f", "g", ""h"", """i""")', str,
            (' "a,b"', ' c', 'd', ' e,f', ' g', ' h', ' "i"')),
        ('(a",b)",c"),(d,e)",f,g)', str, ('a,b)', 'c),(d,e)', 'f', 'g')),
        ('(a"b)', str, ValueError),
        (r'(a\"b)', str, ('a"b',)),
        ('(a""b)', str, ('ab',)),
        ('("a""b")', str, ('a"b',)),
        (r'(a\,b)', str, ('a,b',)),
        (r'(a\bc)', str, ('abc',)),
        (r'("a\bc")', str, ('abc',)),
        (r'(\a\b\c)', str, ('abc',)),
        (r'("\a\b\c")', str, ('abc',)),
        ('("()")', str, ('()',)),
        (r'(\,)', str, (',',)),
        (r'(\(\))', str, ('()',)),
        (r'(\)\()', str, (')(',)),
        ('("(a,b,c)")', str, ('(a,b,c)',)),
        ("('abc')", str, ("'abc'",)),
        ('("abc")', str, ('abc',)),
        (r'(\"abc\")', str, ('"abc"',)),
        (r"(\'abc\')", str, ("'abc'",)),
        ('(Hello World!)', str, ('Hello World!',)),
        ('(Hello, World!)', str, ('Hello', ' World!',)),
        ('(Hello,\ World!)', str, ('Hello', ' World!',)),
        ('(Hello\, World!)', str, ('Hello, World!',)),
        ('("Hello World!")', str, ('Hello World!',)),
        ("(this,shouldn't,be,null)", str, ('this', "shouldn't", 'be', 'null')),
        ('(null,should,be,)', str, ('null', 'should', 'be', None)),
        ('(abcABC0123!?+-*/=&%$\\\\\'\\"{[]}"""":;\\,,)', str,
            ('abcABC0123!?+-*/=&%$\\\'"{[]}":;,', None)),
        ('(3, 2, 1,)', int, (3, 2, 1, None)),
        ('(3, 2, 1, )', int, ValueError),
        ('(, 1, 2, 3)', int, (None, 1, 2, 3)),
        ('( , 1, 2, 3)', int, ValueError),
        ('(,1,,2,,3,)', int, (None, 1, None, 2, None, 3, None)),
        ('(3,17,51)', int, (3, 17, 51)),
        (' ( 3 , 17 , 51 ) ', int, (3, 17, 51)),
        ('(3,17,51)', str, ('3', '17', '51')),
        (' ( 3 , 17 , 51 ) ', str, (' 3 ', ' 17 ', ' 51 ')),
        ('(1,"2",abc,"def")', str, ('1', '2', 'abc', 'def')),
        ('(())', str, ValueError),
        ('()))', str, ValueError),
        ('()()', str, ValueError),
        ('((()', str, ('((',)),
        ('(())', int, ValueError),
        ('((),())', str, ValueError),
        ('("()","()")', str, ('()', '()')),
        ('( " () , () , () " )', str, ('  () , () , ()  ',)),
        ('(20000, 25000, 25000, 25000)', int, (20000, 25000, 25000, 25000)),
        ('("breakfast","consulting","meeting","lunch")', str,
            ('breakfast', 'consulting', 'meeting', 'lunch')),
        ('("breakfast","consulting","meeting","lunch")',
            (str, str, str), ValueError),
        ('("breakfast","consulting","meeting","lunch")', (str, str, str, str),
            ('breakfast', 'consulting', 'meeting', 'lunch')),
        ('("breakfast","consulting","meeting","lunch")',
            (str, str, str, str, str), ValueError),
        ('("fuzzy dice",42,1.9375)', None, ('fuzzy dice', '42', '1.9375')),
        ('("fuzzy dice",42,1.9375)', str, ('fuzzy dice', '42', '1.9375')),
        ('("fuzzy dice",42,1.9375)', int, ValueError),
        ('("fuzzy dice",42,1.9375)', (str, int, float),
            ('fuzzy dice', 42, 1.9375)),
        ('("fuzzy dice",42,1.9375)', (str, int), ValueError),
        ('("fuzzy dice",42,1.9375)', (str, int, float, str), ValueError),
        ('("fuzzy dice",42,)', (str, int, float), ('fuzzy dice', 42, None)),
        ('("fuzzy dice",42,)', (str, int), ValueError),
        ('("",42,)', (str, int, float), ('', 42, None)),
        ('("fuzzy dice","",1.9375)', (str, int, float), ValueError),
        ('(fuzzy dice,"42","1.9375")', (str, int, float),
            ('fuzzy dice', 42, 1.9375))]

    def testParserParams(self):
        f = pg.cast_record
        self.assertRaises(TypeError, f)
        self.assertRaises(TypeError, f, None)
        self.assertRaises(TypeError, f, '()', 1)
        self.assertRaises(TypeError, f, '()', b',',)
        self.assertRaises(TypeError, f, '()', None, None)
        self.assertRaises(TypeError, f, '()', None, 1)
        self.assertRaises(TypeError, f, '()', None, b'')
        self.assertRaises(ValueError, f, '()', None, b'\\')
        self.assertRaises(ValueError, f, '()', None, b'(')
        self.assertRaises(ValueError, f, '()', None, b')')
        self.assertRaises(TypeError, f, '{}', None, b',;')
        self.assertEqual(f('()'), (None,))
        self.assertEqual(f('()', None), (None,))
        self.assertEqual(f('()', None, b';'), (None,))
        self.assertEqual(f('()', str), (None,))
        self.assertEqual(f('()', str, b';'), (None,))

    def testParserSimple(self):
        r = pg.cast_record('(a,b,c)')
        self.assertIsInstance(r, tuple)
        self.assertEqual(len(r), 3)
        self.assertEqual(r, ('a', 'b', 'c'))

    def testParserNested(self):
        f = pg.cast_record
        self.assertRaises(ValueError, f, '((a,b,c))')
        self.assertRaises(ValueError, f, '((a,b),(c,d))')
        self.assertRaises(ValueError, f, '((a),(b),(c))')
        self.assertRaises(ValueError, f, '(((((((abc)))))))')

    def testParserManyElements(self):
        f = pg.cast_record
        for n in 3, 5, 9, 12, 16, 32, 64, 256:
            r = '(%s)' % ','.join(map(str, range(n)))
            r = f(r, int)
            self.assertEqual(r, tuple(range(n)))

    def testParserCastUniform(self):
        f = pg.cast_record
        self.assertEqual(f('(1)'), ('1',))
        self.assertEqual(f('(1)', None), ('1',))
        self.assertEqual(f('(1)', int), (1,))
        self.assertEqual(f('(1)', str), ('1',))
        self.assertEqual(f('(a)'), ('a',))
        self.assertEqual(f('(a)', None), ('a',))
        self.assertRaises(ValueError, f, '(a)', int)
        self.assertEqual(f('(a)', str), ('a',))
        cast = lambda s: '%s is ok' % s
        self.assertEqual(f('(a)', cast), ('a is ok',))

    def testParserCastNonUniform(self):
        f = pg.cast_record
        self.assertEqual(f('(1)', []), ('1',))
        self.assertEqual(f('(1)', [None]), ('1',))
        self.assertEqual(f('(1)', [str]), ('1',))
        self.assertEqual(f('(1)', [int]), (1,))
        self.assertRaises(ValueError, f, '(1)', [None, None])
        self.assertRaises(ValueError, f, '(1)', [str, str])
        self.assertRaises(ValueError, f, '(1)', [int, int])
        self.assertEqual(f('(a)', [None]), ('a',))
        self.assertEqual(f('(a)', [str]), ('a',))
        self.assertRaises(ValueError, f, '(a)', [int])
        self.assertEqual(f('(1,a)', [int, str]), (1, 'a'))
        self.assertRaises(ValueError, f, '(1,a)', [str, int])
        self.assertEqual(f('(a,1)', [str, int]), ('a', 1))
        self.assertRaises(ValueError, f, '(a,1)', [int, str])
        self.assertEqual(f('(1,a,2,b,3,c)',
            [int, str, int, str, int, str]), (1, 'a', 2, 'b', 3, 'c'))
        self.assertEqual(f('(1,a,2,b,3,c)',
            (int, str, int, str, int, str)), (1, 'a', 2, 'b', 3, 'c'))
        cast1 = lambda s: '%s is ok' % s
        self.assertEqual(f('(a)', [cast1]), ('a is ok',))
        cast2 = lambda s: 'and %s is ok, too' % s
        self.assertEqual(f('(a,b)', [cast1, cast2]),
            ('a is ok', 'and b is ok, too'))
        self.assertRaises(ValueError, f, '(a)', [cast1, cast2])
        self.assertRaises(ValueError, f, '(a,b,c)', [cast1, cast2])
        self.assertEqual(f('(1,2,3,4,5,6)',
            [int, float, str, None, cast1, cast2]),
            (1, 2.0, '3', '4', '5 is ok', 'and 6 is ok, too'))

    def testParserDelim(self):
        f = pg.cast_record
        self.assertEqual(f('(1,2)'), ('1', '2'))
        self.assertEqual(f('(1,2)', delim=b','), ('1', '2'))
        self.assertEqual(f('(1;2)'), ('1;2',))
        self.assertEqual(f('(1;2)', delim=b';'), ('1', '2'))
        self.assertEqual(f('(1,2)', delim=b';'), ('1,2',))

    def testParserWithData(self):
        f = pg.cast_record
        for string, cast, expected in self.test_strings:
            if expected is ValueError:
                self.assertRaises(ValueError, f, string, cast)
            else:
                self.assertEqual(f(string, cast), expected)

    def testParserWithoutCast(self):
        f = pg.cast_record

        for string, cast, expected in self.test_strings:
            if cast is not str:
                continue
            if expected is ValueError:
                self.assertRaises(ValueError, f, string)
            else:
                self.assertEqual(f(string), expected)

    def testParserWithDifferentDelimiter(self):
        f = pg.cast_record

        def replace_comma(value):
            if isinstance(value, str):
                return value.replace(';', '@').replace(
                    ',', ';').replace('@', ',')
            elif isinstance(value, tuple):
                return tuple(replace_comma(v) for v in value)
            else:
                return value

        for string, cast, expected in self.test_strings:
            string = replace_comma(string)
            if expected is ValueError:
                self.assertRaises(ValueError, f, string, cast)
            else:
                expected = replace_comma(expected)
                self.assertEqual(f(string, cast, b';'), expected)


class TestParseHStore(unittest.TestCase):
    """Test the hstore parser."""

    test_strings = [
        ('', {}),
        ('=>', ValueError),
        ('""=>', ValueError),
        ('=>""', ValueError),
        ('""=>""', {'': ''}),
        ('NULL=>NULL', {'NULL': None}),
        ('null=>null', {'null': None}),
        ('NULL=>"NULL"', {'NULL': 'NULL'}),
        ('null=>"null"', {'null': 'null'}),
        ('k', ValueError),
        ('k,', ValueError),
        ('k=', ValueError),
        ('k=>', ValueError),
        ('k=>v', {'k': 'v'}),
        ('k=>v,', ValueError),
        (' k => v ', {'k': 'v'}),
        ('   k   =>   v   ', {'k': 'v'}),
        ('" k " => " v "', {' k ': ' v '}),
        ('"k=>v', ValueError),
        ('k=>"v', ValueError),
        ('"1-a" => "anything at all"', {'1-a': 'anything at all'}),
        ('k => v, foo => bar, baz => whatever,'
                ' "1-a" => "anything at all"',
            {'k': 'v', 'foo': 'bar', 'baz': 'whatever',
            '1-a': 'anything at all'}),
        ('"Hello, World!"=>"Hi!"', {'Hello, World!': 'Hi!'}),
        ('"Hi!"=>"Hello, World!"', {'Hi!': 'Hello, World!'}),
        ('"k=>v"=>k\=\>v', {'k=>v': 'k=>v'}),
        ('k\=\>v=>"k=>v"', {'k=>v': 'k=>v'}),
        ('a\\,b=>a,b=>a', {'a,b': 'a', 'b': 'a'})]

    def testParser(self):
        f = pg.cast_hstore

        self.assertRaises(TypeError, f)
        self.assertRaises(TypeError, f, None)
        self.assertRaises(TypeError, f, 42)
        self.assertRaises(TypeError, f, '', None)

        for string, expected in self.test_strings:
            if expected is ValueError:
                self.assertRaises(ValueError, f, string)
            else:
                self.assertEqual(f(string), expected)


class TestCastInterval(unittest.TestCase):
    """Test the interval typecast function."""

    intervals = [
        ((0, 0, 0, 1, 0, 0, 0),
            ('1:00:00', '01:00:00', '@ 1 hour', 'PT1H')),
        ((0, 0, 0, -1, 0, 0, 0),
            ('-1:00:00', '-01:00:00', '@ -1 hour', 'PT-1H')),
        ((0, 0, 0, 1, 0, 0, 0),
            ('0-0 0 1:00:00', '0 years 0 mons 0 days 01:00:00',
            '@ 0 years 0 mons 0 days 1 hour', 'P0Y0M0DT1H')),
        ((0, 0, 0, -1, 0, 0, 0),
            ('-0-0 -1:00:00', '0 years 0 mons 0 days -01:00:00',
            '@ 0 years 0 mons 0 days -1 hour', 'P0Y0M0DT-1H')),
        ((0, 0, 1, 0, 0, 0, 0),
            ('1 0:00:00', '1 day', '@ 1 day', 'P1D')),
        ((0, 0, -1, 0, 0, 0, 0),
            ('-1 0:00:00', '-1 day', '@ -1 day', 'P-1D')),
        ((0, 1, 0, 0, 0, 0, 0),
            ('0-1', '1 mon', '@ 1 mon', 'P1M')),
        ((1, 0, 0, 0, 0, 0, 0),
            ('1-0', '1 year', '@ 1 year', 'P1Y')),
        ((0, 0, 0, 2, 0, 0, 0),
            ('2:00:00', '02:00:00', '@ 2 hours', 'PT2H')),
        ((0, 0, 2, 0, 0, 0, 0),
            ('2 0:00:00', '2 days', '@ 2 days', 'P2D')),
        ((0, 2, 0, 0, 0, 0, 0),
            ('0-2', '2 mons', '@ 2 mons', 'P2M')),
        ((2, 0, 0, 0, 0, 0, 0),
            ('2-0', '2 years', '@ 2 years', 'P2Y')),
        ((0, 0, 0, -3, 0, 0, 0),
            ('-3:00:00', '-03:00:00', '@ 3 hours ago', 'PT-3H')),
        ((0, 0, -3, 0, 0, 0, 0),
            ('-3 0:00:00', '-3 days', '@ 3 days ago', 'P-3D')),
        ((0, -3, 0, 0, 0, 0, 0),
            ('-0-3', '-3 mons', '@ 3 mons ago', 'P-3M')),
        ((-3, 0, 0, 0, 0, 0, 0),
            ('-3-0', '-3 years', '@ 3 years ago', 'P-3Y')),
        ((0, 0, 0, 0, 1, 0, 0),
            ('0:01:00', '00:01:00', '@ 1 min', 'PT1M')),
        ((0, 0, 0, 0, 0, 1, 0),
            ('0:00:01', '00:00:01', '@ 1 sec', 'PT1S')),
        ((0, 0, 0, 0, 0, 0, 1),
            ('0:00:00.000001', '00:00:00.000001',
             '@ 0.000001 secs', 'PT0.000001S')),
        ((0, 0, 0, 0, 2, 0, 0),
            ('0:02:00', '00:02:00', '@ 2 mins', 'PT2M')),
        ((0, 0, 0, 0, 0, 2, 0),
            ('0:00:02', '00:00:02', '@ 2 secs', 'PT2S')),
        ((0, 0, 0, 0, 0, 0, 2),
            ('0:00:00.000002', '00:00:00.000002',
             '@ 0.000002 secs', 'PT0.000002S')),
        ((0, 0, 0, 0, -3, 0, 0),
            ('-0:03:00', '-00:03:00', '@ 3 mins ago', 'PT-3M')),
        ((0, 0, 0, 0, 0, -3, 0),
            ('-0:00:03', '-00:00:03', '@ 3 secs ago', 'PT-3S')),
        ((0, 0, 0, 0, 0, 0, -3),
            ('-0:00:00.000003', '-00:00:00.000003',
             '@ 0.000003 secs ago', 'PT-0.000003S')),
        ((1, 2, 0, 0, 0, 0, 0),
            ('1-2', '1 year 2 mons', '@ 1 year 2 mons', 'P1Y2M')),
        ((0, 0, 3, 4, 5, 6, 0),
            ('3 4:05:06', '3 days 04:05:06',
             '@ 3 days 4 hours 5 mins 6 secs', 'P3DT4H5M6S')),
        ((1, 2, 3, 4, 5, 6, 0),
            ('+1-2 +3 +4:05:06', '1 year 2 mons 3 days 04:05:06',
             '@ 1 year 2 mons 3 days 4 hours 5 mins 6 secs',
             'P1Y2M3DT4H5M6S')),
        ((1, 2, 3, -4, -5, -6, 0),
            ('+1-2 +3 -4:05:06', '1 year 2 mons 3 days -04:05:06',
             '@ 1 year 2 mons 3 days -4 hours -5 mins -6 secs',
             'P1Y2M3DT-4H-5M-6S')),
        ((1, 2, 3, -4, 5, 6, 0),
            ('+1-2 +3 -3:54:54', '1 year 2 mons 3 days -03:54:54',
             '@ 1 year 2 mons 3 days -3 hours -54 mins -54 secs',
             'P1Y2M3DT-3H-54M-54S')),
        ((-1, -2, 3, -4, -5, -6, 0),
            ('-1-2 +3 -4:05:06', '-1 years -2 mons +3 days -04:05:06',
             '@ 1 year 2 mons -3 days 4 hours 5 mins 6 secs ago',
             'P-1Y-2M3DT-4H-5M-6S')),
        ((1, 2, -3, 4, 5, 6, 0),
            ('+1-2 -3 +4:05:06', '1 year 2 mons -3 days +04:05:06',
             '@ 1 year 2 mons -3 days 4 hours 5 mins 6 secs',
             'P1Y2M-3DT4H5M6S')),
        ((0, 0, 0, 1, 30, 0, 0),
            ('1:30:00', '01:30:00', '@ 1 hour 30 mins', 'PT1H30M')),
        ((0, 0, 0, 3, 15, 45, 123456),
            ('3:15:45.123456', '03:15:45.123456',
             '@ 3 hours 15 mins 45.123456 secs', 'PT3H15M45.123456S')),
        ((0, 0, 0, 3, 15, -5, 123),
            ('3:14:55.000123', '03:14:55.000123',
             '@ 3 hours 14 mins 55.000123 secs', 'PT3H14M55.000123S')),
        ((0, 0, 0, 3, -5, 15, -12345),
            ('2:55:14.987655', '02:55:14.987655',
             '@ 2 hours 55 mins 14.987655 secs', 'PT2H55M14.987655S')),
        ((0, 0, 0, 2, -1, 0, 0),
            ('1:59:00', '01:59:00', '@ 1 hour 59 mins', 'PT1H59M')),
        ((0, 0, 0, -1, 2, 0, 0),
            ('-0:58:00', '-00:58:00', '@ 58 mins ago', 'PT-58M')),
        ((1, 11, 0, 0, 0, 0, 0),
            ('1-11', '1 year 11 mons', '@ 1 year 11 mons', 'P1Y11M')),
        ((0, -10, 0, 0, 0, 0, 0),
            ('-0-10', '-10 mons', '@ 10 mons ago', 'P-10M')),
        ((0, 0, 2, -1, 0, 0, 0),
            ('+0-0 +2 -1:00:00', '2 days -01:00:00',
             '@ 2 days -1 hours', 'P2DT-1H')),
        ((0, 0, -1, 2, 0, 0, 0),
            ('+0-0 -1 +2:00:00', '-1 days +02:00:00',
             '@ 1 day -2 hours ago', 'P-1DT2H')),
        ((0, 0, 1, 0, 0, 0, 1),
            ('1 0:00:00.000001', '1 day 00:00:00.000001',
             '@ 1 day 0.000001 secs', 'P1DT0.000001S')),
        ((0, 0, 1, 0, 0, 1, 0),
            ('1 0:00:01', '1 day 00:00:01', '@ 1 day 1 sec', 'P1DT1S')),
        ((0, 0, 1, 0, 1, 0, 0),
            ('1 0:01:00', '1 day 00:01:00', '@ 1 day 1 min', 'P1DT1M')),
        ((0, 0, 0, 0, 1, 0, -1),
            ('0:00:59.999999', '00:00:59.999999',
             '@ 59.999999 secs', 'PT59.999999S')),
        ((0, 0, 0, 0, -1, 0, 1),
            ('-0:00:59.999999', '-00:00:59.999999',
             '@ 59.999999 secs ago', 'PT-59.999999S')),
        ((0, 0, 0, 0, -1, 1, 1),
            ('-0:00:58.999999', '-00:00:58.999999',
             '@ 58.999999 secs ago', 'PT-58.999999S')),
        ((0, 0, 42, 0, 0, 0, 0),
            ('42 0:00:00', '42 days', '@ 42 days', 'P42D')),
        ((0, 0, -7, 0, 0, 0, 0),
            ('-7 0:00:00', '-7 days', '@ 7 days ago', 'P-7D')),
        ((1, 1, 1, 1, 1, 0, 0),
            ('+1-1 +1 +1:01:00', '1 year 1 mon 1 day 01:01:00',
             '@ 1 year 1 mon 1 day 1 hour 1 min', 'P1Y1M1DT1H1M')),
        ((0, -11, -1, -1, 1, 0, 0),
            ('-0-11 -1 -0:59:00', '-11 mons -1 days -00:59:00',
             '@ 11 mons 1 day 59 mins ago', 'P-11M-1DT-59M')),
        ((-1, -1, -1, -1, -1, 0, 0),
            ('-1-1 -1 -1:01:00', '-1 years -1 mons -1 days -01:01:00',
             '@ 1 year 1 mon 1 day 1 hour 1 min ago', 'P-1Y-1M-1DT-1H-1M')),
        ((-1, 0, -3, 1, 0, 0, 0),
            ('-1-0 -3 +1:00:00', '-1 years -3 days +01:00:00',
             '@ 1 year 3 days -1 hours ago', 'P-1Y-3DT1H')),
        ((1, 0, 0, 0, 0, 0, 1),
            ('+1-0 +0 +0:00:00.000001', '1 year 00:00:00.000001',
             '@ 1 year 0.000001 secs', 'P1YT0.000001S')),
        ((1, 0, 0, 0, 0, 0, -1),
            ('+1-0 +0 -0:00:00.000001', '1 year -00:00:00.000001',
             '@ 1 year -0.000001 secs', 'P1YT-0.000001S')),
        ((1, 2, 3, 4, 5, 6, 7),
            ('+1-2 +3 +4:05:06.000007',
             '1 year 2 mons 3 days 04:05:06.000007',
             '@ 1 year 2 mons 3 days 4 hours 5 mins 6.000007 secs',
             'P1Y2M3DT4H5M6.000007S')),
        ((0, 10, 3, -4, 5, -6, 7),
            ('+0-10 +3 -3:55:05.999993', '10 mons 3 days -03:55:05.999993',
             '@ 10 mons 3 days -3 hours -55 mins -5.999993 secs',
             'P10M3DT-3H-55M-5.999993S')),
        ((0, -10, -3, 4, -5, 6, -7),
            ('-0-10 -3 +3:55:05.999993',
             '-10 mons -3 days +03:55:05.999993',
             '@ 10 mons 3 days -3 hours -55 mins -5.999993 secs ago',
             'P-10M-3DT3H55M5.999993S'))]

    def testCastInterval(self):
        for result, values in self.intervals:
            f = pg.cast_interval
            years, mons, days, hours, mins, secs, usecs = result
            days += 365 * years + 30 * mons
            interval = timedelta(days=days, hours=hours, minutes=mins,
                seconds=secs, microseconds=usecs)
            for value in values:
                self.assertEqual(f(value), interval)


class TestEscapeFunctions(unittest.TestCase):
    """Test pg escape and unescape functions.

    The libpq interface memorizes some parameters of the last opened
    connection that influence the result of these functions.
    Therefore we cannot do rigid tests of these functions here.
    We leave this for the test module that runs with a database.

    """

    def testEscapeString(self):
        f = pg.escape_string
        r = f(b'plain')
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, b'plain')
        r = f(u'plain')
        self.assertIsInstance(r, unicode)
        self.assertEqual(r, u'plain')
        r = f("that's cheese")
        self.assertIsInstance(r, str)
        self.assertEqual(r, "that''s cheese")

    def testEscapeBytea(self):
        f = pg.escape_bytea
        r = f(b'plain')
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, b'plain')
        r = f(u'plain')
        self.assertIsInstance(r, unicode)
        self.assertEqual(r, u'plain')
        r = f("that's cheese")
        self.assertIsInstance(r, str)
        self.assertEqual(r, "that''s cheese")

    def testUnescapeBytea(self):
        f = pg.unescape_bytea
        r = f(b'plain')
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, b'plain')
        r = f(u'plain')
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, b'plain')
        r = f(b"das is' k\\303\\244se")
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, u"das is' käse".encode('utf-8'))
        r = f(u"das is' k\\303\\244se")
        self.assertIsInstance(r, bytes)
        self.assertEqual(r, u"das is' käse".encode('utf-8'))
        r = f(b'O\\000ps\\377!')
        self.assertEqual(r, b'O\x00ps\xff!')
        r = f(u'O\\000ps\\377!')
        self.assertEqual(r, b'O\x00ps\xff!')


class TestConfigFunctions(unittest.TestCase):
    """Test the functions for changing default settings.

    The effect of most of these cannot be tested here, because that
    needs a database connection.  So we merely test their existence here.

    """

    def testGetDatestyle(self):
        self.assertIsNone(pg.get_datestyle())

    def testGetDatestyle(self):
        datestyle = pg.get_datestyle()
        try:
            pg.set_datestyle('ISO, YMD')
            self.assertEqual(pg.get_datestyle(), 'ISO, YMD')
            pg.set_datestyle('Postgres, MDY')
            self.assertEqual(pg.get_datestyle(), 'Postgres, MDY')
            pg.set_datestyle('Postgres, DMY')
            self.assertEqual(pg.get_datestyle(), 'Postgres, DMY')
            pg.set_datestyle('SQL, MDY')
            self.assertEqual(pg.get_datestyle(), 'SQL, MDY')
            pg.set_datestyle('SQL, DMY')
            self.assertEqual(pg.get_datestyle(), 'SQL, DMY')
            pg.set_datestyle('German, DMY')
            self.assertEqual(pg.get_datestyle(), 'German, DMY')
            pg.set_datestyle(None)
            self.assertIsNone(pg.get_datestyle())
        finally:
            pg.set_datestyle(datestyle)

    def testGetDecimalPoint(self):
        r = pg.get_decimal_point()
        self.assertIsInstance(r, str)
        self.assertEqual(r, '.')

    def testSetDecimalPoint(self):
        point = pg.get_decimal_point()
        try:
            pg.set_decimal_point('*')
            r = pg.get_decimal_point()
            self.assertIsInstance(r, str)
            self.assertEqual(r, '*')
        finally:
            pg.set_decimal_point(point)
        r = pg.get_decimal_point()
        self.assertIsInstance(r, str)
        self.assertEqual(r, point)

    def testGetDecimal(self):
        r = pg.get_decimal()
        self.assertIs(r, pg.Decimal)

    def testSetDecimal(self):
        decimal_class = pg.Decimal
        try:
            pg.set_decimal(int)
            r = pg.get_decimal()
            self.assertIs(r, int)
        finally:
            pg.set_decimal(decimal_class)
        r = pg.get_decimal()
        self.assertIs(r, decimal_class)

    def testGetBool(self):
        r = pg.get_bool()
        self.assertIsInstance(r, bool)
        self.assertIs(r, True)

    def testSetBool(self):
        use_bool = pg.get_bool()
        try:
            pg.set_bool(False)
            r = pg.get_bool()
            pg.set_bool(use_bool)
            self.assertIsInstance(r, bool)
            self.assertIs(r, False)
            pg.set_bool(True)
            r = pg.get_bool()
            self.assertIsInstance(r, bool)
            self.assertIs(r, True)
        finally:
            pg.set_bool(use_bool)
        r = pg.get_bool()
        self.assertIsInstance(r, bool)
        self.assertIs(r, use_bool)

    def testGetByteaEscaped(self):
        r = pg.get_bytea_escaped()
        self.assertIsInstance(r, bool)
        self.assertIs(r, False)

    def testSetByteaEscaped(self):
        bytea_escaped = pg.get_bytea_escaped()
        try:
            pg.set_bytea_escaped(True)
            r = pg.get_bytea_escaped()
            pg.set_bytea_escaped(bytea_escaped)
            self.assertIsInstance(r, bool)
            self.assertIs(r, True)
            pg.set_bytea_escaped(False)
            r = pg.get_bytea_escaped()
            self.assertIsInstance(r, bool)
            self.assertIs(r, False)
        finally:
            pg.set_bytea_escaped(bytea_escaped)
        r = pg.get_bytea_escaped()
        self.assertIsInstance(r, bool)
        self.assertIs(r, bytea_escaped)

    def testGetJsondecode(self):
        r = pg.get_jsondecode()
        self.assertTrue(callable(r))
        self.assertIs(r, json.loads)

    def testSetJsondecode(self):
        jsondecode = pg.get_jsondecode()
        try:
            pg.set_jsondecode(None)
            r = pg.get_jsondecode()
            self.assertIsNone(r)
            pg.set_jsondecode(str)
            r = pg.get_jsondecode()
            self.assertIs(r, str)
            self.assertRaises(TypeError, pg.set_jsondecode, 'invalid')
        finally:
            pg.set_jsondecode(jsondecode)
        r = pg.get_jsondecode()
        self.assertIs(r, jsondecode)


class TestModuleConstants(unittest.TestCase):
    """Test the existence of the documented module constants."""

    def testVersion(self):
        v = pg.version
        self.assertIsInstance(v, str)
        # make sure the version conforms to PEP440
        re_version = r"""^
            (\d[\.\d]*(?<= \d))
            ((?:[abc]|rc)\d+)?
            (?:(\.post\d+))?
            (?:(\.dev\d+))?
            (?:(\+(?![.])[a-zA-Z0-9\.]*[a-zA-Z0-9]))?
            $"""
        match = re.match(re_version, v, re.X)
        self.assertIsNotNone(match)
        self.assertEqual(pg.__version__, v)


if __name__ == '__main__':
    unittest.main()
