allow "key" to be passed as a parameter
This was initially designed and implemented in #4 Co-authored-by: Sven Seeberg <mail@sven-seeberg.de>
This commit is contained in:
parent
292d77274d
commit
fa8f9b4a51
4 changed files with 24 additions and 6 deletions
|
@ -6,7 +6,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
|
||||||
|
|
||||||
## [Unreleased]
|
## [Unreleased]
|
||||||
|
|
||||||
- update "text" key to "body".
|
- allow "key" to be passed as a parameter
|
||||||
|
- rename "text" to "body".
|
||||||
|
|
||||||
## [3.1.0] - 2021-07-18
|
## [3.1.0] - 2021-07-18
|
||||||
|
|
||||||
|
|
|
@ -38,10 +38,14 @@ async def handler(request):
|
||||||
except json.decoder.JSONDecodeError:
|
except json.decoder.JSONDecodeError:
|
||||||
return create_json_response(HTTPStatus.BAD_REQUEST, "Invalid JSON")
|
return create_json_response(HTTPStatus.BAD_REQUEST, "Invalid JSON")
|
||||||
|
|
||||||
# legacy naming:
|
# legacy naming
|
||||||
if "text" in data and "body" not in data:
|
if "text" in data and "body" not in data:
|
||||||
data["body"] = data["text"]
|
data["body"] = data["text"]
|
||||||
|
|
||||||
|
# allow key to be passed as a parameter
|
||||||
|
if "key" in request.rel_url.query and "key" not in data:
|
||||||
|
data["key"] = request.rel_url.query["key"]
|
||||||
|
|
||||||
if not all(key in data for key in ["body", "key"]):
|
if not all(key in data for key in ["body", "key"]):
|
||||||
return create_json_response(
|
return create_json_response(
|
||||||
HTTPStatus.BAD_REQUEST, "Missing body and/or API key property"
|
HTTPStatus.BAD_REQUEST, "Missing body and/or API key property"
|
||||||
|
|
|
@ -25,12 +25,17 @@ parser.add_argument(
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def bot_req(req=None, key=None, room_id=None):
|
def bot_req(req=None, key=None, room_id=None, key_as_param=False):
|
||||||
"""Bot requests boilerplate."""
|
"""Bot requests boilerplate."""
|
||||||
|
params = {}
|
||||||
|
|
||||||
if key is not None:
|
if key is not None:
|
||||||
|
if key_as_param:
|
||||||
|
params["key"] = key
|
||||||
|
else:
|
||||||
req["key"] = key
|
req["key"] = key
|
||||||
url = BOT_URL if room_id is None else f"{BOT_URL}/{room_id}"
|
url = BOT_URL if room_id is None else f"{BOT_URL}/{room_id}"
|
||||||
return httpx.post(url, json=req).json()
|
return httpx.post(url, params=params, json=req).json()
|
||||||
|
|
||||||
|
|
||||||
def wait_available(url: str, key: str, timeout: int = 10) -> bool:
|
def wait_available(url: str, key: str, timeout: int = 10) -> bool:
|
||||||
|
|
|
@ -18,13 +18,21 @@ class BotTest(unittest.IsolatedAsyncioTestCase):
|
||||||
{"status": 400, "ret": "Missing body and/or API key property"},
|
{"status": 400, "ret": "Missing body and/or API key property"},
|
||||||
)
|
)
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
bot_req({"body": 3, "key": None}), {"status": 401, "ret": "Invalid API key"}
|
bot_req({"body": 3}, "wrong_key"), {"status": 401, "ret": "Invalid API key"}
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
bot_req({"body": 3}, "wrong_key", key_as_param=True),
|
||||||
|
{"status": 401, "ret": "Invalid API key"},
|
||||||
)
|
)
|
||||||
# TODO: if the client from matrix_webhook has olm support, this won't be a 403 from synapse,
|
# TODO: if the client from matrix_webhook has olm support, this won't be a 403 from synapse,
|
||||||
# but a LocalProtocolError from matrix_webhook
|
# but a LocalProtocolError from matrix_webhook
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
bot_req({"body": 3}, KEY), {"status": 403, "ret": "Unknown room"}
|
bot_req({"body": 3}, KEY), {"status": 403, "ret": "Unknown room"}
|
||||||
)
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
bot_req({"body": 3}, KEY, key_as_param=True),
|
||||||
|
{"status": 403, "ret": "Unknown room"},
|
||||||
|
)
|
||||||
|
|
||||||
async def test_message(self):
|
async def test_message(self):
|
||||||
"""Send a markdown message with the old format, and check the result."""
|
"""Send a markdown message with the old format, and check the result."""
|
||||||
|
|
Loading…
Reference in a new issue