marshalizerMock.go

package mock

import (
	"encoding/json"
	"errors"
)

// ErrMockMarshalizer -
var ErrMockMarshalizer = errors.New("MarshalizerMock generic error")

// ErrNilObjectToMarshal -
var ErrNilObjectToMarshal = errors.New("nil object to serialize from")

// MarshalizerMock that will be used for testing
type MarshalizerMock struct {
	MarshalCalled func(obj interface{}) ([]byte, error)
	Fail          bool
}

// Marshal converts the input object in a slice of bytes
func (mm *MarshalizerMock) Marshal(obj interface{}) ([]byte, error) {
	if mm.MarshalCalled != nil {
		return mm.MarshalCalled(obj)
	}

	if mm.Fail {
		return nil, ErrMockMarshalizer
	}

	if obj == nil {
		return nil, ErrNilObjectToMarshal
	}

	return json.Marshal(obj)
}

// Unmarshal applies the serialized values over an instantiated object
func (mm *MarshalizerMock) Unmarshal(obj interface{}, buff []byte) error {
	if mm.Fail {
		return ErrMockMarshalizer
	}

	if obj == nil {
		return errors.New("nil object to serilize to")
	}

	if buff == nil {
		return errors.New("nil byte buffer to deserialize from")
	}

	if len(buff) == 0 {
		return errors.New("empty byte buffer to deserialize from")
	}

	return json.Unmarshal(buff, obj)
}

// IsInterfaceNil returns true if there is no value under the interface
func (mm *MarshalizerMock) IsInterfaceNil() bool {
	return mm == nil
}