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) }