must.go

package testing

import (
	"encoding/json"
	"time"

	"github.com/golang/protobuf/proto"
	"github.com/golang/protobuf/ptypes"
	"github.com/golang/protobuf/ptypes/timestamp"
	"google.golang.org/protobuf/types/known/timestamppb"

	"github.com/hyperledger-labs/cckit/serialize"
)

// PanicIfError
func PanicIfError(err error) {
	if err != nil {
		panic(err)
	}
}

// MustProtoMarshal marshals proto.Message, panics if error
func MustProtoMarshal(pb proto.Message) []byte {
	bb, err := proto.Marshal(pb)
	PanicIfError(err)

	return bb
}

func MustJSONMarshal(val interface{}) []byte {
	bb, err := json.Marshal(val)
	PanicIfError(err)
	return bb
}

// MustProtoUnmarshal unmarshals proto.Message, panics if error
func MustProtoUnmarshal(bb []byte, pm proto.Message) proto.Message {
	p := proto.Clone(pm)
	PanicIfError(proto.Unmarshal(bb, p))
	return p
}

// MustProtoTimestamp creates proto.Timestamp, panics if error
func MustProtoTimestamp(t time.Time) *timestamp.Timestamp {
	return timestamppb.New(t)
}

func MustConvertFromBytes(bb []byte, target interface{}, fromBytesConverter serialize.FromBytesConverter) interface{} {
	v, err := fromBytesConverter.FromBytesTo(bb, target)
	PanicIfError(err)
	return v
}

// MustTime returns Timestamp for date string or panic
func MustTime(s string) *timestamp.Timestamp {
	t, err := time.Parse(time.RFC3339, s)
	if err != nil {
		panic(err)
	}

	ts, err := ptypes.TimestampProto(t)
	if err != nil {
		panic(err)
	}

	return ts
}