func (s *S) TestTokenGenWithExportOn(c *gocheck.C) {
var stdout, stderr bytes.Buffer
context := cmd.Context{
Args: []string{"myapp"},
Stdout: &stdout,
Stderr: &stderr,
}
manager := cmd.NewManager("glb", "0.2", "ad-ver", &stdout, &stderr, nil, nil)
result := `{"token":"secret123"}`
trans := testing.ConditionalTransport{
Transport: testing.Transport{Message: result, Status: http.StatusOK},
CondFunc: func(req *http.Request) bool {
defer req.Body.Close()
body, err := ioutil.ReadAll(req.Body)
c.Assert(err, gocheck.IsNil)
c.Assert(string(body), gocheck.Equals, `{"client":"myapp","export":true}`)
return req.Method == "POST" && req.URL.Path == "/tokens"
},
}
expected := `Application token: "secret123".` + "\n"
client := cmd.NewClient(&http.Client{Transport: &trans}, nil, manager)
command := tokenGen{}
command.Flags().Parse(true, []string{"-e"})
err := command.Run(&context, client)
c.Assert(err, gocheck.IsNil)
c.Assert(stdout.String(), gocheck.Equals, expected)
}
func buildManager() *cmd.Manager {
m := cmd.NewManager("tsr", "0.1.0", "", os.Stdout, os.Stderr, os.Stdin)
m.Register(&apiCmd{})
m.Register(&collectorCmd{})
m.Register(&tokenCmd{})
return m
}
func (s *S) SetUpTest(c *C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", version, header, &stdout, &stderr, os.Stdin)
*tsuru.AppName = ""
*AssumeYes = false
*NumUnits = 1
}
func (s *S) TestCommandsFromBaseManagerAreRegistered(c *gocheck.C) {
baseManager := cmd.NewManager("tsr", "0.3.0", "", os.Stdout, os.Stderr, os.Stdin, nil)
manager := buildManager()
for name, instance := range baseManager.Commands {
command, ok := manager.Commands[name]
c.Assert(ok, gocheck.Equals, true)
c.Assert(command, gocheck.FitsTypeOf, instance)
}
}
func buildManager() *cmd.Manager {
m := cmd.NewManager("tsr", "0.3.0-dev", "", os.Stdout, os.Stderr, os.Stdin)
m.Register(&tsrCommand{Command: &apiCmd{}})
m.Register(&tsrCommand{Command: &collectorCmd{}})
m.Register(&tsrCommand{Command: tokenCmd{}})
m.Register(&tsrCommand{Command: &healerCmd{}})
registerProvisionersCommands(m)
return m
}
func (s *S) TestTokenRun(c *gocheck.C) {
var stdout, stderr bytes.Buffer
context := cmd.Context{
Args: []string{},
Stdout: &stdout,
Stderr: &stderr,
}
manager := cmd.NewManager("glb", "", "", &stdout, &stderr, os.Stdin, nil)
client := cmd.NewClient(&http.Client{}, nil, manager)
command := tokenCmd{}
err := command.Run(&context, client)
c.Assert(err, gocheck.IsNil)
c.Assert(stdout.String(), gocheck.Not(gocheck.Equals), "")
}
func (s *S) SetUpSuite(c *gocheck.C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", version, header, &stdout, &stderr, os.Stdin, nil)
targetFile := os.Getenv("HOME") + "/.tsuru_target"
_, err := os.Stat(targetFile)
if err == nil {
old := targetFile + ".old"
s.recover = []string{"mv", old, targetFile}
exec.Command("mv", targetFile, old).Run()
} else {
s.recover = []string{"rm", targetFile}
}
f, err := os.Create(targetFile)
c.Assert(err, gocheck.IsNil)
f.Write([]byte("http://localhost"))
f.Close()
}
func (s *S) TestTokenRun(c *gocheck.C) {
config.Set("database:url", "127.0.0.1:27017")
config.Set("database:name", "tsuru_tsr_test")
var stdout, stderr bytes.Buffer
context := cmd.Context{
Args: []string{},
Stdout: &stdout,
Stderr: &stderr,
}
manager := cmd.NewManager("glb", "", "", &stdout, &stderr, os.Stdin)
client := cmd.NewClient(&http.Client{}, nil, manager)
command := tokenCmd{}
command.Flags().Parse(true, []string{"--config", "../../etc/tsuru.conf"})
err := command.Run(&context, client)
c.Assert(err, gocheck.IsNil)
c.Assert(stdout.String(), gocheck.Not(gocheck.Equals), "")
}
func (s *S) SetUpTest(c *C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", "0.x", "Foo-Tsuru", &stdout, &stderr, os.Stdin)
AppName = new(string)
AssumeYes = new(bool)
}
func (s *S) SetUpTest(c *gocheck.C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", version, "Supported-Crane", &stdout, &stderr, os.Stdin)
}
func (s *S) SetUpTest(c *gocheck.C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", "0.x", "Foo-Tsuru", &stdout, &stderr, os.Stdin)
}
func (s *S) SetUpTest(c *gocheck.C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", version, header, &stdout, &stderr, os.Stdin, nil)
}
func (s *S) SetUpTest(c *C) {
var stdout, stderr bytes.Buffer
manager = cmd.NewManager("glb", version, header, &stdout, &stderr, os.Stdin)
tsuru.AppName = new(string)
tsuru.AssumeYes = new(bool)
}