test environment ssh config

This commit is contained in:
Joe Kralicky 2025-04-09 19:33:28 +00:00
parent 8d05d3d4b8
commit 0e56f12b83
No known key found for this signature in database
GPG key ID: 75C4875F34A9FB79
3 changed files with 54 additions and 26 deletions

View file

@ -5,6 +5,7 @@ import (
"fmt"
"log"
"net"
"strings"
"github.com/pomerium/pomerium/internal/testenv"
"github.com/pomerium/pomerium/internal/testenv/values"
@ -201,7 +202,7 @@ func (h *sshUpstream) handleConnection(ctx context.Context, conn net.Conn) {
// Dial implements SSHUpstream.
func (h *sshUpstream) Dial(r testenv.Route, config *ssh.ClientConfig) (*ssh.Client, error) {
return ssh.Dial("tcp", h.Env().Config().Options.SSHAddr, config)
return ssh.Dial("tcp", strings.TrimPrefix(r.URL().Value(), "ssh://"), config)
}
// DirectDial implements SSHUpstream.