def main():
print("---------------------")
print("Chassis PSU Unit Test")
print("---------------------")
chassis = Chassis()
for psu in chassis.get_all_psus():
if not psu.get_presence():
print(" Name: {} not present".format(psu.get_name()))
else:
print(" Name:", psu.get_name())
print(" Presence: {}, Status: {}, LED: {}".format(
psu.get_presence(), psu.get_status(), psu.get_status_led()))
print(" Model: {}, Serial#: {}, Part#: {}".format(
psu.get_model(), psu.get_serial(), psu.get_part_number()))
try:
current = psu.get_current()
except NotImplementedError:
current = "NA"
try:
power = psu.get_power()
except NotImplementedError:
power = "NA"
print(" Voltage: {}, Current: {}, Power: {}\n".format(
psu.get_voltage(), current, power))
return
def main():
print("---------------------")
print("Chassis PSU Unit Test")
print("---------------------")
chassis = Chassis()
for psu in chassis.get_all_psus():
print(" Name:", psu.get_name())
print(" Presence: {}, Status: {}, LED: {}".format(
psu.get_presence(), psu.get_status(), psu.get_status_led()))
print(" Model: {}, Serial: {}".format(psu.get_model(),
psu.get_serial()))
print(" Voltage: {}, Current: NO, Power: NO \n".format(
psu.get_voltage()))
return
def test_psu(self):
from sonic_platform.psu import Psu, FixedPsu
# Test creating hot swapable PSU
DeviceDataManager.get_psu_count = mock.MagicMock(return_value=2)
DeviceDataManager.is_psu_hotswapable = mock.MagicMock(
return_value=True)
chassis = Chassis()
chassis.initialize_psu()
assert len(chassis._psu_list) == 2
assert len(
list(filter(lambda x: isinstance(x, Psu), chassis._psu_list))) == 2
# Test creating fixed PSU
DeviceDataManager.get_psu_count = mock.MagicMock(return_value=3)
DeviceDataManager.is_psu_hotswapable = mock.MagicMock(
return_value=False)
chassis._psu_list = []
chassis.initialize_psu()
assert len(chassis._psu_list) == 3
assert len(
list(filter(lambda x: isinstance(x, FixedPsu),
chassis._psu_list))) == 3
# Test chassis.get_all_psus
chassis._psu_list = []
psu_list = chassis.get_all_psus()
assert len(psu_list) == 3
# Test chassis.get_psu
chassis._psu_list = []
psu = chassis.get_psu(0)
assert psu and isinstance(psu, FixedPsu)
psu = chassis.get_psu(3)
assert psu is None
# Test chassis.get_num_psus
chassis._psu_list = []
assert chassis.get_num_psus() == 3