zenizh / go-capturer Goto Github PK

type SDCInstanceOutput []scaleio.SdcInstance

// Output - filter results per cli options then output in the correct format
func (i *SDCInstanceOutput) Output(ctx *cli.Context) {
	var filtered SDCInstanceOutput
	for _, row := range *i {
		if ctx.IsSet(FlagNameClientStatus) && strings.Compare(ctx.String(FlagNameClientStatus), "all") != 0 {
			if strings.Compare(ctx.String(FlagNameClientStatus), strings.ToLower(row.MdmConnectionState)) != 0 {
				continue
			}
		}
		filtered = append(filtered, row)
	}
	switch strings.ToLower(ctx.String(FlagNameOutputMode)) {
	case "csv":
		filtered.CSV()
	case "human":
		filtered.Tabular()
	case "json":
		filtered.JSON(ctx)
	default:
		log.Fatalf("ERROR: invalid mode %q", ctx.String(FlagNameOutputMode))
	}
}

		app.Commands = []*cli.Command{
			&cli.Command{
				Name:        "test-cmd",
				Flags:       []cli.Flag{FlagClientStatus, FlagOutputMode},
				Action: func(x *cli.Context) error {
					output = capturer.CaptureOutput(func() {
						c.givenResult.Output(x)
					})
					log.Infof("OUTPUT=%s", output)
					return nil
				},
			},
		}

		if err := app.Run(c.givenArgs); err != nil {
			t.Fatal(err)
		}

		resultCount := lineCount(output)
		if c.wantedRows != resultCount {
			t.Errorf("expecting %d results, got %d", c.wantedRows, resultCount)
		}

% go test  -v
=== RUN   TestSDCInstanceOutput_Output
Name  IP  Active Version  Installed Version  OS Type  Kernel  Status
                                                              Connected
                                                              Connected
time="2021-09-10T16:57:24.084735-05:00" level=info file="commands_test.go" line=61 msg="OUTPUT="
    commands_test.go:73: expecting 2 results, got 0

cmd := exec.Command("whoami")
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
out := capturer.CaptureOutput(func() {
  _ = cmd.Run()
})
fmt.Println(out) // empty