mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-26 11:35:35 +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>
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build linux || (darwin && !ios)
|
|
|
|
package ipnlocal
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"tailscale.com/tailcfg"
|
|
"tailscale.com/util/must"
|
|
)
|
|
|
|
func TestSSHKeyGen(t *testing.T) {
|
|
dir := t.TempDir()
|
|
lb := &LocalBackend{varRoot: dir}
|
|
keys, err := lb.getTailscaleSSH_HostKeys(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got := map[string]bool{}
|
|
for _, k := range keys {
|
|
got[k.PublicKey().Type()] = true
|
|
}
|
|
want := map[string]bool{
|
|
"ssh-rsa": true,
|
|
"ecdsa-sha2-nistp256": true,
|
|
"ssh-ed25519": true,
|
|
}
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("keys = %v; want %v", got, want)
|
|
}
|
|
|
|
keys2, err := lb.getTailscaleSSH_HostKeys(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(keys, keys2) {
|
|
t.Errorf("got different keys on second call")
|
|
}
|
|
}
|
|
|
|
type fakeSSHServer struct {
|
|
SSHServer
|
|
}
|
|
|
|
func TestGetSSHUsernames(t *testing.T) {
|
|
b := new(LocalBackend)
|
|
b.sshServer = fakeSSHServer{}
|
|
res, err := b.getSSHUsernames(new(tailcfg.C2NSSHUsernamesRequest))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("Got: %s", must.Get(json.Marshal(res)))
|
|
}
|