Unit tests

datatrails · Jul 21, 2024 · c65b076 · c65b076
1 parent 6d3087b
commit c65b076
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/scitt/verify_receipt_signature.py b/scitt/verify_receipt_signature.py
@@ -19,7 +19,7 @@
 HEADER_LABEL_DID = 391
 
 
-def read_cbor_file(cbor_file: str) -> bytes | None:
+def read_cbor_file(cbor_file: str) -> Sign1Message:
     """
     opens the receipt from the receipt file.
     NOTE: the receipt is expected to be in cbor encoding.
@@ -149,7 +149,7 @@ def main():
     """Verifies a counter signed receipt signature"""
 
     parser = argparse.ArgumentParser(
-        description="Verify a counter signed receipt signature from a Receipt or Transparent Statement."
+        description="Verify countersigned signature from a Receipt or Transparent Statement."
     )
 
     options = parser.add_argument_group("Input File Type")

diff --git a/unittests/test_verify_receipt_signature.py b/unittests/test_verify_receipt_signature.py
@@ -4,7 +4,7 @@
 
 import unittest
 
-from scitt.verify_receipt_signature import verify_receipt, open_receipt
+from scitt.verify_receipt_signature import verify_receipt, read_cbor_file
 
 from .constants import KNOWN_RECEIPT_FILE
 
@@ -19,7 +19,7 @@ def test_verify_kat_receipt(self):
         """
         tests we can verify the signature of a known receipt.
         """
-        receipt = open_receipt(KNOWN_RECEIPT_FILE)
+        receipt = read_cbor_file(KNOWN_RECEIPT_FILE)
 
         verified = verify_receipt(receipt)