Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DEX whitelist test #1016

Merged
merged 16 commits into from
Oct 30, 2024
Merged

DEX whitelist test #1016

Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
b2a0eb8
Test DEX messages and queries in wasm
masihyeganeh Oct 18, 2024
9adcb3c
Use module_query_safe for gRPC query accept list
masihyeganeh Oct 21, 2024
a61bd6b
Update dependencies
masihyeganeh Oct 24, 2024
f0d197c
Fix FT bug
masihyeganeh Oct 24, 2024
376f0b5
Fix NFT bug
masihyeganeh Oct 24, 2024
0e8644b
Fix CI errors
masihyeganeh Oct 24, 2024
a79e723
Disable breakingProto
masihyeganeh Oct 24, 2024
f80f6c9
Fix linter errors
masihyeganeh Oct 24, 2024
56fbfa7
Fix naming in ExtensionIssueSettings message
masihyeganeh Oct 24, 2024
45027dc
Fix tests
masihyeganeh Oct 24, 2024
7295ebf
Fix lints
masihyeganeh Oct 24, 2024
39e3759
Fix unchecked nils
masihyeganeh Oct 24, 2024
dde384d
Gave up
masihyeganeh Oct 24, 2024
fecdb2d
Fix FT test
masihyeganeh Oct 25, 2024
202e014
Add whitelist test for DEX
masihyeganeh Oct 29, 2024
89ebed1
Merge branch 'master' into masih/dex-whitelist-test
masihyeganeh Oct 29, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
226 changes: 226 additions & 0 deletions integration-tests/modules/dex_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1841,6 +1841,232 @@ func TestLimitOrdersMatchingWithCommissionRate(t *testing.T) {
requireT.Equal(sdkmath.NewInt(100).String(), acc2Denom1BalanceRes.Balance.Amount.String())
}

// TestLimitOrdersMatchingWithAssetFTWhitelist tests the dex modules ability to place get and match limit orders
// with asset ft with whitelisting.
func TestLimitOrdersMatchingWithAssetFTWhitelist(t *testing.T) {
t.Parallel()
ctx, chain := integrationtests.NewCoreumTestingContext(t)

requireT := require.New(t)
bankClient := banktypes.NewQueryClient(chain.ClientContext)
assetFTClient := assetfttypes.NewQueryClient(chain.ClientContext)
dexClient := dextypes.NewQueryClient(chain.ClientContext)

dexParamsRes, err := dexClient.Params(ctx, &dextypes.QueryParamsRequest{})
requireT.NoError(err)

issuer := chain.GenAccount()
chain.FundAccountWithOptions(ctx, t, issuer, integration.BalancesOptions{
Messages: []sdk.Msg{
&assetfttypes.MsgSetWhitelistedLimit{},
&assetfttypes.MsgSetWhitelistedLimit{},
&banktypes.MsgSend{},
&assetfttypes.MsgSetWhitelistedLimit{},
&dextypes.MsgPlaceOrder{},
},
Amount: dexParamsRes.Params.OrderReserve.Amount,
})

acc1 := chain.GenAccount()
chain.FundAccountWithOptions(ctx, t, acc1, integration.BalancesOptions{
Messages: []sdk.Msg{
&dextypes.MsgPlaceOrder{},
},
Amount: dexParamsRes.Params.OrderReserve.Amount,
})

acc2 := chain.GenAccount()
chain.FundAccountWithOptions(ctx, t, acc2, integration.BalancesOptions{
Messages: []sdk.Msg{
&dextypes.MsgPlaceOrder{},
&dextypes.MsgPlaceOrder{},
},
Amount: dexParamsRes.Params.OrderReserve.Amount.MulRaw(2),
})

denom1 := issueFT(ctx, t, chain, issuer, sdkmath.NewIntWithDecimal(1, 6), assetfttypes.Feature_whitelisting)
denom2 := issueFT(ctx, t, chain, acc2, sdkmath.NewIntWithDecimal(1, 6))

// whitelist denom1 for acc1
msgSetWhitelistedLimit := &assetfttypes.MsgSetWhitelistedLimit{
Sender: issuer.String(),
Account: acc1.String(),
Coin: sdk.NewCoin(denom1, sdkmath.NewInt(1000000)),
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(issuer),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(msgSetWhitelistedLimit)),
msgSetWhitelistedLimit,
)
requireT.NoError(err)

msgSend := &banktypes.MsgSend{
FromAddress: issuer.String(),
ToAddress: acc1.String(),
Amount: sdk.NewCoins(
sdk.NewCoin(denom1, sdkmath.NewInt(150)),
),
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(issuer),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(msgSend)),
msgSend,
)
requireT.NoError(err)

balanceRes, err := assetFTClient.Balance(ctx, &assetfttypes.QueryBalanceRequest{
Account: acc1.String(),
Denom: denom1,
})
requireT.NoError(err)
requireT.Equal(sdkmath.NewInt(150).String(), balanceRes.Balance.String())

