Fix test data generator (#1263)

CosmWasm · Mar 15, 2023 · 2eaa893 · 2eaa893
1 parent d648261
commit 2eaa893
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/x/wasm/keeper/genesis_test.go b/x/wasm/keeper/genesis_test.go
@@ -83,7 +83,8 @@ func TestGenesisExportImport(t *testing.T) {
 		contractAddr := wasmKeeper.ClassicAddressGenerator()(srcCtx, codeID, nil)
 		wasmKeeper.storeContractInfo(srcCtx, contractAddr, &contract)
 		wasmKeeper.appendToContractHistory(srcCtx, contractAddr, history...)
-		wasmKeeper.importContractState(srcCtx, contractAddr, stateModels)
+		err = wasmKeeper.importContractState(srcCtx, contractAddr, stateModels)
+		require.NoError(t, err)
 	}
 	var wasmParams types.Params
 	f.NilChance(0).Fuzz(&wasmParams)

diff --git a/x/wasm/keeper/test_fuzz.go b/x/wasm/keeper/test_fuzz.go
@@ -51,10 +51,12 @@ func FuzzContractCodeHistory(m *types.ContractCodeHistoryEntry, c fuzz.Continue)
 
 func FuzzStateModel(m *types.Model, c fuzz.Continue) {
 	m.Key = tmBytes.HexBytes(c.RandString())
-	if len(m.Key) == 0 {
-		m.Key = tmBytes.HexBytes("non empty key")
+	if len(m.Key) != 0 {
+		c.Fuzz(&m.Value)
+		return
 	}
-	c.Fuzz(&m.Value)
+	// try again, keys must not be empty
+	FuzzStateModel(m, c)
 }
 
 func FuzzAccessType(m *types.AccessType, c fuzz.Continue) {