Handle more State/Event pairs

Property is still failing because we do not hanlde CloseTx in all states

hydra-node/src/Hydra/HeadLogic.hs

@@ -200,7 +200,7 @@ update Environment{party, snapshotStrategy} ledger (HeadState p st) ev = case (s
      in if canCollectCom party pt remainingTokens'
           then newState p newHeadState [OnChainEffect CollectComTx]
           else newState p newHeadState []
-  (CollectingState{}, OnChainEvent CollectComTx) ->
+  (_, OnChainEvent CollectComTx) ->
     let u0 = initUTxO ledger -- TODO(SN): should construct u0 from the collected utxo
      in newState
           p
@@ -277,7 +277,7 @@ update Environment{party, snapshotStrategy} ledger (HeadState p st) ev = case (s
   --
   (ClosedState utxo, ShouldPostFanout) ->
     newState p st [OnChainEffect (FanoutTx utxo)]
-  (ClosedState{}, OnChainEvent (FanoutTx utxo)) ->
+  (_, OnChainEvent (FanoutTx utxo)) ->
     -- NOTE(SN): we might care if we are not in ClosedState
     newState p FinalState [ClientEffect $ HeadIsFinalized utxo]
   --

hydra-node/test/Hydra/HeadLogicSpec.hs

@@ -17,7 +17,7 @@ import Hydra.HeadLogic (
   HeadState (..),
   HeadStatus (..),
   HydraMessage (..),
-  OnChainTx (InitTx),
+  OnChainTx (..),
   Outcome (..),
   SimpleHeadState (..),
   Snapshot (..),
@@ -35,7 +35,7 @@ import Test.Hspec (
  )
 import Test.Hspec.Core.Spec (pending)
 import Test.Hspec.QuickCheck (prop)
-import Test.QuickCheck (Gen, Property, forAll, oneof)
+import Test.QuickCheck (Gen, Property, elements, forAll)
 import Test.QuickCheck.Instances.Time ()
 import Test.QuickCheck.Property (collect)
 
@@ -77,11 +77,23 @@ spec = describe "Hydra Head Logic" $ do
 
 genOnChainTx :: Gen (OnChainTx MockTx)
 genOnChainTx =
-  oneof
-    [pure $ InitTx mempty]
+  elements
+    [ InitTx mempty
+    , -- TODO: maybe handle it differently? , CommitTx (ParticipationToken 1 1) 10
+      CollectComTx
+    , CloseTx
+    , ContestTx
+    , FanoutTx [ValidTx 1]
+    ]
 
 genHeadStatus :: Gen (HeadStatus MockTx)
-genHeadStatus = oneof [pure InitState, pure FinalState]
+genHeadStatus =
+  elements
+    [ InitState
+    , FinalState
+    , CollectingState mempty mempty
+    , OpenState (SimpleHeadState [] mempty mempty (Snapshot 0 mempty mempty))
+    ]
 
 defaultHeadParameters :: HeadParameters
 defaultHeadParameters =