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

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

Hide file tree

Showing 12 changed files with 59 additions and 0 deletions.
diff --git a/test/python/snapshots/nanos/test_sign_when_chain_is_setup/app_context/chain_id.png b/test/python/snapshots/nanos/test_sign_when_chain_is_setup/app_context/chain_id.png
diff --git a/...on/snapshots/nanos/test_sign_when_chain_is_setup/app_context/high_watermark.png b/...on/snapshots/nanos/test_sign_when_chain_is_setup/app_context/high_watermark.png
diff --git a/...snapshots/nanos/test_sign_when_chain_is_setup/app_context/public_key_hash_1.png b/...snapshots/nanos/test_sign_when_chain_is_setup/app_context/public_key_hash_1.png
diff --git a/...snapshots/nanos/test_sign_when_chain_is_setup/app_context/public_key_hash_2.png b/...snapshots/nanos/test_sign_when_chain_is_setup/app_context/public_key_hash_2.png
diff --git a/.../python/snapshots/nanosp/test_sign_when_chain_is_setup/app_context/chain_id.png b/.../python/snapshots/nanosp/test_sign_when_chain_is_setup/app_context/chain_id.png
diff --git a/...n/snapshots/nanosp/test_sign_when_chain_is_setup/app_context/high_watermark.png b/...n/snapshots/nanosp/test_sign_when_chain_is_setup/app_context/high_watermark.png
diff --git a/.../snapshots/nanosp/test_sign_when_chain_is_setup/app_context/public_key_hash.png b/.../snapshots/nanosp/test_sign_when_chain_is_setup/app_context/public_key_hash.png
diff --git a/test/python/snapshots/nanox/test_sign_when_chain_is_setup/app_context/chain_id.png b/test/python/snapshots/nanox/test_sign_when_chain_is_setup/app_context/chain_id.png
diff --git a/...on/snapshots/nanox/test_sign_when_chain_is_setup/app_context/high_watermark.png b/...on/snapshots/nanox/test_sign_when_chain_is_setup/app_context/high_watermark.png
diff --git a/...n/snapshots/nanox/test_sign_when_chain_is_setup/app_context/public_key_hash.png b/...n/snapshots/nanox/test_sign_when_chain_is_setup/app_context/public_key_hash.png
diff --git a/...python/snapshots/stax/test_sign_when_chain_is_setup/app_context/app_context.png b/...python/snapshots/stax/test_sign_when_chain_is_setup/app_context/app_context.png
diff --git a/test/python/test_instructions.py b/test/python/test_instructions.py
@@ -669,6 +669,65 @@ def test_sign_when_no_chain_setup(
     )
 
 
+def test_sign_when_chain_is_setup(
+        client: TezosClient,
+        tezos_navigator: TezosNavigator) -> None:
+    """Check that signing when chain has been setup change main HWM."""
+
+    account = DEFAULT_ACCOUNT
+    main_chain_id = "NetXH12AexHqTQa" # Chain = 1
+
+    tezos_navigator.setup_app_context(
+        account,
+        main_chain_id,
+        main_hwm=Hwm(0),
+        test_hwm=Hwm(0)
+    )
+
+    attestation = build_attestation(
+        1, 0,
+        main_chain_id
+    )
+
+    client.sign_message(account, attestation)
+
+    tezos_navigator.check_app_context(
+        account,
+        chain_id=main_chain_id,
+        main_hwm=Hwm(1, 0),
+        test_hwm=Hwm(0, 0),
+    )
+
+    attestation = build_attestation(
+        2, 0,
+        DEFAULT_CHAIN_ID # Chain = 0
+    )
+
+    client.sign_message(account, attestation)
+
+    tezos_navigator.check_app_context(
+        account,
+        chain_id=main_chain_id,
+        main_hwm=Hwm(1, 0),
+        test_hwm=Hwm(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=main_chain_id,
+        main_hwm=Hwm(1, 0),
+        test_hwm=Hwm(2, 0),
+    )
+
+
 # Data generated by the old application itself
 HMAC_TEST_SET = [
     (DEFAULT_ACCOUNT,