Function TestManager_Dispatch_ghExtensionEnv

pkg/cmd/extension/manager_test.go:201–268

(t *testing.T)

}

func TestManager_Dispatch_ghExtensionEnv(t *testing.T) {
	tests := []struct {
		name     string
		extraEnv []string
		wantOut  string
	}{
		{
			name:    "sets GH_EXTENSION",
			wantOut: "GH_EXTENSION=1\nGH_HELPER_INHERITED=\n",
		},
		{
			name:     "preserves the rest of the environment",
			extraEnv: []string{"GH_HELPER_INHERITED=yes"},
			wantOut:  "GH_EXTENSION=1\nGH_HELPER_INHERITED=yes\n",
		},
		{
			name:     "overrides an inherited GH_EXTENSION",
			extraEnv: []string{"GH_EXTENSION=0", "GH_HELPER_INHERITED=yes"},
			wantOut:  "GH_EXTENSION=1\nGH_HELPER_INHERITED=yes\n",
		},
	}

	for _, tt := range tests {
		t.Run("script extension: "+tt.name, func(t *testing.T) {
			dataDir := t.TempDir()
			updateDir := t.TempDir()
			extDir := filepath.Join(dataDir, "extensions", "gh-hello")
			require.NoError(t, stubExtension(filepath.Join(extDir, "gh-hello")))

			gc, gcOne := &mockGitClient{}, &mockGitClient{}
			gc.On("ForRepo", extDir).Return(gcOne).Once()

			m := newTestManager(dataDir, updateDir, nil, gc, nil, tt.extraEnv...)

			stdout := &bytes.Buffer{}
			stderr := &bytes.Buffer{}
			found, err := m.Dispatch([]string{"hello", "print-env"}, nil, stdout, stderr)
			require.NoError(t, err)
			require.True(t, found)

			assert.Equal(t, tt.wantOut, stdout.String())
			assert.Equal(t, "", stderr.String())
		})

		t.Run("binary extension: "+tt.name, func(t *testing.T) {
			dataDir := t.TempDir()
			updateDir := t.TempDir()
			extDir := filepath.Join(dataDir, "extensions", "gh-hello")
			require.NoError(t, stubBinaryExtension(extDir, binManifest{
				Owner: "owner",
				Name:  "gh-hello",
				Host:  "github.com",
				Tag:   "v1.0.0",
			}))

			m := newTestManager(dataDir, updateDir, nil, nil, nil, tt.extraEnv...)

			stdout := &bytes.Buffer{}