package osbuild2

import (

	"encoding/json"

	"testing"

	"github.com/stretchr/testify/assert"

)

func TestOCIArchiveStage(t *testing.T) {

	expectedStage := &Stage{

		Type:    "org.osbuild.oci-archive",

		Options: &OCIArchiveStageOptions{},

		Inputs:  &OCIArchiveStageInputs{},

	}

	actualStage := NewOCIArchiveStage(&OCIArchiveStageOptions{}, &OCIArchiveStageInputs{})

	assert.Equal(t, expectedStage, actualStage)

}

func TestOCIArchiveInputs(t *testing.T) {

	exp := `{

		"base": {

			"type": "org.osbuild.oci-archive",

			"origin":"org.osbuild.pipeline",

			"references": ["name:container-tree"]

		},

		"layer.1": {

			"type": "org.osbuild.tree",

			"origin": "org.osbuild.pipeline",

			"references": ["name:container-ostree"]

		},

		"layer.2": {

			"type": "org.osbuild.tree",

			"origin": "org.osbuild.pipeline",

			"references": ["name:container-ostree2"]

		}

	}`

	inputs := new(OCIArchiveStageInputs)

	base := &OCIArchiveStageInput{

		References: []string{

			"name:container-tree",

		},

	}

	base.Type = "org.osbuild.oci-archive"

	base.Origin = "org.osbuild.pipeline"

	layer1 := OCIArchiveStageInput{

		References: []string{

			"name:container-ostree",

		},

	}

	layer1.Type = "org.osbuild.tree"

	layer1.Origin = "org.osbuild.pipeline"

	layer2 := OCIArchiveStageInput{

		References: []string{

			"name:container-ostree2",

		},

	}

	layer2.Type = "org.osbuild.tree"

	layer2.Origin = "org.osbuild.pipeline"

	inputs.Base = base

	inputs.Layers = []OCIArchiveStageInput{layer1, layer2}

	data, err := json.Marshal(inputs)

	assert.NoError(t, err)

	assert.JSONEq(t, exp, string(data))

	inputsRead := new(OCIArchiveStageInputs)

	err = json.Unmarshal([]byte(exp), inputsRead)

	assert.NoError(t, err)

	assert.Equal(t, inputs, inputsRead)

}

func TestOCIArchiveInputsErrors(t *testing.T) {

	noBase := `{

		"layer.10": {

			"type": "org.osbuild.tree",

			"origin": "org.osbuild.pipeline",

			"references": ["name:container-ostree"]

		},

		"layer.2": {

			"type": "org.osbuild.tree",

			"origin": "org.osbuild.pipeline",

			"references": ["name:container-ostree2"]

		}

	}`

	inputsRead := new(OCIArchiveStageInputs)

	assert.Error(t, json.Unmarshal([]byte(noBase), inputsRead))

	invalidKey := `{

		"base": {

			"type": "org.osbuild.oci-archive",

			"origin":"org.osbuild.pipeline",

			"references": ["name:container-tree"]

		},

		"not-a-layer": {

			"type": "org.osbuild.tree",

			"origin": "org.osbuild.pipeline",

			"references": ["name:container-ostree2"]

		}

	}`

	assert.Error(t, json.Unmarshal([]byte(invalidKey), inputsRead))

}