From 8c7abc33f27556611bc36f61b651841905bbd1b0 Mon Sep 17 00:00:00 2001
From: Carlos Medeiros <carlos.medeiros@zondax.ch>
Date: Tue, 8 Oct 2024 17:37:56 +0100
Subject: [PATCH] add tests to verify wrong state rejection

---
 tests_zemu/tests/masp.test.ts | 55 +++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/tests_zemu/tests/masp.test.ts b/tests_zemu/tests/masp.test.ts
index afc86af..f75911e 100644
--- a/tests_zemu/tests/masp.test.ts
+++ b/tests_zemu/tests/masp.test.ts
@@ -195,4 +195,59 @@ describe('Masp', function () {
       await sim.close()
     }
   })
+
+  test.concurrent.each(MASP_MODELS)('Wrong MASP starting instruction', async function (m) {
+    const sim = new Zemu(m.path)
+    try {
+      await sim.start({ ...defaultOptions, model: m.name })
+      const app = new NamadaApp(sim.getTransport())
+
+      const respClean = await app.cleanRandomnessBuffers()
+      console.log(respClean)
+      expect(respClean.returnCode).toEqual(0x9000)
+
+      // Wrong MASP starting INS no randomness was computed or spends signed
+      const resp = await app.getSpendSignature()
+
+      // Expect the specific return code and error message
+      expect(resp.returnCode).toEqual(27012)
+      expect(resp.errorMessage).toEqual('Data is invalid')
+    } finally {
+      await sim.close()
+    }
+  })
+
+  test.concurrent.each(MASP_MODELS)('Wrong MASP sequence', async function (m) {
+    const sim = new Zemu(m.path)
+    try {
+      await sim.start({ ...defaultOptions, model: m.name })
+      const app = new NamadaApp(sim.getTransport())
+
+      // First step: compute randomness
+      const respSpend = await app.getSpendRandomness()
+      console.log(respSpend)
+      expect(respSpend.returnCode).toEqual(0x9000)
+      expect(respSpend.errorMessage).toEqual('No errors')
+
+      const respOutput = await app.getOutputRandomness()
+      console.log(respOutput)
+      expect(respOutput.returnCode).toEqual(0x9000)
+      expect(respOutput.errorMessage).toEqual('No errors')
+
+      const respRandomness = await app.getConvertRandomness()
+      console.log(respRandomness)
+      expect(respRandomness.returnCode).toEqual(0x9000)
+      expect(respRandomness.errorMessage).toEqual('No errors')
+
+      // Missing spend signature and trying to extract the signatures
+      const resp = await app.getSpendSignature()
+      console.log(resp)
+
+      // Expect the specific return code and error message
+      expect(resp.returnCode).toEqual(27012)
+      expect(resp.errorMessage).toEqual('Data is invalid')
+    } finally {
+      await sim.close()
+    }
+  })
 })