Function Test_setRun_env

pkg/cmd/variable/set/set_test.go:242–314

(t *testing.T)

}

func Test_setRun_env(t *testing.T) {
	tests := []struct {
		name      string
		opts      *SetOptions
		httpStubs func(*httpmock.Registry)
		wantErr   bool
	}{
		{
			name: "create env variable",
			opts: &SetOptions{},
			httpStubs: func(reg *httpmock.Registry) {
				reg.Register(httpmock.GraphQL(`query MapRepositoryNames\b`),
					httpmock.StringResponse(`{"data":{"repo_0001":{"databaseId":1}}}`))
				reg.Register(httpmock.REST("POST", "repositories/1/environments/release/variables"),
					httpmock.StatusStringResponse(201, `{}`))
			},
		},
		{
			name: "update env variable",
			opts: &SetOptions{},
			httpStubs: func(reg *httpmock.Registry) {
				reg.Register(httpmock.GraphQL(`query MapRepositoryNames\b`),
					httpmock.StringResponse(`{"data":{"repo_0001":{"databaseId":1}}}`))
				reg.Register(httpmock.REST("POST", "repositories/1/environments/release/variables"),
					httpmock.StatusStringResponse(409, `{}`))
				reg.Register(httpmock.REST("PATCH", "repositories/1/environments/release/variables/cool_variable"),
					httpmock.StatusStringResponse(204, `{}`))
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			reg := &httpmock.Registry{}
			defer reg.Verify(t)
			if tt.httpStubs != nil {
				tt.httpStubs(reg)
			}

			ios, _, _, _ := iostreams.Test()

			opts := &SetOptions{
				HttpClient: func() (*http.Client, error) {
					return &http.Client{Transport: reg}, nil
				},
				Config: func() (gh.Config, error) { return config.NewBlankConfig(), nil },
				BaseRepo: func() (ghrepo.Interface, error) {
					return ghrepo.FromFullName("owner/repo")
				},
				IO:           ios,
				VariableName: "cool_variable",
				Body:         "a variable",
				EnvName:      "release",
			}

			err := setRun(opts)
			if tt.wantErr {
				assert.Error(t, err)