package cmd

import (
	"errors"
	"testing"

	"github.com/runconduit/conduit/controller"
	"github.com/runconduit/conduit/controller/api/public"
	pb "github.com/runconduit/conduit/controller/gen/public"
)

func TestGetVersion(t *testing.T) {
	t.Run("Returns existing versions from server and client", func(t *testing.T) {
		expectedServerVersion := "1.2.3"
		expectedClientVersion := controller.Version
		mockClient := &public.MockConduitApiClient{}
		mockClient.VersionInfoToReturn = &pb.VersionInfo{
			ReleaseVersion: expectedServerVersion,
		}

		versions := getVersions(mockClient)

		if versions.Client != expectedClientVersion || versions.Server != expectedServerVersion {
			t.Fatalf("Expected client version to be [%s], was [%s]; expecting server version to be [%s], was [%s]",
				versions.Client, expectedClientVersion, versions.Server, expectedServerVersion)
		}
	})

	t.Run("Returns undefined when cannot get server version", func(t *testing.T) {
		expectedServerVersion := "unavailable"
		expectedClientVersion := controller.Version
		mockClient := &public.MockConduitApiClient{}
		mockClient.ErrorToReturn = errors.New("expected")

		versions := getVersions(mockClient)

		if versions.Client != expectedClientVersion || versions.Server != expectedServerVersion {
			t.Fatalf("Expected client version to be [%s], was [%s]; expecting server version to be [%s], was [%s]",
				expectedClientVersion, versions.Client, expectedServerVersion, versions.Server)
		}
	})
}