mirror of
https://github.com/gravitational/teleport.git
synced 2026-10-16 07:44:31 +08:00
Ev Kontsevoy
cca475924b
Fixes #392 Fixes #396 Teleport now respects `--user` flag and, if --user is specified, forces the certificate to belong to the given user. This changes the file structure in `~/.tsh` directory. If a user logs in under two different accounts, say "ekontsevoy" and "vince", it looks like this: ``` ~/.tsh/ ├── keys │ └── localhost │ ├── ekontsevoy.cert │ ├── ekontsevoy.key │ ├── ekontsevoy.pub │ ├── vince.cert │ ├── vince.key │ └── vince.pub └── known_hosts ``` Also, to make tests more believable, I have added 3 more pre-generated keys to 'testauthority' fixture, so instead of returning the same key over and over, it now returns a random 1 of 4
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
/*
|
|
Copyright 2016 Gravitational, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
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
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package client
|
|
|
|
import (
|
|
"github.com/gravitational/teleport/lib/utils"
|
|
"gopkg.in/check.v1"
|
|
"testing"
|
|
)
|
|
|
|
// register test suite
|
|
type APITestSuite struct {
|
|
}
|
|
|
|
// bootstrap check
|
|
func TestClientAPI(t *testing.T) { check.TestingT(t) }
|
|
|
|
var _ = check.Suite(&APITestSuite{})
|
|
|
|
func (s *APITestSuite) SetUpSuite(c *check.C) {
|
|
utils.InitLoggerForTests()
|
|
}
|
|
|
|
func (s *APITestSuite) TestConfig(c *check.C) {
|
|
var conf Config
|
|
c.Assert(conf.ProxySpecified(), check.Equals, false)
|
|
conf.ProxyHost = "example.org"
|
|
c.Assert(conf.ProxySpecified(), check.Equals, true)
|
|
c.Assert(conf.ProxyHostPort(12), check.Equals, "example.org:12")
|
|
|
|
conf.ProxyHost = "example.org:100"
|
|
c.Assert(conf.ProxySpecified(), check.Equals, true)
|
|
c.Assert(conf.ProxyHostPort(12), check.Equals, "example.org:100")
|
|
}
|
|
|
|
func (s *APITestSuite) TestNew(c *check.C) {
|
|
conf := Config{
|
|
Host: "localhost",
|
|
HostLogin: "vincent",
|
|
HostPort: 22,
|
|
KeysDir: "/tmp",
|
|
Username: "localuser",
|
|
ProxyHost: "proxy",
|
|
SiteName: "site",
|
|
}
|
|
tc, err := NewClient(&conf)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(tc, check.NotNil)
|
|
|
|
la := tc.LocalAgent()
|
|
c.Assert(la, check.NotNil)
|
|
|
|
c.Assert(tc.NodeHostPort(), check.Equals, "localhost:22")
|
|
c.Assert(tc.ProxySpecified(), check.Equals, true)
|
|
c.Assert(tc.ProxyHostPort(12), check.Equals, "proxy:12")
|
|
}
|
|
|
|
func (s *APITestSuite) TestParseLabels(c *check.C) {
|
|
// simplest case:
|
|
m, err := ParseLabelSpec("key=value")
|
|
c.Assert(m, check.NotNil)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(m, check.DeepEquals, map[string]string{
|
|
"key": "value",
|
|
})
|
|
// multiple values:
|
|
m, err = ParseLabelSpec(`type="database";" role"=master,ver="mongoDB v1,2"`)
|
|
c.Assert(m, check.NotNil)
|
|
c.Assert(err, check.IsNil)
|
|
c.Assert(m, check.HasLen, 3)
|
|
c.Assert(m["role"], check.Equals, "master")
|
|
c.Assert(m["type"], check.Equals, "database")
|
|
c.Assert(m["ver"], check.Equals, "mongoDB v1,2")
|
|
// invalid specs
|
|
m, err = ParseLabelSpec(`type="database,"role"=master,ver="mongoDB v1,2"`)
|
|
c.Assert(m, check.IsNil)
|
|
c.Assert(err, check.NotNil)
|
|
m, err = ParseLabelSpec(`type="database",role,master`)
|
|
c.Assert(m, check.IsNil)
|
|
c.Assert(err, check.NotNil)
|
|
}
|
|
|
|
func (s *APITestSuite) TestSCPParsing(c *check.C) {
|
|
user, host, dest := parseSCPDestination("root@remote.host:/etc/nginx.conf")
|
|
c.Assert(user, check.Equals, "root")
|
|
c.Assert(host, check.Equals, "remote.host")
|
|
c.Assert(dest, check.Equals, "/etc/nginx.conf")
|
|
|
|
user, host, dest = parseSCPDestination("remote.host:/etc/nginx.conf")
|
|
c.Assert(user, check.Equals, "")
|
|
c.Assert(host, check.Equals, "remote.host")
|
|
c.Assert(dest, check.Equals, "/etc/nginx.conf")
|
|
}
|