Skip to content

Commit

Permalink
rename wallet to migrate in latency benchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
Piotr Stachyra committed May 25, 2021
1 parent 2c31861 commit c32dafc
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions lib/shelley/bench/Latency.hs
Original file line number Diff line number Diff line change
Expand Up @@ -204,16 +204,16 @@ walletApiBench capture ctx = do
nFixtureWallet n = do
wal1 : wal2 : _ <- replicateM n (fixtureWallet ctx)
walMA <- fixtureMultiAssetWallet ctx
walMAForMigration <- fixtureMultiAssetWallet ctx
pure (wal1, wal2, walMA, walMAForMigration)
maWalletToMigrate <- fixtureMultiAssetWallet ctx
pure (wal1, wal2, walMA, maWalletToMigrate)

-- Creates n fixture wallets and send 1-ada transactions to one of them
-- (m times). The money is sent in batches (see batchSize below) from
-- additionally created source fixture wallet. Then we wait for the money
-- to be accommodated in recipient wallet. After that the source fixture
-- wallet is removed.
nFixtureWalletWithTxs n m = do
(wal1, wal2, walMA, walMAForMigration) <- nFixtureWallet n
(wal1, wal2, walMA, maWalletToMigrate) <- nFixtureWallet n

let amt = minUTxOValue
let batchSize = 10
Expand All @@ -228,12 +228,12 @@ walletApiBench capture ctx = do
let expInflows' = filter (/=0) expInflows

mapM_ (repeatPostTx wal1 amt batchSize . amtExp) expInflows'
pure (wal1, wal2, walMA, walMAForMigration)
pure (wal1, wal2, walMA, maWalletToMigrate)

nFixtureWalletWithUTxOs n utxoNumber = do
let utxoExp = replicate utxoNumber minUTxOValue
wal1 <- fixtureWalletWith @n ctx utxoExp
(_, wal2, walMA, walMAForMigration) <- nFixtureWallet n
(_, wal2, walMA, maWalletToMigrate) <- nFixtureWallet n

eventually "Wallet balance is as expected" $ do
rWal1 <- request @ApiWallet ctx
Expand All @@ -249,7 +249,7 @@ walletApiBench capture ctx = do
(Link.getUTxOsStatistics @'Shelley wal1) Default Empty
expectResponseCode HTTP.status200 rStat
expectWalletUTxO (fromIntegral <$> utxoExp) (snd rStat)
pure (wal1, wal2, walMA, walMAForMigration)
pure (wal1, wal2, walMA, maWalletToMigrate)

repeatPostTx wDest amtToSend batchSize amtExp = do
wSrc <- fixtureWallet ctx
Expand Down Expand Up @@ -285,7 +285,7 @@ walletApiBench capture ctx = do
expectResponseCode HTTP.status202 r
return r

runScenario scenario = runResourceT $ scenario >>= \(wal1, wal2, walMA, walMAForMigration) -> liftIO $ do
runScenario scenario = runResourceT $ scenario >>= \(wal1, wal2, walMA, maWalletToMigrate) -> liftIO $ do
t1 <- measureApiLogs capture
(request @[ApiWallet] ctx (Link.listWallets @'Shelley) Default Empty)
fmtResult "listWallets " t1
Expand Down Expand Up @@ -387,14 +387,14 @@ walletApiBench capture ctx = do
fmtResult "getMultiAsset " t11

-- Create a migration plan:
let endpointPlan = (Link.createMigrationPlan @'Shelley walMAForMigration)
let endpointPlan = (Link.createMigrationPlan @'Shelley maWalletToMigrate)
t12a <- measureApiLogs capture $ request @(ApiWalletMigrationPlan n)
ctx endpointPlan Default $
Json [json|{addresses: #{addresses}}|]
fmtResult "postMigrationPlan " t12a

-- Perform a migration:
let endpointMigrate = Link.migrateWallet @'Shelley walMAForMigration
let endpointMigrate = Link.migrateWallet @'Shelley maWalletToMigrate
t12b <- measureApiLogs capture $ request @[ApiTransaction n]
ctx endpointMigrate Default $
Json [json|
Expand Down

0 comments on commit c32dafc

Please sign in to comment.