|
15 | 15 | package api |
16 | 16 |
|
17 | 17 | import ( |
| 18 | + "fmt" |
| 19 | + "net/http" |
18 | 20 | "os" |
19 | 21 | "testing" |
20 | 22 |
|
| 23 | + "github.com/slackapi/slack-cli/internal/config" |
21 | 24 | "github.com/slackapi/slack-cli/internal/goutils" |
| 25 | + "github.com/slackapi/slack-cli/internal/iostreams" |
| 26 | + "github.com/slackapi/slack-cli/internal/slackcontext" |
| 27 | + "github.com/slackapi/slack-cli/internal/slackdeps" |
| 28 | + "github.com/stretchr/testify/mock" |
22 | 29 | "github.com/stretchr/testify/require" |
23 | 30 | ) |
24 | 31 |
|
| 32 | +func Test_printRequest(t *testing.T) { |
| 33 | + tests := map[string]struct { |
| 34 | + userAgent string |
| 35 | + expected string |
| 36 | + }{ |
| 37 | + "includes User-Agent header in output": { |
| 38 | + userAgent: "slack-cli/v1.2.3 (os: darwin)", |
| 39 | + expected: "HTTP Request User-Agent: slack-cli/v1.2.3 (os: darwin)", |
| 40 | + }, |
| 41 | + "includes empty User-Agent when header is not set": { |
| 42 | + userAgent: "", |
| 43 | + expected: "HTTP Request User-Agent: ", |
| 44 | + }, |
| 45 | + } |
| 46 | + for name, tc := range tests { |
| 47 | + t.Run(name, func(t *testing.T) { |
| 48 | + ctx := slackcontext.MockContext(t.Context()) |
| 49 | + fs := slackdeps.NewFsMock() |
| 50 | + osMock := slackdeps.NewOsMock() |
| 51 | + osMock.AddDefaultMocks() |
| 52 | + cfg := config.NewConfig(fs, osMock) |
| 53 | + cfg.DebugEnabled = true |
| 54 | + ioMock := iostreams.NewIOStreamsMock(cfg, fs, osMock) |
| 55 | + ioMock.On("PrintDebug", mock.Anything, mock.Anything, mock.MatchedBy(func(args ...any) bool { return true })) |
| 56 | + |
| 57 | + c := &Client{io: ioMock} |
| 58 | + req, _ := http.NewRequest("GET", "https://slack.com/api/test", nil) |
| 59 | + if tc.userAgent != "" { |
| 60 | + req.Header.Set("User-Agent", tc.userAgent) |
| 61 | + } |
| 62 | + |
| 63 | + c.printRequest(ctx, req, false) |
| 64 | + |
| 65 | + var output string |
| 66 | + for _, call := range ioMock.Calls { |
| 67 | + if call.Method == "PrintDebug" { |
| 68 | + output = fmt.Sprintf(call.Arguments[1].(string), call.Arguments[2].([]any)...) |
| 69 | + } |
| 70 | + } |
| 71 | + require.Contains(t, output, tc.expected) |
| 72 | + }) |
| 73 | + } |
| 74 | +} |
| 75 | + |
25 | 76 | func Test_RedactPII(t *testing.T) { |
26 | 77 | home, _ := os.UserHomeDir() |
27 | 78 | tests := map[string]struct { |
|
0 commit comments