fix(testing-interface): detect signing wallet instead of hardcoding w1

Author: bogdan-manole

src/testing-interface/lib/Convex/TestingInterface.hs

@@ -81,6 +81,8 @@ import Convex.MockChain.Defaults qualified as Defaults
 import Convex.MonadLog (MonadLog)
 import Convex.NodeParams (NodeParams (..), ledgerProtocolParameters)
 import Convex.ThreatModel (ExceptT, ThreatModel, ThreatModelEnv (..), ThreatModelOutcome (..), getThreatModelName, runExceptT, runThreatModelCheck)
+import Convex.ThreatModel.Cardano.Api (mockWalletHashes, txSigners)
+import Convex.Wallet (Wallet)
 import Convex.Wallet.MockWallet qualified as Wallet
 import Data.Aeson (ToJSON (..), (.=))
 import Data.Aeson qualified as Aeson
@@ -90,7 +92,7 @@ import Data.ByteString.Lazy.Char8 qualified as LBS
 import Data.Foldable (foldl', for_, traverse_)
 import Data.IORef (IORef, modifyIORef, newIORef, readIORef)
 import Data.Map qualified as Map
-import Data.Maybe (fromMaybe)
+import Data.Maybe (fromMaybe, mapMaybe)
 import Data.Set qualified as Set
 import Data.Word (Word32)
 import GHC.Generics (Generic)
@@ -110,6 +112,18 @@ import PlutusTx.Coverage (
 import Prettyprinter qualified as Pretty
 import System.Exit (ExitCode)
 
+{- | Detect which mock wallet signed a transaction by examining its witnesses.
+Returns an error message if no known mock wallet is found among the signers.
+-}
+detectSigningWallet :: C.Tx C.ConwayEra -> Either String Wallet
+detectSigningWallet tx =
+  case txSigners tx of
+    [] -> Left "Transaction has no signers — cannot determine wallet for threat model"
+    signers ->
+      case mapMaybe (\h -> lookup h mockWalletHashes) signers of
+        (w : _) -> Right w
+        [] -> Left "Transaction signers do not match any known mock wallet"
+
 {- | A testing interface defines the state and behavior of one or more smart contracts.
 
 The type parameter @state@ represents the model's view of the world. It should
@@ -460,9 +474,12 @@ positiveTest opts mGetTmResultsRef tms evs (Actions actions) = monadicIO $ do
             -- Run each threat model in an isolated MockchainT context
         liftIO $ forM allToRun $ \tm -> do
           let name = fromMaybe "Unnamed" (getThreatModelName tm)
-          (outcome, tmFinalState) <-
-            runMockchainIO (runThreatModelCheck Wallet.w1 tm [env]) params mcState
-          pure (name, outcome, mcsCoverageData tmFinalState)
+          case detectSigningWallet tx of
+            Left err -> pure (name, TMError err, mempty)
+            Right wallet -> do
+              (outcome, tmFinalState) <-
+                runMockchainIO (runThreatModelCheck wallet tm [env]) params mcState
+              pure (name, outcome, mcsCoverageData tmFinalState)
       _ -> pure []
 
     -- Extract just the (name, outcome) pairs for downstream processing