2018-08-06 12:00:49 -07:00
|
|
|
package types
|
|
|
|
|
|
|
|
import (
|
2019-02-01 17:03:09 -08:00
|
|
|
"math"
|
2018-08-06 12:00:49 -07:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
2020-03-01 16:27:03 -08:00
|
|
|
func TestInfiniteGasMeter(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
meter := NewInfiniteGasMeter()
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, uint64(math.MaxUint64), meter.Limit())
|
|
|
|
require.Equal(t, uint64(math.MaxUint64), meter.GasRemaining())
|
2020-03-01 16:27:03 -08:00
|
|
|
require.Equal(t, uint64(0), meter.GasConsumed())
|
|
|
|
require.Equal(t, uint64(0), meter.GasConsumedToLimit())
|
|
|
|
meter.ConsumeGas(10, "consume 10")
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, uint64(math.MaxUint64), meter.GasRemaining())
|
2020-03-01 16:27:03 -08:00
|
|
|
require.Equal(t, uint64(10), meter.GasConsumed())
|
|
|
|
require.Equal(t, uint64(10), meter.GasConsumedToLimit())
|
2021-06-02 08:14:15 -07:00
|
|
|
meter.RefundGas(1, "refund 1")
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, uint64(math.MaxUint64), meter.GasRemaining())
|
2021-06-02 08:14:15 -07:00
|
|
|
require.Equal(t, uint64(9), meter.GasConsumed())
|
2020-03-01 16:27:03 -08:00
|
|
|
require.False(t, meter.IsPastLimit())
|
|
|
|
require.False(t, meter.IsOutOfGas())
|
|
|
|
meter.ConsumeGas(Gas(math.MaxUint64/2), "consume half max uint64")
|
|
|
|
require.Panics(t, func() { meter.ConsumeGas(Gas(math.MaxUint64/2)+2, "panic") })
|
2021-06-02 08:14:15 -07:00
|
|
|
require.Panics(t, func() { meter.RefundGas(meter.GasConsumed()+1, "refund greater than consumed") })
|
2020-03-01 16:27:03 -08:00
|
|
|
}
|
|
|
|
|
2018-08-06 12:00:49 -07:00
|
|
|
func TestGasMeter(t *testing.T) {
|
2020-03-01 16:27:03 -08:00
|
|
|
t.Parallel()
|
2018-08-06 12:00:49 -07:00
|
|
|
cases := []struct {
|
|
|
|
limit Gas
|
|
|
|
usage []Gas
|
|
|
|
}{
|
|
|
|
{10, []Gas{1, 2, 3, 4}},
|
|
|
|
{1000, []Gas{40, 30, 20, 10, 900}},
|
|
|
|
{100000, []Gas{99999, 1}},
|
|
|
|
{100000000, []Gas{50000000, 40000000, 10000000}},
|
|
|
|
{65535, []Gas{32768, 32767}},
|
|
|
|
{65536, []Gas{32768, 32767, 1}},
|
|
|
|
}
|
|
|
|
|
|
|
|
for tcnum, tc := range cases {
|
|
|
|
meter := NewGasMeter(tc.limit)
|
2018-11-19 09:13:45 -08:00
|
|
|
used := uint64(0)
|
2018-08-06 12:00:49 -07:00
|
|
|
|
|
|
|
for unum, usage := range tc.usage {
|
2019-10-17 06:47:35 -07:00
|
|
|
usage := usage
|
2018-08-06 12:00:49 -07:00
|
|
|
used += usage
|
|
|
|
require.NotPanics(t, func() { meter.ConsumeGas(usage, "") }, "Not exceeded limit but panicked. tc #%d, usage #%d", tcnum, unum)
|
|
|
|
require.Equal(t, used, meter.GasConsumed(), "Gas consumption not match. tc #%d, usage #%d", tcnum, unum)
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, tc.limit-used, meter.GasRemaining(), "Gas left not match. tc #%d, usage #%d", tcnum, unum)
|
2018-11-20 20:07:30 -08:00
|
|
|
require.Equal(t, used, meter.GasConsumedToLimit(), "Gas consumption (to limit) not match. tc #%d, usage #%d", tcnum, unum)
|
|
|
|
require.False(t, meter.IsPastLimit(), "Not exceeded limit but got IsPastLimit() true")
|
|
|
|
if unum < len(tc.usage)-1 {
|
|
|
|
require.False(t, meter.IsOutOfGas(), "Not yet at limit but got IsOutOfGas() true")
|
|
|
|
} else {
|
|
|
|
require.True(t, meter.IsOutOfGas(), "At limit but got IsOutOfGas() false")
|
|
|
|
}
|
2018-08-06 12:00:49 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
require.Panics(t, func() { meter.ConsumeGas(1, "") }, "Exceeded but not panicked. tc #%d", tcnum)
|
2018-11-20 20:07:30 -08:00
|
|
|
require.Equal(t, meter.GasConsumedToLimit(), meter.Limit(), "Gas consumption (to limit) not match limit")
|
|
|
|
require.Equal(t, meter.GasConsumed(), meter.Limit()+1, "Gas consumption not match limit+1")
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, uint64(0), meter.GasRemaining())
|
2021-06-02 08:14:15 -07:00
|
|
|
|
|
|
|
require.NotPanics(t, func() { meter.RefundGas(1, "refund 1") })
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, meter.GasConsumed(), meter.Limit(), "Gas consumption not match with limit")
|
|
|
|
require.Equal(t, uint64(0), meter.GasRemaining())
|
2021-06-02 08:14:15 -07:00
|
|
|
require.Panics(t, func() { meter.RefundGas(meter.GasConsumed()+1, "refund greater than consumed") })
|
|
|
|
|
2021-07-09 07:38:25 -07:00
|
|
|
require.NotPanics(t, func() { meter.RefundGas(meter.GasConsumed(), "refund consumed gas") })
|
|
|
|
require.Equal(t, meter.Limit(), meter.GasRemaining())
|
|
|
|
|
2020-03-01 16:27:03 -08:00
|
|
|
meter2 := NewGasMeter(math.MaxUint64)
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, uint64(math.MaxUint64), meter2.GasRemaining())
|
2020-03-01 16:27:03 -08:00
|
|
|
meter2.ConsumeGas(Gas(math.MaxUint64/2), "consume half max uint64")
|
2021-07-09 07:38:25 -07:00
|
|
|
require.Equal(t, Gas(math.MaxUint64-(math.MaxUint64/2)), meter2.GasRemaining())
|
2020-03-01 16:27:03 -08:00
|
|
|
require.Panics(t, func() { meter2.ConsumeGas(Gas(math.MaxUint64/2)+2, "panic") })
|
2018-08-06 12:00:49 -07:00
|
|
|
}
|
|
|
|
}
|
2019-02-01 17:03:09 -08:00
|
|
|
|
|
|
|
func TestAddUint64Overflow(t *testing.T) {
|
2020-03-01 16:27:03 -08:00
|
|
|
t.Parallel()
|
2019-02-01 17:03:09 -08:00
|
|
|
testCases := []struct {
|
|
|
|
a, b uint64
|
|
|
|
result uint64
|
|
|
|
overflow bool
|
|
|
|
}{
|
|
|
|
{0, 0, 0, false},
|
|
|
|
{100, 100, 200, false},
|
|
|
|
{math.MaxUint64 / 2, math.MaxUint64/2 + 1, math.MaxUint64, false},
|
|
|
|
{math.MaxUint64 / 2, math.MaxUint64/2 + 2, 0, true},
|
|
|
|
}
|
|
|
|
|
|
|
|
for i, tc := range testCases {
|
|
|
|
res, overflow := addUint64Overflow(tc.a, tc.b)
|
|
|
|
require.Equal(
|
|
|
|
t, tc.overflow, overflow,
|
|
|
|
"invalid overflow result; tc: #%d, a: %d, b: %d", i, tc.a, tc.b,
|
|
|
|
)
|
|
|
|
require.Equal(
|
|
|
|
t, tc.result, res,
|
|
|
|
"invalid uint64 result; tc: #%d, a: %d, b: %d", i, tc.a, tc.b,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
2020-03-01 16:27:03 -08:00
|
|
|
|
|
|
|
func TestTransientGasConfig(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
config := TransientGasConfig()
|
|
|
|
require.Equal(t, config, GasConfig{
|
2021-03-05 08:25:19 -08:00
|
|
|
HasCost: 100,
|
|
|
|
DeleteCost: 100,
|
|
|
|
ReadCostFlat: 100,
|
|
|
|
ReadCostPerByte: 0,
|
|
|
|
WriteCostFlat: 200,
|
|
|
|
WriteCostPerByte: 3,
|
|
|
|
IterNextCostFlat: 3,
|
2020-03-01 16:27:03 -08:00
|
|
|
})
|
|
|
|
}
|