imgtool: Add missing encodings to emitter tests
Signed-off-by: David Vincze <david.vincze@arm.com>
Change-Id: Iff37cb62514b181a30f5537d6692d55e3d9c73ed
diff --git a/scripts/imgtool/keys/ecdsa_test.py b/scripts/imgtool/keys/ecdsa_test.py
index 65b709f..55f6cc4 100644
--- a/scripts/imgtool/keys/ecdsa_test.py
+++ b/scripts/imgtool/keys/ecdsa_test.py
@@ -52,17 +52,36 @@
"""Basic sanity check on the code emitters."""
k = ECDSA256P1.generate()
+ pubpem = io.StringIO()
+ k.emit_public_pem(pubpem)
+ self.assertIn("BEGIN PUBLIC KEY", pubpem.getvalue())
+ self.assertIn("END PUBLIC KEY", pubpem.getvalue())
+
ccode = io.StringIO()
k.emit_c_public(ccode)
self.assertIn("ecdsa_pub_key", ccode.getvalue())
self.assertIn("ecdsa_pub_key_len", ccode.getvalue())
+ hashccode = io.StringIO()
+ k.emit_c_public_hash(hashccode)
+ self.assertIn("ecdsa_pub_key_hash", hashccode.getvalue())
+ self.assertIn("ecdsa_pub_key_hash_len", hashccode.getvalue())
+
rustcode = io.StringIO()
k.emit_rust_public(rustcode)
self.assertIn("ECDSA_PUB_KEY", rustcode.getvalue())
+ # raw data - bytes
+ pubraw = io.BytesIO()
+ k.emit_raw_public(pubraw)
+ self.assertTrue(len(pubraw.getvalue()) > 0)
+
+ hashraw = io.BytesIO()
+ k.emit_raw_public_hash(hashraw)
+ self.assertTrue(len(hashraw.getvalue()) > 0)
+
def test_emit_pub(self):
- """Basic sanity check on the code emitters."""
+ """Basic sanity check on the code emitters, from public key."""
pubname = self.tname("public.pem")
k = ECDSA256P1.generate()
k.export_public(pubname)