Tests: add test check HWM after signing when no chain has been setup

trilitech · Feb 5, 2024 · 4ac86f9 · 4ac86f9
1 parent 4b04f50
commit 4ac86f9
Show file tree

Hide file tree

Showing 23 changed files with 63 additions and 2 deletions.
diff --git a/...snapshots/nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png b/...snapshots/nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png
diff --git a/...ots/nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png b/...ots/nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png
diff --git a/.../nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash_1.png b/.../nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash_1.png
diff --git a/.../nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash_2.png b/.../nanos/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash_2.png
diff --git a/...snapshots/nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png b/...snapshots/nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png
diff --git a/...ots/nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png b/...ots/nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png
diff --git a/.../nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash_1.png b/.../nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash_1.png
diff --git a/.../nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash_2.png b/.../nanos/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash_2.png
diff --git a/...napshots/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png b/...napshots/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png
diff --git a/...ts/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png b/...ts/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png
diff --git a/...s/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash.png b/...s/nanosp/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash.png
diff --git a/...napshots/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png b/...napshots/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png
diff --git a/...ts/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png b/...ts/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png
diff --git a/...s/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash.png b/...s/nanosp/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash.png
diff --git a/...snapshots/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png b/...snapshots/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/chain_id.png
diff --git a/...ots/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png b/...ots/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/high_watermark.png
diff --git a/...ts/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash.png b/...ts/nanox/test_sign_when_no_chain_setup/sign_1_0/app_context/public_key_hash.png
diff --git a/...snapshots/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png b/...snapshots/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/chain_id.png
diff --git a/...ots/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png b/...ots/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/high_watermark.png
diff --git a/...ts/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash.png b/...ts/nanox/test_sign_when_no_chain_setup/sign_2_0/app_context/public_key_hash.png
diff --git a/...apshots/stax/test_sign_when_no_chain_setup/sign_1_0/app_context/app_context.png b/...apshots/stax/test_sign_when_no_chain_setup/sign_1_0/app_context/app_context.png
diff --git a/...apshots/stax/test_sign_when_no_chain_setup/sign_2_0/app_context/app_context.png b/...apshots/stax/test_sign_when_no_chain_setup/sign_2_0/app_context/app_context.png
diff --git a/test/python/test_instructions.py b/test/python/test_instructions.py
@@ -177,7 +177,7 @@ def test_setup_app_context(account: Account, tezos_navigator: TezosNavigator) ->
     """Test the SETUP instruction."""
     snap_path = Path(f"{account}")
 
-    main_chain_id = "NetXH12AexHqTQa"
+    main_chain_id = "NetXH12AexHqTQa" # Chain = 1
     main_hwm = Hwm(1)
     test_hwm = Hwm(2)
 
@@ -588,7 +588,7 @@ def test_sign_level_authorized(
 def test_sign_not_authorized_key(
         client: TezosClient,
         tezos_navigator: TezosNavigator) -> None:
-    """Check that signing with a key different from the authorized key is not authorized.."""
+    """Check that signing with a key different from the authorized key is not authorized."""
 
     account_1 = DEFAULT_ACCOUNT
     account_2 = DEFAULT_ACCOUNT_2
@@ -608,6 +608,67 @@ def test_sign_not_authorized_key(
         client.sign_message(account_2, attestation)
 
 
+def test_sign_when_no_chain_setup(
+        client: TezosClient,
+        tezos_navigator: TezosNavigator) -> None:
+    """Check that signing when no chain has been setup change main HWM."""
+
+    account = DEFAULT_ACCOUNT
+
+    tezos_navigator.setup_app_context(
+        account,
+        DEFAULT_CHAIN_ID, # Chain = 0
+        main_hwm=Hwm(0),
+        test_hwm=Hwm(0)
+    )
+
+    attestation = build_attestation(
+        1, 0,
+        DEFAULT_CHAIN_ID # Chain = 0
+    )
+
+    client.sign_message(account, attestation)
+
+    tezos_navigator.check_app_context(
+        account,
+        chain_id=DEFAULT_CHAIN_ID,
+        main_hwm=Hwm(1, 0),
+        test_hwm=Hwm(0, 0),
+        snap_path=Path("sign_1_0")
+    )
+
+    attestation = build_attestation(
+        2, 0,
+        "NetXH12AexHqTQa" # Chain = 1
+    )
+
+    client.sign_message(account, attestation)
+
+    tezos_navigator.check_app_context(
+        account,
+        chain_id=DEFAULT_CHAIN_ID,
+        main_hwm=Hwm(2, 0),
+        test_hwm=Hwm(0, 0),
+        snap_path=Path("sign_2_0")
+    )
+
+    attestation = build_attestation(
+        2, 0,
+        "NetXH12Af5mrXhq" # Chain = 2
+    )
+
+    with StatusCode.WRONG_VALUES.expected():
+        client.sign_message(account, attestation)
+
+    tezos_navigator.check_app_context(
+        account,
+        chain_id=DEFAULT_CHAIN_ID,
+        main_hwm=Hwm(2, 0),
+        test_hwm=Hwm(0, 0),
+        snap_path=Path("sign_2_0")
+    )
+
+
 # Data generated by the old application itself
 HMAC_TEST_SET = [
     (DEFAULT_ACCOUNT,