Update eth_getBlockByNumber to accept words

This commit is contained in:
Taylor Gerring 2015-04-22 13:14:10 -05:00
parent 5f6c8832af
commit 35ad9febce
2 changed files with 21 additions and 6 deletions

View File

@ -145,12 +145,8 @@ func (args *GetBlockByNumberArgs) UnmarshalJSON(b []byte) (err error) {
return NewInsufficientParamsError(len(obj), 2) return NewInsufficientParamsError(len(obj), 2)
} }
if v, ok := obj[0].(float64); ok { if err := blockHeight(obj[0], &args.BlockNumber); err != nil {
args.BlockNumber = int64(v) return err
} else if v, ok := obj[0].(string); ok {
args.BlockNumber = common.Big(v).Int64()
} else {
return NewInvalidTypeError("blockNumber", "not a number or string")
} }
args.IncludeTxs = obj[1].(bool) args.IncludeTxs = obj[1].(bool)

View File

@ -355,6 +355,25 @@ func TestGetBlockByNumberArgsBlockHex(t *testing.T) {
t.Errorf("IncludeTxs should be %v but is %v", expected.IncludeTxs, args.IncludeTxs) t.Errorf("IncludeTxs should be %v but is %v", expected.IncludeTxs, args.IncludeTxs)
} }
} }
func TestGetBlockByNumberArgsWords(t *testing.T) {
input := `["earliest", true]`
expected := new(GetBlockByNumberArgs)
expected.BlockNumber = 0
expected.IncludeTxs = true
args := new(GetBlockByNumberArgs)
if err := json.Unmarshal([]byte(input), &args); err != nil {
t.Error(err)
}
if args.BlockNumber != expected.BlockNumber {
t.Errorf("BlockNumber should be %v but is %v", expected.BlockNumber, args.BlockNumber)
}
if args.IncludeTxs != expected.IncludeTxs {
t.Errorf("IncludeTxs should be %v but is %v", expected.IncludeTxs, args.IncludeTxs)
}
}
func TestGetBlockByNumberEmpty(t *testing.T) { func TestGetBlockByNumberEmpty(t *testing.T) {
input := `[]` input := `[]`