Function Test_proxyWithErrorHandler

cliv2/pkg/basic_workflows/legacycli_test.go:46–101

Test_proxyWithErrorHandler tests if we correctly either set or do not set the "snyk-terminate" header, which is a signal to the legacy TypeScript CLI that it should stop trying to send HTTP requests.

(t *testing.T)

// Test_proxyWithErrorHandler tests if we correctly either set or do not set the "snyk-terminate" header,
// which is a signal to the legacy TypeScript CLI that it should stop trying to send HTTP requests.
func Test_proxyWithErrorHandler(t *testing.T) {
	ctrl := gomock.NewController(t)
	logger := zerolog.Nop()
	config := configuration.NewWithOpts()

	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusUnauthorized)
	})
	server := httptest.NewServer(handler)
	defer server.Close()

	networkAccess := networking.NewNetworkAccess(config)
	networkAccess.AddErrorHandler(func(err error, _ context.Context) error { return err })

	invocationCtxMock := mocks.NewMockInvocationContext(ctrl)
	invocationCtxMock.EXPECT().GetNetworkAccess().Return(networkAccess).AnyTimes()
	invocationCtxMock.EXPECT().GetEnhancedLogger().Return(&logger).AnyTimes()

	testCases := []struct {
		name              string
		configureApiUrl   string
		expectedIntercept bool
	}{
		{
			name:              "intercepts traffic based on configuration",
			configureApiUrl:   server.URL,
			expectedIntercept: true,
		},
		{
			name:              "does not intercept external traffic",
			configureApiUrl:   "http://api.snyk.io",
			expectedIntercept: false,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			config.Set(configuration.API_URL, tc.configureApiUrl)
			wp, err := createInternalProxy(config, &logger, invocationCtxMock)
			assert.Nil(t, err)

			defer wp.Close()

			client := http.Client{Transport: &http.Transport{Proxy: func(*http.Request) (*url.URL, error) {
				return &url.URL{Scheme: "http", Host: fmt.Sprintf("127.0.0.1:%d", wp.ProxyInfo().Port)}, nil
			}}}

			res, err := client.Get(server.URL)
			assert.NotNil(t, res)
			defer func() { _ = res.Body.Close() }()

			assert.Equal(t, tc.expectedIntercept, res.Header.Get(proxy.HeaderSnykTerminate) == "true")
			assert.Nil(t, err)
		})
	}
}

func Test_ValidateGlibcVersion_doesNotApplyOnNonLinux(t *testing.T) {