68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package basecoin
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func init() {
|
|
TxMapper.
|
|
RegisterImplementation(Demo{}, TypeDemo, ByteDemo).
|
|
RegisterImplementation(Fake{}, TypeFake, ByteFake)
|
|
}
|
|
|
|
const (
|
|
ByteDemo = 0xF0
|
|
TypeDemo = "test/demo"
|
|
ByteFake = 0xF1
|
|
TypeFake = "test/fake"
|
|
)
|
|
|
|
// define a Demo struct that implements TxLayer
|
|
type Demo struct{}
|
|
|
|
var _ TxLayer = Demo{}
|
|
|
|
func (d Demo) Next() Tx { return Tx{} }
|
|
func (d Demo) Wrap() Tx { return Tx{d} }
|
|
func (d Demo) ValidateBasic() error { return nil }
|
|
|
|
// define a Fake struct that doesn't implement TxLayer
|
|
type Fake struct{}
|
|
|
|
func (f Fake) Wrap() Tx { return Tx{f} }
|
|
func (f Fake) ValidateBasic() error { return nil }
|
|
|
|
// Make sure the layer
|
|
func TestLayer(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
// a fake tx, just don't use it...
|
|
nl := Fake{}.Wrap()
|
|
assert.False(nl.IsLayer())
|
|
assert.Nil(nl.GetLayer())
|
|
|
|
// a tx containing a TxLayer should respond properly
|
|
l := Demo{}.Wrap()
|
|
assert.True(l.IsLayer())
|
|
assert.NotNil(l.GetLayer())
|
|
}
|
|
|
|
func TestKind(t *testing.T) {
|
|
cases := []struct {
|
|
tx Tx
|
|
kind string
|
|
}{
|
|
{Demo{}.Wrap(), TypeDemo},
|
|
{Fake{}.Wrap(), TypeFake},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
kind, err := tc.tx.GetKind()
|
|
require.Nil(t, err, "%+v", err)
|
|
assert.Equal(t, tc.kind, kind)
|
|
}
|
|
}
|