Use random data in TestTestProtocol

Author: Steve-Tech

tests/general.py

@@ -1,5 +1,6 @@
 import unittest
 import sys
+import random
 from cros_ec_python import get_cros_ec, ECError, general as ec_general
 
 if sys.platform == "win32":
@@ -9,6 +10,8 @@
 
 ec = get_cros_ec()
 
+random.seed(0)  # For reproducibility of random tests
+
 
 class TestProtoVersion(unittest.TestCase):
     def test_version0(self):
@@ -86,11 +89,11 @@ def test_invalid(self):
 
 class TestTestProtocol(unittest.TestCase):
     def test_version0(self):
-        length = 32
-        buf = bytes(range(length))
-        resp = ec_general.test_protocol(ec, 0, length, buf)
-        print(type(self).__name__, "-", resp)
-        self.assertEqual(resp, buf)
+        for length in range(0, 8):
+            buf = random.randbytes(length)
+            resp = ec_general.test_protocol(ec, 0, length, buf)
+            print(type(self).__name__, "-", resp)
+            self.assertEqual(resp, buf)
 
     def test_error(self):
         with self.assertRaises(ECError):
@@ -102,7 +105,7 @@ def test_error(self):
     )
     def test_warning(self):
         with self.assertWarns(RuntimeWarning):
-            length = 16
+            length = 8
             buf = bytes(range(length))
             resp = ec_general.test_protocol(ec, 0, length, buf, length * 2)
             print(type(self).__name__, "-", resp)