Skip to content

Commit

Permalink
testUSB1: Add tests for USBTransfer bytearray buffer.
Browse files Browse the repository at this point in the history
  • Loading branch information
vpelletier committed Jan 2, 2017
1 parent 5742ea8 commit 8c3122b
Showing 1 changed file with 9 additions and 5 deletions.
14 changes: 9 additions & 5 deletions usb1/testUSB1.py
Expand Up @@ -23,13 +23,15 @@
import libusb1
from ctypes import pointer

buff_len = 1024
buffer_base = [x % 256 for x in xrange(buff_len)]
if sys.version_info[0] == 3:
buff = bytes([0, 0xff])
other_buff = bytes((ord(x) for x in 'foo'))
buff = bytes(buffer_base)
other_buff = bytes(reversed(buffer_base))
else:
buff = '\x00\xff'
other_buff = 'foo'
buff_len = 2
buff = ''.join(chr(x) for x in buffer_base)
other_buff = ''.join(chr(x) for x in reversed(buffer_base))
bytearray_buff = bytearray(buffer_base)

class USBContext(usb1.USBContext):
def open(self):
Expand Down Expand Up @@ -129,6 +131,8 @@ def callback(transfer):
self.assertEqual(buff, transfer.getBuffer())
transfer.setBuffer(other_buff)
self.assertEqual(other_buff, transfer.getBuffer())
transfer.setBuffer(bytearray_buff)
self.assertEqual(bytearray_buff, transfer.getBuffer())
transfer.setBuffer(buff_len)
self.assertEqual(buff_len, len(transfer.getBuffer()))
# All provided, buffer length variant
Expand Down

0 comments on commit 8c3122b

Please sign in to comment.