fix: nil expirations no longer pruned

x/ecocredit/server/marketplace/prune.go

@@ -24,16 +24,26 @@ func (k Keeper) PruneSellOrders(ctx context.Context) error {
 		return err
 	}
 	defer it.Close()
+
+	expiredOrderIds := make([]uint64, 0)
 	for it.Next() {
 		sellOrder, err := it.Value()
 		if err != nil {
 			return err
 		}
-		if err = k.unescrowCredits(ctx, sellOrder.Seller, sellOrder.BatchId, sellOrder.Quantity); err != nil {
+		if sellOrder.Expiration != nil {
+			if err = k.unescrowCredits(ctx, sellOrder.Seller, sellOrder.BatchId, sellOrder.Quantity); err != nil {
+				return err
+			}
+			expiredOrderIds = append(expiredOrderIds, sellOrder.Id)
+		}
+	}
+	for _, id := range expiredOrderIds {
+		if err := k.stateStore.SellOrderTable().Delete(ctx, &marketplaceapi.SellOrder{Id: id}); err != nil {
 			return err
 		}
 	}
-	return k.stateStore.SellOrderTable().DeleteRange(ctx, fromKey, toKey)
+	return nil
 }
 
 // unescrowCredits moves `amount` of credits from the sellerAddr's escrowed balance, into their tradable balance.

x/ecocredit/server/marketplace/prune_test.go

@@ -66,3 +66,41 @@ func TestSell_Prune(t *testing.T) {
 	_, err = s.marketStore.SellOrderTable().Get(s.ctx, shouldBeValid)
 	assert.NilError(t, err)
 }
+
+// TestPrune_NilExpiration tests that sell orders with nil expirations are not deleted when PruneOrders is called.
+func TestPrune_NilExpiration(t *testing.T) {
+	t.Parallel()
+	s := setupBase(t)
+	s.testSellSetup(batchDenom, ask.Denom, ask.Denom[1:], "C01", start, end, creditType)
+
+	blockTime, err := time.Parse("2006-01-02", "2020-01-01")
+	assert.NilError(t, err)
+	expired, err := time.Parse("2006-01-02", "2010-01-01")
+	assert.NilError(t, err)
+
+	msg := &marketplace.MsgSell{
+		Owner: s.addr.String(),
+		Orders: []*marketplace.MsgSell_Order{
+			{BatchDenom: batchDenom, Quantity: "5", AskPrice: &ask, Expiration: nil},
+			{BatchDenom: batchDenom, Quantity: "10", AskPrice: &ask, Expiration: &expired},
+		},
+	}
+	res, err := s.k.Sell(s.ctx, msg)
+	assert.NilError(t, err)
+
+	shouldExistOrder := res.SellOrderIds[0]
+	shouldNotExistOrder := res.SellOrderIds[1]
+
+	s.sdkCtx = s.sdkCtx.WithBlockTime(blockTime)
+	s.ctx = sdk.WrapSDKContext(s.sdkCtx)
+
+	err = s.k.PruneSellOrders(s.ctx)
+	assert.NilError(t, err)
+
+	order, err := s.marketStore.SellOrderTable().Get(s.ctx, shouldExistOrder)
+	assert.NilError(t, err)
+	assert.Equal(t, "5", order.Quantity)
+
+	order, err = s.marketStore.SellOrderTable().Get(s.ctx, shouldNotExistOrder)
+	assert.ErrorIs(t, err, ormerrors.NotFound)
+}