make mockdevice function public

furiosa-ai · Aug 21, 2024 · 442373d · 442373d
1 parent 0693bef
commit 442373d
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 13 deletions.
diff --git a/pkg/smi/device_error_test.go b/pkg/smi/device_error_test.go
@@ -8,7 +8,7 @@ import (
 )
 
 func testDeviceErrorInfo(arch Arch, t *testing.T, expected DeviceErrorInfo) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	device_error_info, err := mockdevice.DeviceErrorInfo()
 

diff --git a/pkg/smi/device_file_test.go b/pkg/smi/device_file_test.go
@@ -8,7 +8,7 @@ import (
 )
 
 func testDeviceFiles(arch Arch, t *testing.T, expected []DeviceFile) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	device_files, err := mockdevice.DeviceFiles()
 

diff --git a/pkg/smi/device_info_test.go b/pkg/smi/device_info_test.go
@@ -8,7 +8,7 @@ import (
 )
 
 func testDeviceInfo(arch Arch, t *testing.T, expected DeviceInfo) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	device_info, err := mockdevice.DeviceInfo()
 

diff --git a/pkg/smi/device_test.go b/pkg/smi/device_test.go
@@ -6,7 +6,7 @@ import (
 )
 
 func testCoreStatus(arch Arch, t *testing.T, expected map[uint32]CoreStatus) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	core_status, err := mockdevice.CoreStatus()
 
@@ -40,7 +40,7 @@ func TestRngdCoreStatus(t *testing.T) {
 }
 
 func testLiveness(arch Arch, t *testing.T, expected bool) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	liveness, err := mockdevice.Liveness()
 
@@ -84,13 +84,13 @@ func testGetDeviceToDeviceLinkType(devices []Device, t *testing.T) {
 }
 
 func TestWarboyGetDeviceToDeviceLinkType(t *testing.T) {
-	devices := getStaticMockDevices(ArchRngd)
+	devices := GetStaticMockDevices(ArchRngd)
 
 	testGetDeviceToDeviceLinkType(devices, t)
 }
 
 func TestRngdGetDeviceToDeviceLinkType(t *testing.T) {
-	devices := getStaticMockDevices(ArchRngd)
+	devices := GetStaticMockDevices(ArchRngd)
 
 	testGetDeviceToDeviceLinkType(devices, t)
 }
diff --git a/pkg/smi/mock_common.go b/pkg/smi/mock_common.go
@@ -54,15 +54,15 @@ var linkTypeHintMap = map[int]map[int]LinkType{
 	7: {0: LinkTypeInterconnect, 1: LinkTypeInterconnect, 2: LinkTypeInterconnect, 3: LinkTypeInterconnect, 4: LinkTypeCpu, 5: LinkTypeCpu, 6: LinkTypeHostBridge, 7: LinkTypeNoc},
 }
 
-func getStaticMockDevices(arch Arch) (mockDevices []Device) {
+func GetStaticMockDevices(arch Arch) (mockDevices []Device) {
 	for i := range iter.N(8) {
-		mockDevices = append(mockDevices, getStaticMockDevice(arch, i))
+		mockDevices = append(mockDevices, GetStaticMockDevice(arch, i))
 	}
 
 	return
 }
 
-func getStaticMockDevice(arch Arch, nodeIdx int) Device {
+func GetStaticMockDevice(arch Arch, nodeIdx int) Device {
 	switch arch {
 	case ArchWarboy:
 		return &staticWarboyMockDevice{

diff --git a/pkg/smi/performance_test.go b/pkg/smi/performance_test.go
@@ -8,7 +8,7 @@ import (
 )
 
 func testDeviceTemperature(arch Arch, t *testing.T, expected deviceTemperature) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	temperature, err := mockdevice.DeviceTemperature()
 	if err != nil {
@@ -36,7 +36,7 @@ func TestRngdDeviceTemperature(t *testing.T) {
 }
 
 func testPowerConsumption(arch Arch, t *testing.T, expected float64) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	power, err := mockdevice.PowerConsumption()
 	if err != nil {
@@ -61,7 +61,7 @@ func TestRngdPowerConsumption(t *testing.T) {
 }
 
 func testDeviceUtilization(arch Arch, t *testing.T) {
-	mockdevice := getStaticMockDevice(arch, 0)
+	mockdevice := GetStaticMockDevice(arch, 0)
 
 	_, err := mockdevice.DeviceUtilization() // Currenlty, not to check the value.