Integration test: Delegating addresses API roundtrip

lib/core-integration/src/Test/Integration/Scenario/API/Shelley/Addresses.hs

@@ -17,18 +17,21 @@ import Cardano.Wallet.Api.Types
     ( AnyAddress
     , ApiAddress
     , ApiTransaction
+    , ApiVerificationKey
     , ApiWallet
     , DecodeAddress
     , DecodeStakeAddress
     , EncodeAddress
     , WalletStyle (..)
     )
+import Cardano.Wallet.Primitive.AddressDerivation
+    ( AccountingStyle (..), DerivationIndex (..) )
 import Cardano.Wallet.Primitive.AddressDiscovery.Sequential
     ( defaultAddressPoolGap, getAddressPoolGap )
 import Cardano.Wallet.Primitive.Types.Address
     ( AddressState (..) )
 import Control.Monad
-    ( forM_ )
+    ( forM, forM_ )
 import Control.Monad.Trans.Resource
     ( runResourceT )
 import Data.Aeson
@@ -62,6 +65,7 @@ import Test.Integration.Framework.DSL
     , listAddresses
     , minUTxOValue
     , request
+    , unsafeRequest
     , verify
     , walletId
     )
@@ -73,6 +77,7 @@ import qualified Data.Aeson as Aeson
 import qualified Data.Aeson.Lens as Aeson
 import qualified Data.Text as T
 import qualified Network.HTTP.Types.Status as HTTP
+import qualified Test.Hspec.Expectations.Lifted as Expectations
 
 spec :: forall n t.
     ( DecodeAddress n
@@ -519,6 +524,33 @@ spec = describe "SHELLEY_ADDRESSES" $ do
                 "addr1y9qthemrg5kczwfjjnahwt65elhrl95e9hcgufnajtp6wfffxzlhgvlzh\
                 \6drdsm04j43jk2wpsnqw7uketsgelghm3lsch4g8t" :: Text
         validateAddr r goldenAddr
+
+    it "ANY_ADDRESS_POST_12 - Delegating addresses API roundtrip" $ \ctx -> runResourceT $ do
+        w <- emptyWallet ctx
+
+        -- Generate first 20 addresses using payment and stake keys derived from
+        -- wallet API
+        let indexes = [0..19]
+        generatedAddresses <- forM indexes $ \index -> do
+            let paymentPath = Link.getWalletKey w UtxoExternal (DerivationIndex index)
+            (_, paymentKey) <- unsafeRequest @ApiVerificationKey ctx paymentPath Empty
+
+            let stakePath = Link.getWalletKey w MutableAccount (DerivationIndex 0)
+            (_, stakeKey) <- unsafeRequest @ApiVerificationKey ctx stakePath Empty
+
+            let payload = Json [json|{
+                    "payment": #{paymentKey},
+                    "stake": #{stakeKey}
+                }|]
+            (_, addr) <- unsafeRequest @AnyAddress ctx Link.postAnyAddress payload
+            pure (addr ^. #payload)
+
+        -- Make sure the same addresses are already available in the wallet
+        addrs <- listAddresses @n ctx w
+        forM_ (zip (fmap fromIntegral indexes) generatedAddresses)
+            $ \(idx, genAddr) -> do
+                let walAddr = fst (addrs !! idx ^. #id) ^. (#getApiT . #unAddress)
+                walAddr `Expectations.shouldBe` genAddr
   where
     validateAddr resp expected = do
         let addr = getFromResponse id resp