(t *testing.T)
| 382 | } |
| 383 | |
| 384 | func Test_wrapTransportWithEntitlementsCheck(t *testing.T) { |
| 385 | t.Parallel() |
| 386 | |
| 387 | lines := []string{"First Warning", "Second Warning"} |
| 388 | var buf bytes.Buffer |
| 389 | rt := wrapTransportWithEntitlementsCheck(roundTripper(func(req *http.Request) (*http.Response, error) { |
| 390 | return &http.Response{ |
| 391 | StatusCode: http.StatusOK, |
| 392 | Header: http.Header{ |
| 393 | codersdk.EntitlementsWarningHeader: lines, |
| 394 | }, |
| 395 | Body: io.NopCloser(nil), |
| 396 | }, nil |
| 397 | }), &buf) |
| 398 | res, err := rt.RoundTrip(httptest.NewRequest(http.MethodGet, "http://example.com", nil)) |
| 399 | require.NoError(t, err) |
| 400 | defer res.Body.Close() |
| 401 | expectedOutput := fmt.Sprintf("%s\n%s\n", pretty.Sprint(cliui.DefaultStyles.Warn, lines[0]), |
| 402 | pretty.Sprint(cliui.DefaultStyles.Warn, lines[1])) |
| 403 | require.Equal(t, expectedOutput, buf.String()) |
| 404 | } |
| 405 | |
| 406 | func Test_ensureTLSConfig(t *testing.T) { |
| 407 | t.Parallel() |
nothing calls this directly
no test coverage detected