go-buildkite/optional_test.go at main · buildkite/go-buildkite · 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
package buildkite

import (
	"encoding/json"
	"testing"
)

func TestOptionalMarshalJSON(t *testing.T) {
	t.Parallel()

	type payload struct {
		Name    Optional[string]            `json:"name,omitzero"`
		Tags    Optional[[]string]          `json:"tags,omitzero"`
		Env     Optional[map[string]string] `json:"env,omitzero"`
		Enabled Optional[bool]              `json:"enabled,omitzero"`
	}

	tests := []struct {
		name string
		in   payload
		want string
	}{
		{
			name: "unset fields omitted",
			in:   payload{},
			want: `{}`,
		},
		{
			name: "empty string is sent",
			in: payload{
				Name: Some(""),
			},
			want: `{"name":""}`,
		},
		{
			name: "false is sent",
			in: payload{
				Enabled: Some(false),
			},
			want: `{"enabled":false}`,
		},
		{
			name: "empty slice is sent",
			in: payload{
				Tags: Some([]string{}),
			},
			want: `{"tags":[]}`,
		},
		{
			name: "empty map is sent",
			in: payload{
				Env: Some(map[string]string{}),
			},
			want: `{"env":{}}`,
		},
		{
			name: "nil slice is sent as null when set",
			in: payload{
				Tags: Some([]string(nil)),
			},
			want: `{"tags":null}`,
		},
		{
			name: "nil map is sent as null when set",
			in: payload{
				Env: Some(map[string]string(nil)),
			},
			want: `{"env":null}`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			got, err := json.Marshal(tt.in)
			if err != nil {
				t.Fatalf("Marshal returned error: %v", err)
			}
			assertJSONEqual(t, string(got), tt.want)
		})
	}
}