diff options
| author | Ash McKenzie <amckenzie@gitlab.com> | 2022-04-14 05:47:34 +0000 |
|---|---|---|
| committer | Ash McKenzie <amckenzie@gitlab.com> | 2022-04-14 05:47:34 +0000 |
| commit | 5fe0d17f11eb122d45bad9126c7c0646e196b1c5 (patch) | |
| tree | bc89cbdc9ec3ca400305975e83363669fe639e69 /internal/sshd/session_test.go | |
| parent | 9abd6d79043f90057ff9a7bf9deecb868e83c46d (diff) | |
| parent | cc353a57f3aa8f3f751c175ea596d2baca0b1f19 (diff) | |
| download | gitlab-shell-5fe0d17f11eb122d45bad9126c7c0646e196b1c5.tar.gz | |
Merge branch 'id-metrics-for-results' into 'main'
Add additional metrics to gitlab-sshd
See merge request gitlab-org/gitlab-shell!593
Diffstat (limited to 'internal/sshd/session_test.go')
| -rw-r--r-- | internal/sshd/session_test.go | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/internal/sshd/session_test.go b/internal/sshd/session_test.go index f135825..d0cc8d4 100644 --- a/internal/sshd/session_test.go +++ b/internal/sshd/session_test.go @@ -99,6 +99,7 @@ func TestHandleExec(t *testing.T) { expectedExecCmd string sentRequestName string sentRequestPayload []byte + success bool }{ { desc: "invalid payload", @@ -111,6 +112,7 @@ func TestHandleExec(t *testing.T) { expectedExecCmd: "discover", sentRequestName: "exit-status", sentRequestPayload: ssh.Marshal(exitStatusReq{ExitStatus: 0}), + success: true, }, } @@ -130,6 +132,7 @@ func TestHandleExec(t *testing.T) { require.Equal(t, false, s.handleExec(context.Background(), r)) require.Equal(t, tc.sentRequestName, f.sentRequestName) require.Equal(t, tc.sentRequestPayload, f.sentRequestPayload) + require.Equal(t, tc.success, s.success) }) } } @@ -141,6 +144,7 @@ func TestHandleShell(t *testing.T) { errMsg string gitlabKeyId string expectedExitCode uint32 + success bool }{ { desc: "fails to parse command", |