Add explicit states for TestState wallet flavors.

cardano-foundation · May 26, 2023 · 371e45f · 371e45f
1 parent c9f170c
commit 371e45f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/lib/wallet/src/Cardano/Wallet/Flavor.hs b/lib/wallet/src/Cardano/Wallet/Flavor.hs
@@ -76,7 +76,8 @@ data WalletFlavors
     | SharedF
     | BenchByronF
     | BenchShelleyF
-    | TestStateF
+    | TestAddressesWithCredsF
+    | TestOnlyAddressesF
 
 type family FlavorOf s where
     FlavorOf (SeqState n ShelleyKey) = 'ShelleyF
@@ -94,7 +95,8 @@ type AllFlavors =
      , 'SharedF
      , 'BenchByronF
      , 'BenchShelleyF
-     , 'TestStateF
+     , 'TestAddressesWithCredsF
+     , 'TestOnlyAddressesF
      ]
 
 type IncludingStates ss s = Including AllFlavors ss s

diff --git a/lib/wallet/test/unit/Cardano/WalletSpec.hs b/lib/wallet/test/unit/Cardano/WalletSpec.hs
@@ -44,13 +44,7 @@ import Cardano.Wallet
     , throttle
     )
 import Cardano.Wallet.Address.Derivation
-    ( Depth (..)
-    , DerivationIndex (..)
-    , DerivationType (..)
-    , HardDerivation (..)
-    , Index
-    , Role (..)
-    )
+    ( Depth (..), DerivationIndex (..), HardDerivation (..), Role (..) )
 import Cardano.Wallet.Address.Derivation.Shelley
     ( ShelleyKey (..), generateKeyFromSeed )
 import Cardano.Wallet.Address.Discovery
@@ -202,8 +196,6 @@ import Data.List
     ( nubBy, sort, sortOn )
 import Data.List.NonEmpty
     ( NonEmpty (..) )
-import Data.Map.Strict
-    ( Map )
 import Data.Maybe
     ( catMaybes, fromMaybe, isJust, isNothing, mapMaybe )
 import Data.Ord