def test_mode2_login(self, mode2_scenario):
scenario_param = mode2_scenario["param"]
if (not any(
[scenario_param.get("dps_name"),
scenario_param.get("hub_name")]) and not scenario_param["login"]
and mode2_scenario["cmd"].name.startswith("iot")):
with pytest.raises(CLIError):
mode2_iot_login_handler(cmd=mode2_scenario["cmd"],
namespace=mode2_scenario["namespace"])
else:
mode2_iot_login_handler(cmd=mode2_scenario["cmd"],
namespace=mode2_scenario["namespace"])
def test_mode2_login(self, mocker, hub_name, dps_name, login):
mock_cmd = mocker.MagicMock(name='mock cmd')
mock_cmd.name = 'iot '
mock_ns = mocker.MagicMock(name='mock ns')
if login != '[]':
mock_ns.login = login
if hub_name != '[]':
mock_ns.hub_name = hub_name
if dps_name != '[]':
mock_ns.dps_name = dps_name
mode2_iot_login_handler(mock_cmd, mock_ns)
