Add check for num_keys in test_build_header_prealpha

2017-09-11 12:50:36 -07:00 · 2017-09-11 12:50:36 -07:00 · 0feb161d7b
parent de59deb7b1
commit 0feb161d7b
1 changed files with 2 additions and 1 deletions
--- a/tests/test_client.py
+++ b/tests/test_client.py
@ -43,8 +43,9 @@ class TestClient(unittest.TestCase):
        try:
            msgpack.loads(header)
        except Exception as E:
-            self.fail("Failed to unpack header: {}".format(E))
+            self.fail("Failed to unpack header:\n{}".format(E))

+        self.assertIn((3).to_bytes(4, byteorder='big'), header)
        for key in enc_keys:
            self.assertIn(key, header)