package python

import (
	"bytes"
	"os"
	"strings"
	"testing"
	cli "v/cli"
	logger "v/logger"
	state "v/state"
	testutils "v/testutils"
)

func TestListVersionOutputsNoticeIfNoVersionsInstalled(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	os.MkdirAll(state.GetStatePath("runtimes", "python"), 0750)
	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	listVersions([]string{}, cli.Flags{}, state.State{})

	captured := out.String()
	if captured != "No versions installed!\n" {
		t.Errorf("Unexpected message: %s", captured)
	}
}

func TestListVersionOutputsVersionsInstalled(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	os.MkdirAll(state.GetStatePath("runtimes", "python", "1.2.3"), 0750)
	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	listVersions([]string{}, cli.Flags{}, state.State{})

	captured := out.String()
	if captured != "1.2.3\n" {
		t.Errorf("Unexpected message: %s", captured)
	}
}

func TestListVersionReturnsErrorOnFailure(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	err := listVersions([]string{}, cli.Flags{}, state.State{})

	captured := out.String()
	if captured != "" {
		t.Errorf("Captured unexpected message: %s", captured)
	}

	if err == nil {
		t.Errorf("Expected error returned, did not get one.")
	}
}

func TestListVersionOutputsVersionSelectedAndWarnsNotInstalled(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	which([]string{}, cli.Flags{}, state.State{GlobalVersion: "1.2.3"})

	captured := out.String()
	if captured != "The desired version (1.2.3) is not installed.\n" {
		t.Errorf("Unexpected message: %s", captured)
	}
}

func TestWhichOutputsVersionSelectedIfInstalled(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	os.MkdirAll(state.GetStatePath("runtimes", "python", "1.2.3"), 0750)
	which([]string{}, cli.Flags{}, state.State{GlobalVersion: "1.2.3"})

	captured := strings.TrimSpace(out.String())
	expected := state.GetStatePath("runtimes", "python", "1.2.3", "bin", "python1.2")
	if !strings.Contains(captured, expected) {
		t.Errorf("Unexpected message: %s, not %s", captured, expected)
	}
}

func TestWhichOutputsSystemVersionIfNoneSelected(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	which([]string{}, cli.Flags{RawOutput: true}, state.State{})

	captured := strings.TrimSpace(out.String())

	if captured != "/bin/python (system)" {
		t.Errorf("%s != %s", captured, "/bin/python (system)")
	}
}

func TestWhichOutputsVersionWithoutPrefixesIfRawOutput(t *testing.T) {
	defer testutils.SetupAndCleanupEnvironment(t)()

	var out bytes.Buffer

	logger.InfoLogger.SetOutput(&out)
	defer logger.InfoLogger.SetOutput(os.Stdout)

	os.MkdirAll(state.GetStatePath("runtimes", "python", "1.2.3"), 0750)
	which([]string{}, cli.Flags{RawOutput: true}, state.State{GlobalVersion: "1.2.3"})

	captured := strings.TrimSpace(out.String())
	expected := state.GetStatePath("runtimes", "python", "1.2.3", "bin", "python1.2")
	if captured != expected {
		t.Errorf("Unexpected message: %s, not %s", captured, expected)
	}
}