utils_test.go

package bitxid

import (
	"testing"

	"github.com/meshplus/bitxhub-kit/log"
	"github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"
)

const (
	loggerAccountDID = "account-did"
	loggerChainDID   = "chain-did"
)

var w *loggerWrapper

type loggerWrapper struct {
	loggers map[string]*logrus.Entry
}

// Init initializes all loggers
func loggerInit() {
	// config *repo.Config
	m := make(map[string]*logrus.Entry)
	m[loggerAccountDID] = log.NewWithModule(loggerAccountDID)
	m[loggerAccountDID].Logger.SetLevel(log.ParseLevel("info"))
	m[loggerChainDID] = log.NewWithModule(loggerChainDID)
	m[loggerChainDID].Logger.SetLevel(log.ParseLevel("info"))

	w = &loggerWrapper{loggers: m}
}

// Get gets logger for specific module
func loggerGet(name string) logrus.FieldLogger {
	return w.loggers[name]
}

type s struct {
	A string
	B int
}

var testBytes []byte

func TestMarshal(t *testing.T) {
	testStruct := s{
		A: "aaa",
		B: 1,
	}
	expectedByte := []byte{27, 255, 161, 3, 1, 1, 1, 115,
		1, 255, 162, 0, 1, 2, 1, 1,
		65, 1, 12, 0, 1, 1, 66, 1,
		4, 0, 0, 0, 10, 255, 162, 1,
		3, 97, 97, 97, 1, 2, 0}

	testBytes1, err := Marshal(testStruct)
	assert.Nil(t, err)
	assert.Equal(t, expectedByte, testBytes1)
	testBytes = testBytes1
}

func TestUnmarshal(t *testing.T) {
	expectedStruct := s{
		A: "aaa",
		B: 1,
	}
	var testStruct s
	err := Unmarshal(testBytes, &testStruct)
	assert.Nil(t, err)
	assert.Equal(t, expectedStruct, testStruct)
}