src/cmd/cos_gpu_installer/features/config_reader_test.go

package features

import (
	"os"
	"reflect"
	"testing"
)

func TestReadCOSGPUConfigFileExists(t *testing.T) {
	tmpFile, err := os.CreateTemp("", "cos-gpu-config")
	if err != nil {
		t.Fatalf("Failed to create temp file: %v", err)
	}
	defer os.Remove(tmpFile.Name())
	_, err = tmpFile.WriteString(`{"use-proto": true}`)
	if err != nil {
		t.Fatalf("Failed to write to tmp file: %v", err)
	}
	tmpFile.Close()
	gpuConfig, err := ReadCOSGPUConfig(tmpFile.Name())
	if err != nil {
		t.Fatalf("Read cos gpu config failed: %v", err)
	}
	var expectedGPUConfig = &GPUConfig{UseProto: true}
	if !reflect.DeepEqual(gpuConfig, expectedGPUConfig) {
		t.Errorf("Expected result: %+v, got result: %+v", expectedGPUConfig, gpuConfig)
	}
}

func TestReadCOSGPUConfigFileNotExists(t *testing.T) {
	gpuConfig, err := ReadCOSGPUConfig("not-exist-config-file.json")
	if err != nil {
		t.Errorf("Expected no error when reding not exist config file, but we got: %v", err)
	}
	var expectedDefaultGPUConfig = &GPUConfig{UseProto: false}
	if !reflect.DeepEqual(gpuConfig, expectedDefaultGPUConfig) {
		t.Errorf("Expected result: %+v, got result: %+v", expectedDefaultGPUConfig, gpuConfig)
	}
}