|
@@ -66,3 +66,69 @@ def test__main(
|
|
mqtt_username=expected_username,
|
|
mqtt_username=expected_username,
|
|
mqtt_password=expected_password,
|
|
mqtt_password=expected_password,
|
|
)
|
|
)
|
|
|
|
+
|
|
|
|
+
|
|
|
|
+@pytest.mark.parametrize(
|
|
|
|
+ ("password_file_content", "expected_password",),
|
|
|
|
+ [
|
|
|
|
+ ("secret", "secret"),
|
|
|
|
+ ("secret space", "secret space"),
|
|
|
|
+ ("secret ", "secret "),
|
|
|
|
+ (" secret ", " secret "),
|
|
|
|
+ ("secret\n", "secret"),
|
|
|
|
+ ("secret\n\n", "secret\n"),
|
|
|
|
+ ("secret\r\n", "secret"),
|
|
|
|
+ ("secret\n\r\n", "secret\n"),
|
|
|
|
+ ("你好\n", "你好"),
|
|
|
|
+ ],
|
|
|
|
+)
|
|
|
|
+def test__main_password_file(tmpdir, password_file_content, expected_password):
|
|
|
|
+ mqtt_password_path = tmpdir.join("mqtt-password")
|
|
|
|
+ with mqtt_password_path.open("w") as mqtt_password_file:
|
|
|
|
+ mqtt_password_file.write(password_file_content)
|
|
|
|
+ with unittest.mock.patch("switchbot_mqtt._run") as run_mock, unittest.mock.patch(
|
|
|
|
+ "sys.argv",
|
|
|
|
+ [
|
|
|
|
+ "",
|
|
|
|
+ "--mqtt-host",
|
|
|
|
+ "localhost",
|
|
|
|
+ "--mqtt-username",
|
|
|
|
+ "me",
|
|
|
|
+ "--mqtt-password-file",
|
|
|
|
+ str(mqtt_password_path),
|
|
|
|
+ ],
|
|
|
|
+ ):
|
|
|
|
+
|
|
|
|
+ switchbot_mqtt._main()
|
|
|
|
+ run_mock.assert_called_once_with(
|
|
|
|
+ mqtt_host="localhost",
|
|
|
|
+ mqtt_port=1883,
|
|
|
|
+ mqtt_username="me",
|
|
|
|
+ mqtt_password=expected_password,
|
|
|
|
+ )
|
|
|
|
+
|
|
|
|
+
|
|
|
|
+def test__main_password_file_collision(capsys):
|
|
|
|
+ with unittest.mock.patch(
|
|
|
|
+ "sys.argv",
|
|
|
|
+ [
|
|
|
|
+ "",
|
|
|
|
+ "--mqtt-host",
|
|
|
|
+ "localhost",
|
|
|
|
+ "--mqtt-username",
|
|
|
|
+ "me",
|
|
|
|
+ "--mqtt-password",
|
|
|
|
+ "secret",
|
|
|
|
+ "--mqtt-password-file",
|
|
|
|
+ "/var/lib/secrets/mqtt/password",
|
|
|
|
+ ],
|
|
|
|
+ ):
|
|
|
|
+ with pytest.raises(SystemExit):
|
|
|
|
+
|
|
|
|
+ switchbot_mqtt._main()
|
|
|
|
+ out, err = capsys.readouterr()
|
|
|
|
+ assert not out
|
|
|
|
+ assert (
|
|
|
|
+ "argument --mqtt-password-file: not allowed with argument --mqtt-password\n"
|
|
|
|
+ in err
|
|
|
|
+ )
|