// Copyright © 2018 The Knative Authors
//
// 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 commands

import (
	"bytes"
	"testing"
	"text/template"

	"github.com/spf13/cobra"
	"gotest.tools/assert"
)

type versionOutput struct {
	Version      string
	BuildDate    string
	GitRevision  string
	VersionsAPIs *VersionsAPIs
}

var versionOutputTemplate = `Version:      {{.Version}}
Build Date:   {{.BuildDate}}
Git Revision: {{.GitRevision}}
Support:
- Serving: {{index .VersionsAPIs.Versions 0}}  {{index .VersionsAPIs.Versions 1}}
- API(s):  {{index .VersionsAPIs.APIs 0}}
`

const (
	fakeVersion     = "fake-version"
	fakeBuildDate   = "fake-build-date"
	fakeGitRevision = "fake-git-revision"
)

func TestVersion(t *testing.T) {
	var (
		versionCmd            *cobra.Command
		knParams              *KnParams
		expectedVersionOutput string
		output                *bytes.Buffer
	)

	setup := func() {
		Version = fakeVersion
		BuildDate = fakeBuildDate
		GitRevision = fakeGitRevision
		ServingVersion = knServingDep

		expectedVersionOutput = genVersionOuput(t, versionOutputTemplate,
			versionOutput{
				fakeVersion,
				fakeBuildDate,
				fakeGitRevision,
				supportMatrix[ServingVersion]})

		knParams = &KnParams{}
		versionCmd = NewVersionCommand(knParams)
		output = new(bytes.Buffer)
		versionCmd.SetOutput(output)
	}

	t.Run("creates a VersionCommand", func(t *testing.T) {
		setup()

		assert.Equal(t, versionCmd.Use, "version")
		assert.Equal(t, versionCmd.Short, "Prints the client version")
		assert.Assert(t, versionCmd.RunE != nil)
	})

	t.Run("prints version, build date, git revision, supported serving version and APIs", func(t *testing.T) {
		setup()

		err := versionCmd.RunE(versionCmd, []string{})
		assert.NilError(t, err)
		assert.Equal(t, output.String(), expectedVersionOutput)
	})

}

// Private

func genVersionOuput(t *testing.T, templ string, vOutput versionOutput) string {
	tmpl, err := template.New("versionOutput").Parse(versionOutputTemplate)
	assert.NilError(t, err)

	buf := bytes.Buffer{}
	err = tmpl.Execute(&buf, vOutput)
	assert.NilError(t, err)

	return buf.String()
}