Fix user order spend

onomyprotocol · Jun 11, 2024 · 649ef61 · 649ef61
1 parent a58e73a
commit 649ef61
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/x/market/keeper/msg_server_simulate_test.go b/x/market/keeper/msg_server_simulate_test.go
@@ -86,7 +86,7 @@ func TestSimulate(t *testing.T) {
 
 	users := []User{user1, user2, user3}
 
-	for i := 1; i < 100; i++ {
+	for i := 1; i < 10000; i++ {
 
 		j := rand.IntN(6)
 		k := rand.IntN(2)
@@ -98,7 +98,9 @@ func TestSimulate(t *testing.T) {
 		userAddr, _ := sdk.AccAddressFromBech32(users[k].Addr)
 
 		spendable := testInput.BankKeeper.SpendableCoins(testInput.Context, userAddr)
-
+		if len(spendable) < 2 {
+			continue
+		}
 		l := rand.Uint64N(99)
 		if l == 0 {
 			continue
@@ -165,6 +167,12 @@ func TestSimulate(t *testing.T) {
 			// Cancel Order
 			if len(users[k].Orders) > 0 {
 				p := rand.IntN(len(users[k].Orders))
+				order, found := testInput.MarketKeeper.GetOrder(testInput.Context, users[k].Orders[p])
+				require.True(t, found)
+				if order.Status == "filled" || order.Status == "cancelled" {
+					users[k].Orders, _ = removeUid(users[k].Orders, users[k].Orders[p])
+					continue
+				}
 				co := types.MsgCancelOrder{Creator: addr, Uid: strconv.FormatUint(users[k].Orders[p], 10)}
 				_, err = keeper.NewMsgServerImpl(*testInput.MarketKeeper).CancelOrder(sdk.WrapSDKContext(testInput.Context), &co)
 				require.NoError(t, err)