import unittest

import objc

class TestAllocateBuffer(unittest.TestCase):
    def testBadLengths(self):
        self.assertRaises(ValueError, objc.allocateBuffer, 0)
        self.assertRaises(ValueError, objc.allocateBuffer, -1000)

    def testBuffer(self):
        b = objc.allocateBuffer(10000)
        self.assertEquals(len(b), 10000)

        for i in range(0,10000):
            b[i] = chr(i % 256)

        b[5:10] = b[1:6]
        b[5:10] = 'abcde'
        try:
            b[5:10] = 'abcdefghijk'
        except TypeError, r:
            if str(r).find("right operand length must match slice length") is not 0:
                raise

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