package main

import (
	"io/ioutil"
	"os"
	"strings"
	"testing"
	cli "v/cli"
	python "v/python"
	state "v/state"
	testutils "v/testutils"
)

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

	os.Mkdir(state.GetStatePath("shims"), 0775)
	testShimPath := state.GetStatePath("shims", "testshim")
	e := writeShim(testShimPath, "testcommand")

	shimContent, _ := ioutil.ReadFile(testShimPath)

	if e != nil {
		t.Errorf("Errored while writing shim")
	}

	if !strings.Contains(string(shimContent), "testcommand") {
		t.Errorf("Expected shim to contain pass-through via 'which', got %s", shimContent)
	}

}

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

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

	if err != nil {
		t.Errorf("Unexpected error initializing")
	}

	if _, err = os.Stat(state.GetStatePath()); os.IsNotExist(err) {
		t.Errorf("Root state directory not found")
	}

	if _, err = os.Stat(state.GetStatePath("shims")); os.IsNotExist(err) {
		t.Errorf("Shims directory not found")
	}

	if _, err = os.Stat(state.GetStatePath("cache")); os.IsNotExist(err) {
		t.Errorf("Cache directory not found")
	}

	if _, err = os.Stat(state.GetStatePath("runtimes")); os.IsNotExist(err) {
		t.Errorf("Runtimes directory not found")
	}
}

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

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

	if err != nil {
		t.Errorf("Unexpected error initializing")
	}

	expectedShims := python.Shims

	for shimLabel, shimCall := range expectedShims {
		shimContent, err := os.ReadFile(state.GetStatePath("shims", shimLabel))

		if os.IsNotExist(err) {
			t.Errorf("%s shim not created", shimLabel)
		}

		if !strings.Contains(string(shimContent), shimCall) {
			t.Errorf("%s shim does not contain expected call (%s not in %s)", shimLabel, shimCall, shimContent)
		}
	}
}