package pythontest

import (
	"context"
	"os/exec"
	"path/filepath"
	"testing"

	"github.com/databricks/cli/libs/python"
	"github.com/stretchr/testify/require"
)

func TestVenvSuccess(t *testing.T) {
	// Test at least two version to ensure we capture a case where venv version does not match system one
	for _, pythonVersion := range []string{"3.11", "3.12"} {
		t.Run(pythonVersion, func(t *testing.T) {
			ctx := context.Background()
			dir := t.TempDir()
			opts := VenvOpts{
				PythonVersion: pythonVersion,
				Dir:           dir,
			}
			RequireActivatedPythonEnv(t, ctx, &opts)
			require.DirExists(t, opts.EnvPath)
			require.DirExists(t, opts.BinPath)
			require.FileExists(t, opts.PythonExe)

			pythonExe, err := exec.LookPath(python.GetExecutable())
			require.NoError(t, err)
			require.Equal(t, filepath.Dir(pythonExe), filepath.Dir(opts.PythonExe))
			require.FileExists(t, pythonExe)
		})
	}
}

func TestWrongVersion(t *testing.T) {
	require.Error(t, CreatePythonEnv(&VenvOpts{PythonVersion: "4.0"}))
}

func TestMissingVersion(t *testing.T) {
	require.Error(t, CreatePythonEnv(nil))
	require.Error(t, CreatePythonEnv(&VenvOpts{}))
}