Merge branch 'id-sshd-tests' into 'main'

Add TestInvalidClientConfig and TestNewServerWithoutHosts for sshd.go

See merge request gitlab-org/gitlab-shell!518

1 files changed, 16 insertions, 0 deletions

diff --git a/internal/sshd/sshd_test.go b/internal/sshd/sshd_test.go
index 32946af..cba1c3f 100644
--- a/internal/sshd/sshd_test.go
+++ b/internal/sshd/sshd_test.go
@@ -104,6 +104,22 @@ func TestLivenessProbe(t *testing.T) {
 	require.Equal(t, 200, r.Result().StatusCode)
 }
 
+func TestNewServerWithoutHosts(t *testing.T) {
+	_, err := NewServer(&config.Config{GitlabUrl: "http://localhost"})
+
+	require.Error(t, err)
+	require.Equal(t, "No host keys could be loaded, aborting", err.Error())
+}
+
+func TestInvalidClientConfig(t *testing.T) {
+	setupServer(t)
+
+	cfg := clientConfig(t)
+	cfg.User = "unknown"
+	_, err := ssh.Dial("tcp", serverUrl, cfg)
+	require.Error(t, err)
+}
+
 func setupServer(t *testing.T) *Server {
 	t.Helper()