def test_update_master_status(self):
chassis = MockChassis()
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
# Test good values while in bad state
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == True
assert chassis_info.master_status_good == True
# Test good values while in good state
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test unknown total_supplied_power (0.0)
chassis_info.total_supplied_power = 0.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test unknown total_consumed_power (0.0)
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 0.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test bad values while in good state
chassis_info.total_supplied_power = 300.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = True
ret = chassis_info.update_master_status()
assert ret == True
assert chassis_info.master_status_good == False
# Test bad values while in good state
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == False
# Test set_status_master_led not implemented
with mock.patch.object(MockPsu, "set_status_master_led",
mock.Mock(side_effect=NotImplementedError)):
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = False
chassis_info.log_warning = mock.MagicMock()
ret = chassis_info.update_master_status()
assert ret == True
assert chassis_info.master_status_good == True
chassis_info.log_warning.assert_called_with(
"set_status_master_led() not implemented")
def test_power_budget(self):
chassis = MockChassis()
psu1 = MockPsu("PSU 1", 0, True, True)
psu1_power = 510.0
psu1.set_maximum_supplied_power(psu1_power)
chassis._psu_list.append(psu1)
fan_drawer1 = MockFanDrawer("FanDrawer 1", 0, True, True)
fan_drawer1_power = 510.0
fan_drawer1.set_maximum_consumed_power(fan_drawer1_power)
chassis._fan_drawer_list.append(fan_drawer1)
module1 = MockModule("Module 1", 0, True, True)
module1_power = 700.0
module1.set_maximum_consumed_power(module1_power)
chassis._module_list.append(module1)
state_db = daemon_base.db_connect("STATE_DB")
chassis_tbl = mock_swsscommon.Table(state_db, CHASSIS_INFO_TABLE)
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
chassis_info.first_run = False
# Check case where supplied_power < consumed_power
chassis_info.run_power_budget(chassis_tbl)
chassis_info.update_master_status()
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert float(fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD]) < float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
assert chassis_info.master_status_good == False
# We cannot call get_status_master_led() on our mocked PSUs, because
# they are not instantiated from the same Psu class loaded in psud,
# so we must call it on the class there.
assert psud.Psu.get_status_master_led() == MockPsu.STATUS_LED_COLOR_RED
# Add a PSU
psu2 = MockPsu("PSU 2", 1, True, True)
psu2_power = 800.0
psu2.set_maximum_supplied_power(psu2_power)
chassis._psu_list.append(psu2)
# Check case where supplied_power > consumed_power
chassis_info.run_power_budget(chassis_tbl)
chassis_info.update_master_status()
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert float(fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD]) > float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
assert chassis_info.master_status_good == True
# We cannot call get_status_master_led() on our mocked PSUs, because
# they are not instantiated from the same Psu class loaded in psud,
# so we must call it on the class there.
assert psud.Psu.get_status_master_led(
) == MockPsu.STATUS_LED_COLOR_GREEN
def test_power_budget(self):
chassis = MockChassis()
psu = MockPsu(True, True, "PSU 1", 0)
psu1_power = 510.0
psu.set_maximum_supplied_power(psu1_power)
chassis.psu_list.append(psu)
fan_drawer1 = MockFanDrawer(True, True, "FanDrawer 1")
fan_drawer1_power = 510.0
fan_drawer1.set_maximum_consumed_power(fan_drawer1_power)
chassis.fan_drawer_list.append(fan_drawer1)
module1 = MockFanDrawer(True, True, "Module 1")
module1_power = 700.0
module1.set_maximum_consumed_power(module1_power)
chassis.module_list.append(module1)
state_db = daemon_base.db_connect("STATE_DB")
chassis_tbl = mock_swsscommon.Table(state_db, CHASSIS_INFO_TABLE)
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
# Check if supplied_power < consumed_power
chassis_info.run_power_budget(chassis_tbl)
chassis_info.update_master_status()
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert float(fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD]) < float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
assert chassis_info.master_status_good == False
assert MockPsu.get_status_master_led() == MockPsu.STATUS_LED_COLOR_RED
# Add a PSU
psu = MockPsu(True, True, "PSU 2", 1)
psu2_power = 800.0
psu.set_maximum_supplied_power(psu2_power)
chassis.psu_list.append(psu)
# Check if supplied_power > consumed_power
chassis_info.run_power_budget(chassis_tbl)
chassis_info.update_master_status()
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert float(fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD]) > float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
assert chassis_info.master_status_good == True
assert MockPsu.get_status_master_led(
) == MockPsu.STATUS_LED_COLOR_GREEN
def test_first_run(self):
chassis = MockChassis()
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
chassis_info.total_supplied_power = 0.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == True
assert psud.Psu.get_status_master_led() == MockPsu.STATUS_LED_COLOR_RED
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = True
ret = chassis_info.update_master_status()
assert ret == True
assert psud.Psu.get_status_master_led(
) == MockPsu.STATUS_LED_COLOR_GREEN
def test_update_master_status(self):
chassis = MockChassis()
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
# Test good values while in bad state
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == True
assert chassis_info.master_status_good == True
# Test good values while in good state
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test unknown total_supplied_power (0.0)
chassis_info.total_supplied_power = 0.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test unknown total_consumed_power (0.0)
chassis_info.total_supplied_power = 510.0
chassis_info.total_consumed_power = 0.0
chassis_info.master_status_good = False
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == True
# Test bad values while in good state
chassis_info.total_supplied_power = 300.0
chassis_info.total_consumed_power = 350.0
chassis_info.master_status_good = True
ret = chassis_info.update_master_status()
assert ret == True
assert chassis_info.master_status_good == False
# Test bad values while in good state
ret = chassis_info.update_master_status()
assert ret == False
assert chassis_info.master_status_good == False
def test_supplied_power(self):
chassis = MockChassis()
psu1 = MockPsu("PSU 1", 0, True, True)
psu1_power = 510.0
psu1.set_maximum_supplied_power(psu1_power)
chassis._psu_list.append(psu1)
psu2 = MockPsu("PSU 2", 1, True, True)
psu2_power = 800.0
psu2.set_maximum_supplied_power(psu2_power)
chassis._psu_list.append(psu2)
psu3 = MockPsu("PSU 3", 2, True, True)
psu3_power = 350.0
psu3.set_maximum_supplied_power(psu3_power)
chassis._psu_list.append(psu3)
total_power = psu1_power + psu2_power + psu3_power
state_db = daemon_base.db_connect("STATE_DB")
chassis_tbl = mock_swsscommon.Table(state_db, CHASSIS_INFO_TABLE)
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
# Check if supplied power is recorded in DB
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD])
# Check if psu1 is not present
psu1.set_presence(False)
total_power = psu2_power + psu3_power
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD])
# Check if psu2 status is NOT_OK
psu2.set_status(False)
total_power = psu3_power
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_SUPPLIED_FIELD])
def test_consumed_power(self):
chassis = MockChassis()
fan_drawer1 = MockFanDrawer("FanDrawer 1", 0, True, True)
fan_drawer1_power = 510.0
fan_drawer1.set_maximum_consumed_power(fan_drawer1_power)
chassis._fan_drawer_list.append(fan_drawer1)
module1 = MockModule("Module 1", 0, True, True)
module1_power = 700.0
module1.set_maximum_consumed_power(module1_power)
chassis._module_list.append(module1)
total_power = fan_drawer1_power + module1_power
state_db = daemon_base.db_connect("STATE_DB")
chassis_tbl = mock_swsscommon.Table(state_db, CHASSIS_INFO_TABLE)
chassis_info = psud.PsuChassisInfo(SYSLOG_IDENTIFIER, chassis)
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
# Check if supplied power is recorded in DB
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
# Check if fan_drawer1 present
fan_drawer1.set_presence(False)
total_power = module1_power
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])
# Check if module1 present
fan_drawer1.set_presence(True)
module1.set_presence(False)
total_power = fan_drawer1_power
chassis_info.run_power_budget(chassis_tbl)
fvs = chassis_tbl.get(CHASSIS_INFO_POWER_KEY_TEMPLATE.format(1))
assert total_power == float(
fvs[CHASSIS_INFO_TOTAL_POWER_CONSUMED_FIELD])