// place order should fail because acc2 is out of whitelisted coins
placeSellOrderMsg := &dextypes.MsgPlaceOrder{
Sender: acc2.String(),
Type: dextypes.ORDER_TYPE_LIMIT,
ID: "id1",
BaseDenom: denom1,
QuoteDenom: denom2,
Price: lo.ToPtr(dextypes.MustNewPriceFromString("11e-2")),
Quantity: sdkmath.NewInt(300),
Side: dextypes.SIDE_BUY,
TimeInForce: dextypes.TIME_IN_FORCE_GTC,
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(acc2),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(placeSellOrderMsg)),
placeSellOrderMsg,
)
requireT.ErrorContains(err, assetfttypes.ErrWhitelistedLimitExceeded.Error())

balanceRes, err = assetFTClient.Balance(ctx, &assetfttypes.QueryBalanceRequest{
Account: acc2.String(),
Denom: denom1,
})
requireT.NoError(err)
requireT.Equal(sdkmath.NewInt(0).String(), balanceRes.LockedInDEX.String())

msgSetWhitelistedLimit = &assetfttypes.MsgSetWhitelistedLimit{
Sender: issuer.String(),
Account: acc2.String(),
Coin: sdk.NewCoin(denom1, sdkmath.NewInt(300)),
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(issuer),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(msgSetWhitelistedLimit)),
msgSetWhitelistedLimit,
)
requireT.NoError(err)

// now placing order should succeed because the receiving amount is within the whitelist limit
placeBuyOrderMsg := &dextypes.MsgPlaceOrder{
Sender: acc2.String(),
Type: dextypes.ORDER_TYPE_LIMIT,
ID: "id1",
BaseDenom: denom1,
QuoteDenom: denom2,
Price: lo.ToPtr(dextypes.MustNewPriceFromString("11e-2")),
Quantity: sdkmath.NewInt(300),
Side: dextypes.SIDE_BUY,
TimeInForce: dextypes.TIME_IN_FORCE_GTC,
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(acc2),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(placeBuyOrderMsg)),
placeBuyOrderMsg,
)
requireT.NoError(err)

balanceRes, err = assetFTClient.Balance(ctx, &assetfttypes.QueryBalanceRequest{
Account: acc2.String(),
Denom: denom2,
})
requireT.NoError(err)
requireT.Equal("33", balanceRes.LockedInDEX.String())

// Reducing the whitelist limit will not interfere with DEX order after order placement
msgSetWhitelistedLimit = &assetfttypes.MsgSetWhitelistedLimit{
Sender: issuer.String(),
Account: acc2.String(),
Coin: sdk.NewCoin(denom1, sdkmath.NewInt(0)),
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(issuer),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(msgSetWhitelistedLimit)),
msgSetWhitelistedLimit,
)
requireT.NoError(err)

// place sell order to match the buy
placeSellOrderMsg = &dextypes.MsgPlaceOrder{
Sender: acc1.String(),
Type: dextypes.ORDER_TYPE_LIMIT,
ID: "id1", // same ID allowed for different user
BaseDenom: denom1,
QuoteDenom: denom2,
Price: lo.ToPtr(dextypes.MustNewPriceFromString("1e-1")),
Quantity: sdkmath.NewInt(100),
Side: dextypes.SIDE_SELL,
TimeInForce: dextypes.TIME_IN_FORCE_GTC,
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(acc1),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(placeSellOrderMsg)),
placeSellOrderMsg,
)
requireT.NoError(err)

acc1Denom2BalanceRes, err := bankClient.Balance(ctx, &banktypes.QueryBalanceRequest{
Address: acc1.String(),
Denom: denom2,
})
requireT.NoError(err)
requireT.Equal(sdkmath.NewInt(11).String(), acc1Denom2BalanceRes.Balance.Amount.String())

acc2Denom1BalanceRes, err := bankClient.Balance(ctx, &banktypes.QueryBalanceRequest{
Address: acc2.String(),
Denom: denom1,
})
requireT.NoError(err)
requireT.Equal(sdkmath.NewInt(100).String(), acc2Denom1BalanceRes.Balance.Amount.String())

// place order should succeed as issuer because admin (issuer) doesn't have whitelist limit
placeSellOrderMsg = &dextypes.MsgPlaceOrder{
Sender: issuer.String(),
Type: dextypes.ORDER_TYPE_LIMIT,
ID: "id1",
BaseDenom: denom1,
QuoteDenom: denom2,
Price: lo.ToPtr(dextypes.MustNewPriceFromString("1e-1")),
Quantity: sdkmath.NewInt(100),
Side: dextypes.SIDE_SELL,
TimeInForce: dextypes.TIME_IN_FORCE_GTC,
}

_, err = client.BroadcastTx(
ctx,
chain.ClientContext.WithFromAddress(issuer),
chain.TxFactory().WithGas(chain.GasLimitByMsgs(placeSellOrderMsg)),
placeSellOrderMsg,
)
requireT.NoError(err)
}

// TestCancelOrdersByDenom tests the dex modules ability to cancel all orders of the account and by denom.
func TestCancelOrdersByDenom(t *testing.T) {
t.Parallel()
Expand Down