Use in-memory SSH agent
This commit is contained in:
45
examples/main_test.go
Normal file
45
examples/main_test.go
Normal file
@@ -0,0 +1,45 @@
|
||||
//go:build nodejs || python || dotnet || java || all
|
||||
// +build nodejs python dotnet java all
|
||||
|
||||
package examples
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/crypto/ssh/agent"
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
if os.Getenv("SSH_AUTH_SOCK") == "" {
|
||||
sock := sshagent()
|
||||
os.Setenv("SSH_AUTH_SOCK", sock)
|
||||
}
|
||||
|
||||
os.Exit(m.Run())
|
||||
}
|
||||
|
||||
func sshagent() string {
|
||||
dir := os.TempDir()
|
||||
sock := filepath.Join(dir, "test.sock")
|
||||
|
||||
_ = os.Remove(sock)
|
||||
|
||||
l, err := net.Listen("unix", sock)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
go func() {
|
||||
conn, err := l.Accept()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
a := agent.NewKeyring()
|
||||
agent.ServeAgent(a, conn)
|
||||
}()
|
||||
|
||||
return sock
|
||||
}
|
||||
@@ -4,7 +4,7 @@
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
|
||||
Reference in New Issue
Block a user