add init unit test

precompiles/prototype/prototype.go

@@ -33,6 +33,10 @@
 )
 
 func init() {
+	initABI()
+}
+
+func initABI() {
 	if prototypeABI == "" {
 		panic("missing prototype ABI")
 	}
@@ -124,16 +128,16 @@
 
 	bech32Prefix := strings.SplitN(address, "1", 2)[0]
 	if bech32Prefix == address {
 		return nil, fmt.Errorf("invalid bech32 address: %s", address)
 	}
 
 	addressBz, err := sdk.GetFromBech32(address, bech32Prefix)
 	if err != nil {
 		return nil, err
 	}
 
 	if err := sdk.VerifyAddressFormat(addressBz); err != nil {
 		return nil, err
 	}
 
 	return method.Outputs.Pack(common.BytesToAddress(addressBz))
@@ -169,21 +173,21 @@
 
 	// NOTE: safety check, should not happen given that the address is 20 bytes.
 	if err := sdk.VerifyAddressFormat(address.Bytes()); err != nil {
 		return nil, err
 	}
 
 	bech32Str, err := sdk.Bech32ifyAddressBytes(prefix, address.Bytes())
 	if err != nil {
 		return nil, err
 	}
 
 	addressBz, err := sdk.GetFromBech32(bech32Str, "zeta")
 	if err != nil {
 		return nil, err
 	}
 
 	if err := sdk.VerifyAddressFormat(addressBz); err != nil {
 		return nil, err
 	}
 
 	return method.Outputs.Pack(bech32Str)
@@ -214,43 +218,43 @@
 		return nil, fmt.Errorf("error calling fungible keeper: %s", err.Error())
 	}
 
 	return method.Outputs.Pack(balance)
 }
 
 // Run is the entrypoint of the precompiled contract, it switches over the input method,
 // and execute them accordingly.
 func (c *Contract) Run(evm *vm.EVM, contract *vm.Contract, _ bool) ([]byte, error) {
 	method, err := ABI.MethodById(contract.Input[:4])
 	if err != nil {
 		return nil, err
 	}
 
 	args, err := method.Inputs.Unpack(contract.Input[4:])
 	if err != nil {
 		return nil, err
 	}
 
 	stateDB := evm.StateDB.(ptypes.ExtStateDB)
 
 	switch method.Name {
 	case GetGasStabilityPoolBalanceName:
 		var res []byte
 		execErr := stateDB.ExecuteNativeAction(contract.Address(), nil, func(ctx sdk.Context) error {
 			res, err = c.GetGasStabilityPoolBalance(ctx, method, args)
 			return err
 		})
 		if execErr != nil {
 			return nil, err
 		}
 		return res, nil
 
 	case Bech32ToHexAddrMethodName:
 		return c.Bech32ToHexAddr(method, args)
 	case Bech32ifyMethodName:
 		return c.Bech32ify(method, args)
 	default:
 		return nil, ptypes.ErrInvalidMethod{
 			Method: method.Name,
 		}
 	}
 }

precompiles/prototype/prototype_test.go

@@ -1,6 +1,7 @@
 package prototype
 
 import (
+	"encoding/json"
 	"testing"
 
 	storetypes "github.com/cosmos/cosmos-sdk/store/types"
@@ -277,4 +278,26 @@ func Test_InvalidMethod(t *testing.T) {
 	// Test for non existent method.
 	_, doNotExist := abi.Methods["invalidMethod"]
 	require.False(t, doNotExist, "invalidMethod should not be present in the ABI")
+}
+
+func Test_MissingABI(t *testing.T) {
+	prototypeABI = ""
+	defer func() {
+		if r := recover(); r != nil {
+			require.Equal(t, "missing prototype ABI", r, "expected error: missing ABI, got: %v", r)
+		}
+	}()
+
+	initABI()
+}
+
+func Test_InvalidABI(t *testing.T) {
+	prototypeABI = "invalid json"
+	defer func() {
+		if r := recover(); r != nil {
+			require.IsType(t, &json.SyntaxError{}, r, "expected error type: json.SyntaxError, got: %T", r)
+		}
+	}()
+
+	initABI()
 }