sonic-buildimage/platform/mellanox/mlnx-platform-api/tests/mock_platform.py

59 lines
1.2 KiB
Python

class MockFan:
speed = 60
def __init__(self):
self.presence = True
self.status = True
def get_presence(self):
return self.presence
def set_speed(self, speed):
MockFan.speed = speed
def get_status(self):
return self.status
def get_target_speed(self):
return MockFan.speed
class MockPsu:
def __init__(self):
self.presence = True
self.powergood = True
def get_presence(self):
return self.presence
def get_powergood_status(self):
return self.powergood
def get_all_fans(self):
return []
class MockChassis:
def __init__(self):
self.fan_list = []
self.psu_list = []
def get_all_psus(self):
return self.psu_list
def get_all_fans(self):
return self.fan_list
def get_thermal_manager(self):
from sonic_platform.thermal_manager import ThermalManager
return ThermalManager
def make_fan_absence(self):
fan = MockFan()
fan.presence = False
self.fan_list.append(fan)
def make_psu_absence(self):
psu = MockPsu()
psu.presence = False
self.psu_list.append(psu)