From a5dd494aaab8072e9fa7ea7f3bbfe4b1fc0ae7a8 Mon Sep 17 00:00:00 2001
From: Ethan Frey <ethanfrey@users.noreply.github.com>
Date: Wed, 27 Apr 2022 21:30:32 +0200
Subject: [PATCH] Start testing snapshots

---
 x/wasm/keeper/snapshotter_test.go | 62 +++++++++++++++++++++++++++++++
 x/wasm/keeper/test_common.go      |  2 +
 2 files changed, 64 insertions(+)
 create mode 100644 x/wasm/keeper/snapshotter_test.go

diff --git a/x/wasm/keeper/snapshotter_test.go b/x/wasm/keeper/snapshotter_test.go
new file mode 100644
index 00000000..341a40ca
--- /dev/null
+++ b/x/wasm/keeper/snapshotter_test.go
@@ -0,0 +1,62 @@
+package keeper
+
+import (
+	"encoding/json"
+	"fmt"
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	dbm "github.com/tendermint/tm-db"
+
+	"github.com/CosmWasm/wasmd/x/wasm/types"
+)
+
+func TestSnapshotRestoreMovesFile(t *testing.T) {
+	// we hack this to "fake" copying over all the iavl data
+	sharedDB := dbm.NewMemDB()
+
+	ctx, keepers := createTestInput(t, false, SupportedFeatures, types.DefaultWasmConfig(), sharedDB)
+	deposit := sdk.NewCoins(sdk.NewInt64Coin("denom", 100000))
+	creator := keepers.Faucet.NewFundedAccount(ctx, deposit...)
+	_, _, bob := keyPubAddr()
+	_, _, fred := keyPubAddr()
+
+	// create a contact
+	codeID, err := keepers.ContractKeeper.Create(ctx, creator, hackatomWasm, nil)
+	require.NoError(t, err)
+	require.Equal(t, uint64(1), codeID)
+
+	// instantiate it
+	initMsg := HackatomExampleInitMsg{
+		Verifier:    fred,
+		Beneficiary: bob,
+	}
+	initMsgBz, err := json.Marshal(initMsg)
+	require.NoError(t, err)
+	contractAddr, _, err := keepers.ContractKeeper.Instantiate(ctx, codeID, creator, nil, initMsgBz, "demo contract 1", deposit)
+	require.NoError(t, err)
+	require.Equal(t, "cosmos14hj2tavq8fpesdwxxcu44rty3hh90vhujrvcmstl4zr3txmfvw9s4hmalr", contractAddr.String())
+
+	// successfully query it
+	queryBz := []byte(`{"verifier":{}}`)
+	res, err := keepers.WasmKeeper.QuerySmart(ctx, contractAddr, queryBz)
+	require.NoError(t, err)
+	expected := fmt.Sprintf(`{"verifier":"%s"}`, fred.String())
+	assert.JSONEq(t, string(res), expected)
+
+	// failed attempt to copy state
+	// // now, we make a new app with a copy of the "iavl" db, but no contracts
+	// copyCtx, copyKeepers := createTestInput(t, false, SupportedFeatures, types.DefaultWasmConfig(), sharedDB)
+
+	// // contract exists
+	// info := copyKeepers.WasmKeeper.GetContractInfo(ctx, contractAddr)
+	// require.NotNil(t, info)
+	// require.Equal(t, info.CodeID, codeID)
+
+	// // querying the existing contract errors, as there is no wasm file
+	// res, err = copyKeepers.WasmKeeper.QuerySmart(copyCtx, contractAddr, queryBz)
+	// require.Error(t, err)
+
+}
diff --git a/x/wasm/keeper/test_common.go b/x/wasm/keeper/test_common.go
index d609771c..4cd92917 100644
--- a/x/wasm/keeper/test_common.go
+++ b/x/wasm/keeper/test_common.go
@@ -178,6 +178,7 @@ type TestKeepers struct {
 	Router         *baseapp.Router
 	EncodingConfig wasmappparams.EncodingConfig
 	Faucet         *TestFaucet
+	MultiStore     sdk.MultiStore
 }
 
 // CreateDefaultTestInput common settings for CreateTestInput
@@ -431,6 +432,7 @@ func createTestInput(
 		Router:         router,
 		EncodingConfig: encodingConfig,
 		Faucet:         faucet,
+		MultiStore:     ms,
 	}
 	return ctx, keepers
 }