Add unit test for gas calculation

CosmWasm · Sep 6, 2023 · ee4a8bf · ee4a8bf
1 parent 543355a
commit ee4a8bf
Showing 1 changed file with 55 additions and 0 deletions.
diff --git a/x/wasm/keeper/keeper_test.go b/x/wasm/keeper/keeper_test.go
@@ -2432,6 +2432,61 @@ func TestSetContractAdmin(t *testing.T) {
 	}
 }
 
+func TestGasConsumed(t *testing.T) {
+	specs := map[string]struct {
+		originalMeter            stypes.GasMeter
+		gasRegister              WasmGasRegister
+		consumeGas               sdk.Gas
+		expPanic                 bool
+		expMultipliedGasConsumed uint64
+	}{
+		"all good": {
+			originalMeter:            sdk.NewGasMeter(100),
+			gasRegister:              NewWasmGasRegister(DefaultGasRegisterConfig()),
+			consumeGas:               sdk.Gas(1),
+			expMultipliedGasConsumed: 140000000,
+		},
+		"consumeGas = limit": {
+			originalMeter:            sdk.NewGasMeter(1),
+			gasRegister:              NewWasmGasRegister(DefaultGasRegisterConfig()),
+			consumeGas:               sdk.Gas(1),
+			expMultipliedGasConsumed: 140000000,
+		},
+		"consumeGas > limit": {
+			originalMeter: sdk.NewGasMeter(10),
+			gasRegister:   NewWasmGasRegister(DefaultGasRegisterConfig()),
+			consumeGas:    sdk.Gas(11),
+			expPanic:      true,
+		},
+		"nil original meter": {
+			gasRegister: NewWasmGasRegister(DefaultGasRegisterConfig()),
+			consumeGas:  sdk.Gas(1),
+			expPanic:    true,
+		},
+		"nil gas register": {
+			originalMeter: sdk.NewGasMeter(100),
+			consumeGas:    sdk.Gas(1),
+			expPanic:      true,
+		},
+	}
+
+	for name, spec := range specs {
+		t.Run(name, func(t *testing.T) {
+			m := NewMultipliedGasMeter(spec.originalMeter, spec.gasRegister)
+			if spec.expPanic {
+				assert.Panics(t, func() {
+					m.originalMeter.ConsumeGas(spec.consumeGas, "test-panic")
+					_ = m.GasConsumed()
+				})
+				return
+			}
+
+			m.originalMeter.ConsumeGas(spec.consumeGas, "test")
+			assert.Equal(t, spec.expMultipliedGasConsumed, m.GasConsumed())
+		})
+	}
+}
+
 func attrsToStringMap(attrs []abci.EventAttribute) map[string]string {
 	r := make(map[string]string, len(attrs))
 	for _, v := range attrs {