mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-29 04:55:31 +00:00
da8def8e13
The //go:build syntax was introduced in Go 1.17: https://go.dev/doc/go1.17#build-lines gofmt has kept the +build and go:build lines in sync since then, but enough time has passed. Time to remove them. Done with: perl -i -npe 's,^// \+build.*\n,,' $(git grep -l -F '+build') Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
//go:build glidertests
|
|
|
|
package ssh
|
|
|
|
import "testing"
|
|
|
|
func TestSetPermissions(t *testing.T) {
|
|
t.Parallel()
|
|
permsExt := map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
session, _, cleanup := newTestSessionWithOptions(t, &Server{
|
|
Handler: func(s Session) {
|
|
if _, ok := s.Permissions().Extensions["foo"]; !ok {
|
|
t.Fatalf("got %#v; want %#v", s.Permissions().Extensions, permsExt)
|
|
}
|
|
},
|
|
}, nil, PasswordAuth(func(ctx Context, password string) bool {
|
|
ctx.Permissions().Extensions = permsExt
|
|
return true
|
|
}))
|
|
defer cleanup()
|
|
if err := session.Run(""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestSetValue(t *testing.T) {
|
|
t.Parallel()
|
|
value := map[string]string{
|
|
"foo": "bar",
|
|
}
|
|
key := "testValue"
|
|
session, _, cleanup := newTestSessionWithOptions(t, &Server{
|
|
Handler: func(s Session) {
|
|
v := s.Context().Value(key).(map[string]string)
|
|
if v["foo"] != value["foo"] {
|
|
t.Fatalf("got %#v; want %#v", v, value)
|
|
}
|
|
},
|
|
}, nil, PasswordAuth(func(ctx Context, password string) bool {
|
|
ctx.SetValue(key, value)
|
|
return true
|
|
}))
|
|
defer cleanup()
|
|
if err := session.Run(""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|