package llbbuild

import (
	"testing"

	"github.com/moby/buildkit/client/llb"
	"github.com/moby/buildkit/solver/pb"
	digest "github.com/opencontainers/go-digest"
	"github.com/stretchr/testify/require"
)

func TestMarshal(t *testing.T) {
	b := NewBuildOp(newDummyOutput("foobar"), WithFilename("myfilename"))
	dt, opMeta, err := b.Marshal()
	_ = opMeta
	require.NoError(t, err)

	var op pb.Op
	err = op.Unmarshal(dt)
	require.NoError(t, err)

	buildop := op.GetBuild()
	require.NotEqual(t, buildop, nil)

	require.Equal(t, len(op.Inputs), 1)
	require.Equal(t, buildop.Builder, pb.LLBBuilder)
	require.Equal(t, len(buildop.Inputs), 1)
	require.Equal(t, buildop.Inputs[pb.LLBDefinitionInput], &pb.BuildInput{pb.InputIndex(0)})

	require.Equal(t, buildop.Attrs[pb.AttrLLBDefinitionFilename], "myfilename")
}

func newDummyOutput(key string) llb.Output {
	dgst := digest.FromBytes([]byte(key))
	return &dummyOutput{dgst: dgst}
}

type dummyOutput struct {
	dgst digest.Digest
}

func (d *dummyOutput) ToInput() (*pb.Input, error) {
	return &pb.Input{
		Digest: d.dgst,
		Index:  pb.OutputIndex(7), // random constant
	}, nil
}
func (d *dummyOutput) Vertex() llb.Vertex {
	return nil
}