cosmos-sdk/examples/democoin/x/pow/types_test.go

package pow

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"

	sdk "github.com/cosmos/cosmos-sdk/types"
)

func TestNewMsgMine(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 0, 0, 0, []byte("")}
	equiv := NewMsgMine(addr, 0, 0, 0, []byte(""))
	assert.Equal(t, msg, equiv, "%s != %s", msg, equiv)
}

func TestMsgMineType(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 0, 0, 0, []byte("")}
	assert.Equal(t, msg.Type(), "pow")
}

func TestMsgMineValidation(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	otherAddr := sdk.Address([]byte("another"))
	count := uint64(0)

	for difficulty := uint64(1); difficulty < 1000; difficulty += 100 {

		count++
		nonce, proof := mine(addr, count, difficulty)
		msg := MsgMine{addr, difficulty, count, nonce, proof}
		err := msg.ValidateBasic()
		assert.Nil(t, err, "error with difficulty %d - %+v", difficulty, err)

		msg.Count++
		err = msg.ValidateBasic()
		assert.NotNil(t, err, "count was wrong, should have thrown error with msg %s", msg)

		msg.Count--
		msg.Nonce++
		err = msg.ValidateBasic()
		assert.NotNil(t, err, "nonce was wrong, should have thrown error with msg %s", msg)

		msg.Nonce--
		msg.Sender = otherAddr
		err = msg.ValidateBasic()
		assert.NotNil(t, err, "sender was wrong, should have thrown error with msg %s", msg)
	}
}

func TestMsgMineString(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 0, 0, 0, []byte("abc")}
	res := msg.String()
	assert.Equal(t, res, "MsgMine{Sender: 73656E646572, Difficulty: 0, Count: 0, Nonce: 0, Proof: abc}")
}

func TestMsgMineGet(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 0, 0, 0, []byte("")}
	res := msg.Get(nil)
	assert.Nil(t, res)
}

func TestMsgMineGetSignBytes(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 1, 1, 1, []byte("abc")}
	res := msg.GetSignBytes()
	assert.Equal(t, string(res), `{"sender":"73656E646572","difficulty":1,"count":1,"nonce":1,"proof":"YWJj"}`)
}

func TestMsgMineGetSigners(t *testing.T) {
	addr := sdk.Address([]byte("sender"))
	msg := MsgMine{addr, 1, 1, 1, []byte("abc")}
	res := msg.GetSigners()
	assert.Equal(t, fmt.Sprintf("%v", res), "[73656E646572]")
}