buildx/util/buildflags/attests_test.go

package buildflags

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/require"
	"github.com/zclconf/go-cty/cty"
)

func TestAttests(t *testing.T) {
	t.Run("MarshalJSON", func(t *testing.T) {
		attests := Attests{
			{Type: "provenance", Attrs: map[string]string{"mode": "max"}},
			{Type: "sbom", Disabled: true},
			{Type: "sbom", Attrs: map[string]string{
				"generator": "scanner",
				"ENV1":      `"foo,bar"`,
				"Env2":      "hello",
			}},
		}

		expected := `[{"type":"provenance","mode":"max"},{"type":"sbom","disabled":true},{"ENV1":"\"foo,bar\"","Env2":"hello","generator":"scanner","type":"sbom"}]`
		actual, err := json.Marshal(attests)
		require.NoError(t, err)
		require.JSONEq(t, expected, string(actual))
	})

	t.Run("UnmarshalJSON", func(t *testing.T) {
		in := `[{"type":"provenance","mode":"max"},{"type":"sbom","disabled":true},{"ENV1":"\"foo,bar\"","Env2":"hello","generator":"scanner","type":"sbom"}]`

		var actual Attests
		err := json.Unmarshal([]byte(in), &actual)
		require.NoError(t, err)

		expected := Attests{
			{Type: "provenance", Attrs: map[string]string{"mode": "max"}},
			{Type: "sbom", Disabled: true, Attrs: map[string]string{}},
			{Type: "sbom", Disabled: false, Attrs: map[string]string{
				"generator": "scanner",
				"ENV1":      `"foo,bar"`,
				"Env2":      "hello",
			}},
		}
		require.Equal(t, expected, actual)
	})

	t.Run("FromCtyValue", func(t *testing.T) {
		in := cty.TupleVal([]cty.Value{
			cty.ObjectVal(map[string]cty.Value{
				"type": cty.StringVal("provenance"),
				"mode": cty.StringVal("max"),
			}),
			cty.ObjectVal(map[string]cty.Value{
				"type":      cty.StringVal("sbom"),
				"generator": cty.StringVal("scan"),
				"ENV1":      cty.StringVal(`foo,bar`),
				"Env2":      cty.StringVal(`hello`),
			}),
			cty.StringVal("type=sbom,disabled=true"),
			cty.StringVal(`type=sbom,generator=scan,"FOO=bar,baz",Hello=World`),
		})

		var actual Attests
		err := actual.FromCtyValue(in, nil)
		require.NoError(t, err)

		expected := Attests{
			{Type: "provenance", Attrs: map[string]string{"mode": "max"}},
			{Type: "sbom", Attrs: map[string]string{
				"generator": "scan",
				"ENV1":      "foo,bar",
				"Env2":      "hello",
			}},
			{Type: "sbom", Disabled: true, Attrs: map[string]string{}},
			{Type: "sbom", Attrs: map[string]string{
				"generator": "scan",
				"FOO":       "bar,baz",
				"Hello":     "World",
			}},
		}
		require.Equal(t, expected, actual)
	})

	t.Run("ToCtyValue", func(t *testing.T) {
		attests := Attests{
			{Type: "provenance", Attrs: map[string]string{"mode": "max"}},
			{Type: "sbom", Disabled: true},
			{Type: "sbom", Attrs: map[string]string{
				"generator": "scan",
				"ENV1":      `"foo,bar"`,
				"Env2":      "hello",
			}},
		}

		actual := attests.ToCtyValue()
		expected := cty.ListVal([]cty.Value{
			cty.MapVal(map[string]cty.Value{
				"type": cty.StringVal("provenance"),
				"mode": cty.StringVal("max"),
			}),
			cty.MapVal(map[string]cty.Value{
				"type":     cty.StringVal("sbom"),
				"disabled": cty.StringVal("true"),
			}),
			cty.MapVal(map[string]cty.Value{
				"type":      cty.StringVal("sbom"),
				"generator": cty.StringVal("scan"),
				"ENV1":      cty.StringVal(`"foo,bar"`),
				"Env2":      cty.StringVal("hello"),
			}),
		})

		result := actual.Equals(expected)
		require.True(t, result.True())
	})
}