Add tests for pruning wrapper.

app/store_loader_test.go

@@ -2,11 +2,13 @@ package app
 
 import (
 	"testing"
+	"time"
 
 	"github.com/cosmos/cosmos-sdk/baseapp"
 	"github.com/cosmos/cosmos-sdk/store/rootmulti"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/stretchr/testify/assert"
+	"github.com/tendermint/tendermint/libs/log"
 	dbm "github.com/tendermint/tm-db"
 )
 
@@ -58,7 +60,39 @@ func TestWrapStoreLoader(t *testing.T) {
 }
 
 func TestPruningWrapper(t *testing.T) {
+	tests := []struct {
+		name    string
+		pruning string
+		delta   uint64
+	}{
+		{
+			name:    "recommended pruning should not wait",
+			pruning: "13",
+			delta:   0,
+		},
+		{
+			name:    "non-recommended pruning should wait",
+			pruning: "1000",
+			delta:   30,
+		},
+	}
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			logger := log.NewNopLogger()
+			appOpts := MockAppOptions{pruning: tc.pruning}
+			storeLoader := PruningWrapper(logger, appOpts, createMockStoreLoader())
+			db := dbm.GoLevelDB{}
+			ms := rootmulti.NewStore(&db, nil)
+			assert.NotNil(t, ms, "should create a new multistore for testing")
 
+			start := time.Now()
+			err := storeLoader(ms)
+			delta := uint64(time.Now().Sub(start).Seconds())
+			assert.NoError(t, err, "should not throw error")
+			assert.GreaterOrEqual(t, delta, tc.delta, "should wait with non recommended pruning")
+		})
+	}
 }
 
 func createMockStoreLoader() baseapp.StoreLoader {
@@ -80,3 +114,11 @@ func createMockStoreWrapper(flag *bool) StoreLoaderWrapper {
 		return nil
 	}
 }
+
+type MockAppOptions struct {
+	pruning string
+}
+
+func (m MockAppOptions) Get(string) interface{} {
+	return m.pruning
+}