/*
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * openFuyao is licensed under Mulan PSL v2.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *          http://license.coscl.org.cn/MulanPSL2
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 */

package util

import (
	"os"
	"testing"

	"huawei.com/vxpu-device-plugin/pkg/lock"
	"huawei.com/vxpu-device-plugin/pkg/plugin/config"
	v1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes/fake"
)

func TestParseNodeModeFromConfigMap(t *testing.T) {
	yamlContent := `
nodeConfigs:
- name: node1
  vnpu-mode: soft
- name: node2
  vnpu-mode: hard
`
	configMap := &v1.ConfigMap{
		ObjectMeta: metav1.ObjectMeta{
			Name:      nodeConfigCmName,
			Namespace: "xpu",
		},
		Data: map[string]string{
			nodeConfigYamlName: yamlContent,
		},
	}
	lock.SetClient(fake.NewClientset(configMap))

	os.Setenv("POD_NAMESPACE", "xpu")
	config.NodeName = "node1"
	config.NodeMode = ""
	parseNodeModeFromConfigMap()
	if config.NodeMode != "soft" {
		t.Error("parse node mode from configmap failed")
	}

	config.NodeName = "node2"
	config.NodeMode = ""
	parseNodeModeFromConfigMap()
	if config.NodeMode != "hard" {
		t.Error("parse node mode from configmap failed")
	}

	config.NodeName = "node3"
	config.NodeMode = ""
	parseNodeModeFromConfigMap()
	if config.NodeMode != "soft" {
		t.Error("parse node mode from configmap failed")
	}
}

func TestParseEmptyNodeModeFromConfigMap(t *testing.T) {
	yamlContent := `
nodeConfigs:
`
	configMap := &v1.ConfigMap{
		ObjectMeta: metav1.ObjectMeta{
			Name:      nodeConfigCmName,
			Namespace: "xpu",
		},
		Data: map[string]string{
			nodeConfigYamlName: yamlContent,
		},
	}
	lock.SetClient(fake.NewClientset(configMap))

	os.Setenv("POD_NAMESPACE", "xpu")
	config.NodeName = "node"
	config.NodeMode = ""
	parseNodeModeFromConfigMap()
	if config.NodeMode != "soft" {
		t.Error("parse node mode from configmap failed")
	}
}