buildx/util/buildflags/ssh_test.go at fe17ebda89d07e43d113303d95a0065bbf77f32d · crazy-max/buildx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package buildflags

import (
	"encoding/json"
	"testing"

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

func TestSSHKeys(t *testing.T) {
	t.Run("MarshalJSON", func(t *testing.T) {
		sshkeys := SSHKeys{
			{ID: "default", Paths: []string{}},
			{ID: "key", Paths: []string{"path/to/key"}},
		}

		expected := `[{"id":"default"},{"id":"key","paths":["path/to/key"]}]`
		actual, err := json.Marshal(sshkeys)
		require.NoError(t, err)
		require.JSONEq(t, expected, string(actual))
	})

	t.Run("UnmarshalJSON", func(t *testing.T) {
		in := `[{"id":"default"},{"id":"key","paths":["path/to/key"]}]`

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

		expected := SSHKeys{
			{ID: "default"},
			{ID: "key", Paths: []string{"path/to/key"}},
		}
		require.Equal(t, expected, actual)
	})

	t.Run("FromCtyValue", func(t *testing.T) {
		in := cty.TupleVal([]cty.Value{
			cty.ObjectVal(map[string]cty.Value{
				"id": cty.StringVal("default"),
			}),
			cty.ObjectVal(map[string]cty.Value{
				"id": cty.StringVal("key"),
				"paths": cty.TupleVal([]cty.Value{
					cty.StringVal("path/to/key"),
				}),
			}),
		})

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

		expected := SSHKeys{
			{ID: "default"},
			{ID: "key", Paths: []string{"path/to/key"}},
		}
		require.Equal(t, expected, actual)
	})

	t.Run("ToCtyValue", func(t *testing.T) {
		sshkeys := SSHKeys{
			{ID: "default", Paths: []string{}},
			{ID: "key", Paths: []string{"path/to/key"}},
		}

		actual := sshkeys.ToCtyValue()
		expected := cty.ListVal([]cty.Value{
			cty.ObjectVal(map[string]cty.Value{
				"id":    cty.StringVal("default"),
				"paths": cty.ListValEmpty(cty.String),
			}),
			cty.ObjectVal(map[string]cty.Value{
				"id": cty.StringVal("key"),
				"paths": cty.ListVal([]cty.Value{
					cty.StringVal("path/to/key"),
				}),
			}),
		})

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

	t.Run("RemoveDupes", func(t *testing.T) {
		sshkeys := SSHKeys{
			{ID: "default"},
			{ID: "key", Paths: []string{"path/to/foo"}},
			{ID: "key", Paths: []string{"path/to/bar"}},
		}.Normalize()

		expected := `[{"id":"default"},{"id":"key","paths":["path/to/bar"]}]`
		actual, err := json.Marshal(sshkeys)
		require.NoError(t, err)
		require.JSONEq(t, expected, string(actual))
	})
}