jackcartersmith/blackmagic
1
0
Fork 0
Code Issues Pull Requests Releases Activity

tests: gadget0: remove magic constants

Browse Source
This commit is contained in:
Jochen Hoenicke 2016-10-04 17:25:58 +02:00 committed by Karl Palsson
parent 0259102560
commit afb66508a4
1 changed files with 29 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
tests/gadget-zero/test_gadget0.py
View File

@ -6,6 +6,9 @@ import logging
import unittest import unittest
VENDOR_ID=0xcafe
PRODUCT_ID=0xcafe
#DUT_SERIAL = "stm32f429i-disco" #DUT_SERIAL = "stm32f429i-disco"
DUT_SERIAL = "stm32f4disco" DUT_SERIAL = "stm32f4disco"
#DUT_SERIAL = "stm32f103-generic" #DUT_SERIAL = "stm32f103-generic"
@ -13,6 +16,13 @@ DUT_SERIAL = "stm32f4disco"
#DUT_SERIAL = "stm32f072disco" #DUT_SERIAL = "stm32f072disco"
#DUT_SERIAL = "stm32l053disco" #DUT_SERIAL = "stm32l053disco"
GZ_REQ_SET_PATTERN=1
GZ_REQ_PRODUCE=2
GZ_REQ_SET_ALIGNED=3
GZ_REQ_SET_UNALIGNED=4
GZ_REQ_WRITE_LOOPBACK_BUFFER=10
GZ_REQ_READ_LOOPBACK_BUFFER=11
class find_by_serial(object): class find_by_serial(object):
def __init__(self, serial): def __init__(self, serial):
self._serial = serial self._serial = serial
@ -25,7 +35,7 @@ class TestGadget0(unittest.TestCase):
# TODO - parameterize this with serial numbers so we can find # TODO - parameterize this with serial numbers so we can find
# gadget 0 code for different devices. (or use different PIDs?) # gadget 0 code for different devices. (or use different PIDs?)
def setUp(self): def setUp(self):
self.dev = usb.core.find(idVendor=0xcafe, idProduct=0xcafe, custom_match=find_by_serial(DUT_SERIAL)) self.dev = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID, custom_match=find_by_serial(DUT_SERIAL))
self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device") self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device")
self.longMessage = True self.longMessage = True
@ -77,7 +87,7 @@ class TestConfigSourceSink(unittest.TestCase):
""" """
def setUp(self): def setUp(self):
self.dev = usb.core.find(idVendor=0xcafe, idProduct=0xcafe, custom_match=find_by_serial(DUT_SERIAL)) self.dev = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID, custom_match=find_by_serial(DUT_SERIAL))
self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device") self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device")
self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2) self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2)
@ -121,7 +131,7 @@ class TestConfigSourceSink(unittest.TestCase):
self.assertEqual(written, len(data), "should have written all bytes plz") self.assertEqual(written, len(data), "should have written all bytes plz")
def test_read_zeros(self): def test_read_zeros(self):
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 0) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 0)
self.ep_in.read(self.ep_in.wMaxPacketSize) # Clear out any prior pattern data self.ep_in.read(self.ep_in.wMaxPacketSize) # Clear out any prior pattern data
# unless, you know _exactly_ how much will be written by the device, always read # unless, you know _exactly_ how much will be written by the device, always read
# an integer multiple of max packet size, to avoid overflows. # an integer multiple of max packet size, to avoid overflows.
@ -136,9 +146,9 @@ class TestConfigSourceSink(unittest.TestCase):
def test_read_sequence(self): def test_read_sequence(self):
# switching to the mod63 pattern requires resynching carefully to read out any zero frames already # switching to the mod63 pattern requires resynching carefully to read out any zero frames already
# queued, but still make sure we start the sequence at zero. # queued, but still make sure we start the sequence at zero.
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 1) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 1)
self.ep_in.read(self.ep_in.wMaxPacketSize) # Potentially queued zeros, or would have been safe. self.ep_in.read(self.ep_in.wMaxPacketSize) # Potentially queued zeros, or would have been safe.
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 1) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 1)
self.ep_in.read(self.ep_in.wMaxPacketSize) # definitely right pattern now, but need to restart at zero. self.ep_in.read(self.ep_in.wMaxPacketSize) # definitely right pattern now, but need to restart at zero.
read_size = self.ep_in.wMaxPacketSize * 3 read_size = self.ep_in.wMaxPacketSize * 3
data = self.dev.read(self.ep_in, read_size) data = self.dev.read(self.ep_in, read_size)
@ -155,10 +165,10 @@ class TestConfigSourceSink(unittest.TestCase):
self.assertEqual(oo, len(dd), "should have written full packet") self.assertEqual(oo, len(dd), "should have written full packet")
def test_control_known(self): def test_control_known(self):
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 0) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 0)
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 1) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 1)
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 99) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 99)
self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, 0x1, 0) self.dev.ctrl_transfer(uu.CTRL_TYPE_VENDOR | uu.CTRL_RECIPIENT_INTERFACE, GZ_REQ_SET_PATTERN, 0)
def test_control_unknown(self): def test_control_unknown(self):
try: try:
@ -176,7 +186,7 @@ class TestConfigSourceSinkPerformance(unittest.TestCase):
""" """
def setUp(self): def setUp(self):
self.dev = usb.core.find(idVendor=0xcafe, idProduct=0xcafe, custom_match=find_by_serial(DUT_SERIAL)) self.dev = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID, custom_match=find_by_serial(DUT_SERIAL))
self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device") self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device")
self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2) self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2)
@ -227,7 +237,7 @@ class TestControlTransfer_Reads(unittest.TestCase):
""" """
def setUp(self): def setUp(self):
self.dev = usb.core.find(idVendor=0xcafe, idProduct=0xcafe, custom_match=find_by_serial(DUT_SERIAL)) self.dev = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID, custom_match=find_by_serial(DUT_SERIAL))
self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device") self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device")
self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2) self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2)
@ -238,14 +248,14 @@ class TestControlTransfer_Reads(unittest.TestCase):
def inner_t(self, wVal, read_len): def inner_t(self, wVal, read_len):
wVal = int(wVal) wVal = int(wVal)
read_len = int(read_len) read_len = int(read_len)
q = self.dev.ctrl_transfer(self.req, 2, wVal, 0, read_len) q = self.dev.ctrl_transfer(self.req, GZ_REQ_PRODUCE, wVal, 0, read_len)
self.assertEqual(len(q), wVal, "Should have read as much as we asked for?") self.assertEqual(len(q), wVal, "Should have read as much as we asked for?")
def tearDown(self): def tearDown(self):
uu.dispose_resources(self.dev) uu.dispose_resources(self.dev)
def test_basic(self): def test_basic(self):
x = self.dev.ctrl_transfer(self.req, 2, 32, 0, 32) x = self.dev.ctrl_transfer(self.req, GZ_REQ_PRODUCE, 32, 0, 32)
self.assertEqual(32, len(x)) self.assertEqual(32, len(x))
def test_matching_sizes(self): def test_matching_sizes(self):
@ -255,7 +265,7 @@ class TestControlTransfer_Reads(unittest.TestCase):
""" """
def inner(x): def inner(x):
x = int(x) x = int(x)
q = self.dev.ctrl_transfer(self.req, 2, x, 0, x) q = self.dev.ctrl_transfer(self.req, GZ_REQ_PRODUCE, x, 0, x)
self.assertEqual(len(q), x, "Should have read as much as we asked for") self.assertEqual(len(q), x, "Should have read as much as we asked for")
ep0_size = self.dev.bMaxPacketSize0 ep0_size = self.dev.bMaxPacketSize0
@ -272,7 +282,7 @@ class TestControlTransfer_Reads(unittest.TestCase):
(Don't make them too, big, or libusb will reject you outright, see MAX_CTRL_BUFFER_LENGTH in libusb sources) (Don't make them too, big, or libusb will reject you outright, see MAX_CTRL_BUFFER_LENGTH in libusb sources)
""" """
try: try:
self.dev.ctrl_transfer(self.req, 2, 10 * self.dev.bMaxPacketSize0, 0, 10 * self.dev.bMaxPacketSize0) self.dev.ctrl_transfer(self.req, GZ_REQ_PRODUCE, 10 * self.dev.bMaxPacketSize0, 0, 10 * self.dev.bMaxPacketSize0)
self.fail("Should have got a stall") self.fail("Should have got a stall")
except usb.core.USBError as e: except usb.core.USBError as e:
# Note, this might not be as portable as we'd like. # Note, this might not be as portable as we'd like.
@ -314,7 +324,7 @@ class TestControlTransfer_Reads(unittest.TestCase):
tell the device to produce more than we ask for. tell the device to produce more than we ask for.
Note, this doesn't test the usb stack, it tests the application code behaves. Note, this doesn't test the usb stack, it tests the application code behaves.
""" """
q = self.dev.ctrl_transfer(self.req, 2, 100, 0, 10) q = self.dev.ctrl_transfer(self.req, GZ_REQ_PRODUCE, 100, 0, 10)
self.assertEqual(len(q), 10, "In this case, should have gotten wLen back") self.assertEqual(len(q), 10, "In this case, should have gotten wLen back")
@ -327,7 +337,7 @@ class TestUnaligned(unittest.TestCase):
""" """
def setUp(self): def setUp(self):
self.dev = usb.core.find(idVendor=0xcafe, idProduct=0xcafe, custom_match=find_by_serial(DUT_SERIAL)) self.dev = usb.core.find(idVendor=VENDOR_ID, idProduct=PRODUCT_ID, custom_match=find_by_serial(DUT_SERIAL))
self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device") self.assertIsNotNone(self.dev, "Couldn't find locm3 gadget0 device")
self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2) self.cfg = uu.find_descriptor(self.dev, bConfigurationValue=2)
@ -344,11 +354,11 @@ class TestUnaligned(unittest.TestCase):
def set_unaligned(self): def set_unaligned(self):
# GZ_REQ_SET_UNALIGNED # GZ_REQ_SET_UNALIGNED
x = self.dev.ctrl_transfer(self.req, 4, 0, 0) x = self.dev.ctrl_transfer(self.req, GZ_REQ_SET_UNALIGNED, 0, 0)
def set_aligned(self): def set_aligned(self):
# GZ_REQ_SET_ALIGNED # GZ_REQ_SET_ALIGNED
x = self.dev.ctrl_transfer(self.req, 3, 0, 0) x = self.dev.ctrl_transfer(self.req, GZ_REQ_SET_ALIGNED, 0, 0)
def do_readwrite(self): def do_readwrite(self):
""" """