version_test.go

// Copyright 2012, 2013 Canonical Ltd.
// Licensed under the LGPLv3, see LICENSE file for details.

package cmd_test

import (
	"fmt"

	"github.com/juju/loggo/v2"
	"github.com/juju/testing"
	gc "gopkg.in/check.v1"

	"github.com/juju/cmd/v4"
	"github.com/juju/cmd/v4/cmdtesting"
)

type VersionSuite struct {
	testing.LoggingSuite

	ctx *cmd.Context
}

var _ = gc.Suite(&VersionSuite{})

type versionDetail struct {
	Version       string `json:"version"`
	GitCommitHash string `json:"git-commit-hash"`
	GitTreeState  string `json:"git-tree-state"`
}

func (s *VersionSuite) SetUpTest(c *gc.C) {
	s.LoggingSuite.SetUpTest(c)
	s.ctx = cmdtesting.Context(c)
	loggo.ReplaceDefaultWriter(cmd.NewWarningWriter(s.ctx.Stderr))
}

func (s *VersionSuite) TestVersion(c *gc.C) {
	const version = "999.888.777"

	code := cmd.Main(cmd.NewVersionCommand(version, nil), s.ctx, nil)
	c.Check(code, gc.Equals, 0)
	c.Assert(cmdtesting.Stderr(s.ctx), gc.Equals, "")
	c.Assert(cmdtesting.Stdout(s.ctx), gc.Equals, version+"\n")
}

func (s *VersionSuite) TestVersionExtraArgs(c *gc.C) {
	code := cmd.Main(cmd.NewVersionCommand("xxx", nil), s.ctx, []string{"foo"})
	c.Check(code, gc.Equals, 2)
	c.Assert(cmdtesting.Stdout(s.ctx), gc.Equals, "")
	c.Assert(cmdtesting.Stderr(s.ctx), gc.Matches, "ERROR unrecognized args.*\n")
}

func (s *VersionSuite) TestVersionJson(c *gc.C) {
	const version = "999.888.777"

	code := cmd.Main(cmd.NewVersionCommand(version, nil), s.ctx, []string{"--format", "json"})
	c.Check(code, gc.Equals, 0)
	c.Assert(cmdtesting.Stderr(s.ctx), gc.Equals, "")
	c.Assert(cmdtesting.Stdout(s.ctx), gc.Equals, fmt.Sprintf("%q\n", version))
}

func (s *VersionSuite) TestVersionDetailJson(c *gc.C) {
	const version = "999.888.777"
	detail := versionDetail{
		Version:       version,
		GitCommitHash: "46f1a0bd5592a2f9244ca321b129902a06b53e03",
		GitTreeState:  "dirty",
	}

	code := cmd.Main(cmd.NewVersionCommand(version, detail), s.ctx, []string{"--all", "--format", "json"})
	c.Check(code, gc.Equals, 0)
	c.Assert(cmdtesting.Stderr(s.ctx), gc.Equals, "")
	c.Assert(cmdtesting.Stdout(s.ctx), gc.Equals, `
{"version":"999.888.777","git-commit-hash":"46f1a0bd5592a2f9244ca321b129902a06b53e03","git-tree-state":"dirty"}
`[1:])
}