"""Module for testing long and long raw variables."""

class TestLongVar(BaseTestCase):

    def __PerformTest(self, a_Type, a_InputType):
        self.cursor.execute(u"truncate table Test%ss" % a_Type)
        if a_InputType == cx_Oracle.LONG_STRING:
            longString = u""
        else:
            longString = ""
        for i in range(1, 11):
            if a_InputType == cx_Oracle.LONG_STRING:
                char = unichr(ord('A') + i - 1)
            else:
                char = chr(ord('A') + i - 1)
            longString += char * 25000
            self.cursor.setinputsizes(p_LongString = a_InputType)
            self.cursor.execute(u"""
                    insert into Test%ss (
                      IntCol,
                      %sCol
                    ) values (
                      :p_IntegerValue,
                      :p_LongString
                    )""" % (a_Type, a_Type),
                    p_IntegerValue = i,
                    p_LongString = longString)
        self.connection.commit()
        self.cursor.setoutputsize(250000, 2)
        self.cursor.execute(u"""
                select *
                from Test%ss
                order by IntCol""" % a_Type)
        longString = ""
        while 1:
            row = self.cursor.fetchone()
            if row is None:
                break
            integerValue, fetchedValue = row
            char = unichr(ord('A') + integerValue - 1)
            longString += char * 25000
            self.failUnlessEqual(len(fetchedValue), integerValue * 25000)
            self.failUnlessEqual(fetchedValue, longString)

    def testLongs(self):
        "test binding and fetching long data"
        self.__PerformTest("Long", cx_Oracle.LONG_STRING)

    def testLongRaws(self):
        "test binding and fetching long raw data"
        self.__PerformTest("LongRaw", cx_Oracle.LONG_BINARY)

    def testLongCursorDescription(self):
        "test cursor description is accurate for longs"
        self.cursor.execute(u"select * from TestLongs")
        self.failUnlessEqual(self.cursor.description,
                [ (u'INTCOL', cx_Oracle.NUMBER, 10, 22, 9, 0, 0),
                  (u'LONGCOL', cx_Oracle.LONG_STRING, -1, 0, 0, 0, 0) ])

    def testLongRawCursorDescription(self):
        "test cursor description is accurate for long raws"
        self.cursor.execute(u"select * from TestLongRaws")
        self.failUnlessEqual(self.cursor.description,
                [ (u'INTCOL', cx_Oracle.NUMBER, 10, 22, 9, 0, 0),
                  (u'LONGRAWCOL', cx_Oracle.LONG_BINARY, -1, 0, 0, 0, 0) ])

    def testSetOutputSizesAll(self):
        "test setoutputsizes is valid (all)"
        self.cursor.setoutputsize(25000)
        self.cursor.execute(u"select * from TestLongRaws")
        longVar = self.cursor.fetchvars[1]
        self.failUnlessEqual(longVar.size, 25000)
        self.failUnlessEqual(longVar.bufferSize, 25004)

    def testSetOutputSizesWrongColumn(self):
        "test setoutputsizes is valid (wrong column)"
        self.cursor.setoutputsize(25000, 1)
        self.cursor.execute(u"select * from TestLongRaws")
        longVar = self.cursor.fetchvars[1]
        self.failUnlessEqual(longVar.size, 131072)
        self.failUnlessEqual(longVar.bufferSize, 131076)

    def testSetOutputSizesRightColumn(self):
        "test setoutputsizes is valid (right column)"
        self.cursor.setoutputsize(35000, 2)
        self.cursor.execute(u"select * from TestLongs")
        longVar = self.cursor.fetchvars[1]
        self.failUnlessEqual(longVar.size, 35000)
        self.failUnlessEqual(longVar.bufferSize, 70004)

    def testArraySizeTooLarge(self):
        "test array size too large generates an exception"
        self.cursor.arraysize = 65536
        self.failUnlessRaises(ValueError, self.cursor.execute,
                u"select * from TestLongRaws")

