chore: add unit test cases for payment module

x/payment/keeper/auto_resume_record_test.go

@@ -0,0 +1,52 @@
+package keeper_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/bnb-chain/greenfield/testutil/sample"
+	"github.com/bnb-chain/greenfield/x/payment/types"
+)
+
+func TestAutoResumeRecord(t *testing.T) {
+	keeper, ctx, _ := makePaymentKeeper(t)
+
+	addr1 := sample.RandAccAddress()
+	record1 := &types.AutoResumeRecord{
+		Addr:      addr1.String(),
+		Timestamp: 100,
+	}
+
+	addr2 := sample.RandAccAddress()
+	record2 := &types.AutoResumeRecord{
+		Addr:      addr2.String(),
+		Timestamp: 200,
+	}
+
+	// set
+	keeper.SetAutoResumeRecord(ctx, record1)
+	keeper.SetAutoResumeRecord(ctx, record2)
+
+	// exits
+	// before the timestamp
+	exist := keeper.ExistsAutoResumeRecord(ctx, 90, addr1)
+	require.True(t, !exist)
+	exist = keeper.ExistsAutoResumeRecord(ctx, 101, addr1)
+	require.True(t, exist)
+
+	// at any time
+	exist = keeper.ExistsAutoResumeRecord(ctx, 0, addr1)
+	require.True(t, exist)
+	exist = keeper.ExistsAutoResumeRecord(ctx, 0, addr2)
+	require.True(t, exist)
+
+	// remove
+	keeper.RemoveAutoResumeRecord(ctx, record1.Timestamp, addr1)
+	keeper.RemoveAutoResumeRecord(ctx, record2.Timestamp, addr2)
+
+	exist = keeper.ExistsAutoResumeRecord(ctx, 0, addr1)
+	require.True(t, !exist)
+	exist = keeper.ExistsAutoResumeRecord(ctx, 0, addr2)
+	require.True(t, !exist)
+}

x/payment/keeper/auto_settle_record.go

@@ -18,28 +18,6 @@ func (k Keeper) SetAutoSettleRecord(ctx sdk.Context, autoSettleRecord *types.Aut
 	), b)
 }
 
-// GetAutoSettleRecord returns a autoSettleRecord from its index
-func (k Keeper) GetAutoSettleRecord(
-	ctx sdk.Context,
-	timestamp int64,
-	addr sdk.AccAddress,
-) (*types.AutoSettleRecord, bool) {
-	store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AutoSettleRecordKeyPrefix)
-
-	b := store.Get(types.AutoSettleRecordKey(
-		timestamp,
-		addr,
-	))
-	if b == nil {
-		return nil, false
-	}
-
-	return &types.AutoSettleRecord{
-		Timestamp: timestamp,
-		Addr:      addr.String(),
-	}, true
-}
-
 // RemoveAutoSettleRecord removes a autoSettleRecord from the store
 func (k Keeper) RemoveAutoSettleRecord(
 	ctx sdk.Context,

x/payment/keeper/auto_settle_record_test.go

@@ -0,0 +1,42 @@
+package keeper_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/bnb-chain/greenfield/testutil/sample"
+	"github.com/bnb-chain/greenfield/x/payment/types"
+)
+
+func TestAutoSettleRecord(t *testing.T) {
+	keeper, ctx, _ := makePaymentKeeper(t)
+
+	addr1 := sample.RandAccAddress()
+	record1 := &types.AutoSettleRecord{
+		Addr:      addr1.String(),
+		Timestamp: 100,
+	}
+
+	addr2 := sample.RandAccAddress()
+	record2 := &types.AutoSettleRecord{
+		Addr:      addr2.String(),
+		Timestamp: 200,
+	}
+
+	// set
+	keeper.SetAutoSettleRecord(ctx, record1)
+	keeper.SetAutoSettleRecord(ctx, record2)
+
+	// update to new time
+	keeper.UpdateAutoSettleRecord(ctx, addr1, record1.Timestamp, 110)
+
+	// update to remove
+	keeper.UpdateAutoSettleRecord(ctx, addr2, record2.Timestamp, 0)
+
+	// get all
+	records := keeper.GetAllAutoSettleRecord(ctx)
+	require.True(t, len(records) == 1)
+	require.True(t, records[0].Addr == addr1.String())
+	require.True(t, records[0].Timestamp == 110)
+}