Raise error if could not resolve attached datum

OpShin · Apr 11, 2023 · 0f23048 · 0f23048
1 parent 8897aa5
commit 0f23048
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/opshin/prelude.py b/opshin/prelude.py
@@ -91,12 +91,15 @@ def resolve_datum(
     Otherwise it returns NoOutputDatum.
     """
     attached_datum = txout.datum
-    res: Union[SomeOutputDatum, NoOutputDatum] = NoOutputDatum()
     if isinstance(attached_datum, SomeOutputDatumHash):
         if attached_datum.datum_hash in tx_info.data.keys():
-            res = SomeOutputDatum(
+            res: Union[SomeOutputDatum, NoOutputDatum] = SomeOutputDatum(
                 tx_info.data.get(attached_datum.datum_hash, Nothing())
             )
+        else:
+            assert (
+                False
+            ), "Could not resolve attached datum from datum hash, because it was not embedded in the transaction"
     else:
-        res = attached_datum
+        res: Union[SomeOutputDatum, NoOutputDatum] = attached_datum
     return res