50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package rpctypes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type SampleResult struct {
|
|
Value string
|
|
}
|
|
|
|
func TestResponses(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
a := NewRPCSuccessResponse("1", &SampleResult{"hello"})
|
|
b, _ := json.Marshal(a)
|
|
s := `{"jsonrpc":"2.0","id":"1","result":{"Value":"hello"}}`
|
|
assert.Equal(string(s), string(b))
|
|
|
|
d := RPCParseError("1", errors.New("Hello world"))
|
|
e, _ := json.Marshal(d)
|
|
f := `{"jsonrpc":"2.0","id":"1","error":{"code":-32700,"message":"Parse error. Invalid JSON","data":"Hello world"}}`
|
|
assert.Equal(string(f), string(e))
|
|
|
|
g := RPCMethodNotFoundError("2")
|
|
h, _ := json.Marshal(g)
|
|
i := `{"jsonrpc":"2.0","id":"2","error":{"code":-32601,"message":"Method not found"}}`
|
|
assert.Equal(string(h), string(i))
|
|
}
|
|
|
|
func TestRPCError(t *testing.T) {
|
|
assert.Equal(t, "RPC error 12 - Badness: One worse than a code 11",
|
|
fmt.Sprintf("%v", &RPCError{
|
|
Code: 12,
|
|
Message: "Badness",
|
|
Data: "One worse than a code 11",
|
|
}))
|
|
|
|
assert.Equal(t, "RPC error 12 - Badness",
|
|
fmt.Sprintf("%v", &RPCError{
|
|
Code: 12,
|
|
Message: "Badness",
|
|
}))
|
|
}
|