146 lines
3.7 KiB
Go
146 lines
3.7 KiB
Go
package mock
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
protov2 "google.golang.org/protobuf/proto"
|
|
|
|
bankv1beta1 "cosmossdk.io/api/cosmos/bank/v1beta1"
|
|
errorsmod "cosmossdk.io/errors"
|
|
|
|
cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
txsigning "github.com/cosmos/cosmos-sdk/types/tx/signing"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/signing"
|
|
)
|
|
|
|
// An sdk.Tx which is its own sdk.Msg.
|
|
type KVStoreTx struct {
|
|
key []byte
|
|
value []byte
|
|
bytes []byte
|
|
address sdk.AccAddress
|
|
}
|
|
|
|
// testPubKey is a dummy implementation of PubKey used for testing.
|
|
type testPubKey struct {
|
|
address sdk.AccAddress
|
|
}
|
|
|
|
func (t testPubKey) Reset() { panic("not implemented") }
|
|
|
|
func (t testPubKey) String() string { panic("not implemented") }
|
|
|
|
func (t testPubKey) ProtoMessage() { panic("not implemented") }
|
|
|
|
func (t testPubKey) Address() cryptotypes.Address { return t.address.Bytes() }
|
|
|
|
func (t testPubKey) Bytes() []byte { panic("not implemented") }
|
|
|
|
func (t testPubKey) VerifySignature(msg, sig []byte) bool { panic("not implemented") }
|
|
|
|
func (t testPubKey) Equals(key cryptotypes.PubKey) bool { panic("not implemented") }
|
|
|
|
func (t testPubKey) Type() string { panic("not implemented") }
|
|
|
|
func (msg *KVStoreTx) GetSignaturesV2() (res []txsigning.SignatureV2, err error) {
|
|
res = append(res, txsigning.SignatureV2{
|
|
PubKey: testPubKey{address: msg.address},
|
|
Data: nil,
|
|
Sequence: 1,
|
|
})
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (msg *KVStoreTx) VerifySignature(msgByte, sig []byte) bool {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (msg *KVStoreTx) Address() cryptotypes.Address {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (msg *KVStoreTx) Bytes() []byte {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (msg *KVStoreTx) Equals(key cryptotypes.PubKey) bool {
|
|
panic("implement me")
|
|
}
|
|
|
|
// dummy implementation of proto.Message
|
|
func (msg *KVStoreTx) Reset() {}
|
|
func (msg *KVStoreTx) String() string { return "TODO" }
|
|
func (msg *KVStoreTx) ProtoMessage() {}
|
|
|
|
var (
|
|
_ sdk.Tx = &KVStoreTx{}
|
|
_ sdk.Msg = &KVStoreTx{}
|
|
_ signing.SigVerifiableTx = &KVStoreTx{}
|
|
_ cryptotypes.PubKey = &KVStoreTx{}
|
|
_ cryptotypes.PubKey = &testPubKey{}
|
|
)
|
|
|
|
func NewTx(key, value string, accAddress sdk.AccAddress) *KVStoreTx {
|
|
bytes := fmt.Sprintf("%s=%s=%s", key, value, accAddress)
|
|
return &KVStoreTx{
|
|
key: []byte(key),
|
|
value: []byte(value),
|
|
bytes: []byte(bytes),
|
|
address: accAddress,
|
|
}
|
|
}
|
|
|
|
func (msg *KVStoreTx) Type() string {
|
|
return "kvstore_tx"
|
|
}
|
|
|
|
func (msg *KVStoreTx) GetMsgs() []sdk.Msg {
|
|
return []sdk.Msg{msg}
|
|
}
|
|
|
|
func (msg *KVStoreTx) GetMsgsV2() ([]protov2.Message, error) {
|
|
return []protov2.Message{&bankv1beta1.MsgSend{FromAddress: msg.address.String()}}, nil // this is a hack for tests
|
|
}
|
|
|
|
func (msg *KVStoreTx) GetSignBytes() []byte {
|
|
return msg.bytes
|
|
}
|
|
|
|
// Should the app be calling this? Or only handlers?
|
|
func (msg *KVStoreTx) ValidateBasic() error {
|
|
return nil
|
|
}
|
|
|
|
func (msg *KVStoreTx) GetSigners() ([][]byte, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (msg *KVStoreTx) GetPubKeys() ([]cryptotypes.PubKey, error) { panic("GetPubKeys not implemented") }
|
|
|
|
// takes raw transaction bytes and decodes them into an sdk.Tx. An sdk.Tx has
|
|
// all the signatures and can be used to authenticate.
|
|
func decodeTx(txBytes []byte) (sdk.Tx, error) {
|
|
var tx sdk.Tx
|
|
|
|
split := bytes.Split(txBytes, []byte("="))
|
|
switch len(split) {
|
|
case 1:
|
|
k := split[0]
|
|
tx = &KVStoreTx{k, k, txBytes, nil}
|
|
case 2:
|
|
k, v := split[0], split[1]
|
|
tx = &KVStoreTx{k, v, txBytes, nil}
|
|
case 3:
|
|
k, v, addr := split[0], split[1], split[2]
|
|
tx = &KVStoreTx{k, v, txBytes, addr}
|
|
default:
|
|
return nil, errorsmod.Wrap(sdkerrors.ErrTxDecode, "too many '='")
|
|
}
|
|
|
|
return tx, nil
|
|
}
|