environment_test.go

package support

import (
	"fmt"
	"os"
	"testing"

	"github.com/onsi/gomega"
)

func TestGetRayVersion(t *testing.T) {

	g := gomega.NewGomegaWithT(t)
	// Set the environment variable.
	os.Setenv(CodeFlareTestRayVersion, "1.4.5")

	// Get the version.
	version := GetRayVersion()

	// Assert that the version is correct.

	g.Expect(version).To(gomega.Equal("1.4.5"), "Expected version 1.4.5, but got %s", version)

}

func TestGetRayImage(t *testing.T) {

	g := gomega.NewGomegaWithT(t)
	// Set the environment variable.
	os.Setenv(CodeFlareTestRayImage, "ray/ray:latest")

	// Get the image.
	image := GetRayImage()

	// Assert that the image is correct.

	g.Expect(image).To(gomega.Equal("ray/ray:latest"), "Expected image ray/ray:latest, but got %s", image)

}

func TestGetPyTorchImage(t *testing.T) {

	g := gomega.NewGomegaWithT(t)
	// Set the environment variable.
	os.Setenv(CodeFlareTestPyTorchImage, "pytorch/pytorch:latest")

	// Get the image.
	image := GetPyTorchImage()

	// Assert that the image is correct.

	g.Expect(image).To(gomega.Equal("pytorch/pytorch:latest"), "Expected image pytorch/pytorch:latest, but got %s", image)

}

func TestGetClusterID(t *testing.T) {

	g := gomega.NewGomegaWithT(t)
	os.Setenv(ClusterID, "my-cluster-id")
	clusterId, ok := GetClusterId()
	if !ok {
		gomega.Expect(ok).To(gomega.BeTrue(), "Expected GetClusterId() to return true, but got false.")
	}

	g.Expect(clusterId).To(gomega.Equal("my-cluster-id"), "Expected GetClusterId() to return 'my-cluster-id', but got '%s'.", clusterId)

}

func TestGetInstascaleOcmSecret(t *testing.T) {

	g := gomega.NewGomegaWithT(t)
	// Set the Instascale OCM secret environment variable.
	os.Setenv(InstaScaleOcmSecret, "default/instascale-ocm-secret")
	// Get the Instascale OCM secret namespace and secret name.
	namespace, secretName := GetInstascaleOcmSecret()

	// Verify that the namespace and secret name are correct.

	g.Expect(fmt.Sprintf("%s/%s", namespace, secretName)).To(
		gomega.Equal("default/instascale-ocm-secret"),
		"Expected GetInstascaleOcmSecret() to return 'default/instascale-ocm-secret', but got '%s/%s'.",
		namespace, secretName,
	)

}

func TestGetClusterType(t *testing.T) {

	g := gomega.NewGomegaWithT(t)

	tests := []struct {
		name        string
		envVarValue string
		expected    ClusterType
	}{
		{
			name:        "OSD cluster",
			envVarValue: "OSD",
			expected:    OsdCluster,
		},
		{
			name:        "OCP cluster",
			envVarValue: "OCP",
			expected:    OcpCluster,
		},
		{
			name:        "Hypershift cluster",
			envVarValue: "HYPERSHIFT",
			expected:    HypershiftCluster,
		},
		{
			name:        "KIND cluster",
			envVarValue: "KIND",
			expected:    KindCluster,
		},
	}
	ttt := With(t)
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			os.Setenv(ClusterTypeEnvVar, tt.envVarValue)
			actual := GetClusterType(ttt)

			g.Expect(actual).To(
				gomega.Equal(tt.expected),
				"Expected GetClusterType() to return %v, but got %v", tt.expected, actual,
			)

		})
	}
}