Use send_json_auto_id in some collection tests (#119570)

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
This commit is contained in:
Erik Montnemery 2024-06-13 16:04:32 +02:00 committed by GitHub
parent 8871090463
commit 9f322b20d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -450,9 +450,8 @@ async def test_storage_collection_websocket(
client = await hass_ws_client(hass)
# Create invalid
await client.send_json(
await client.send_json_auto_id(
{
"id": 1,
"type": "test_item/collection/create",
"name": 1,
# Forgot to add immutable_string
@ -464,9 +463,8 @@ async def test_storage_collection_websocket(
assert len(changes) == 0
# Create
await client.send_json(
await client.send_json_auto_id(
{
"id": 2,
"type": "test_item/collection/create",
"name": "Initial Name",
"immutable_string": "no-changes",
@ -483,7 +481,7 @@ async def test_storage_collection_websocket(
assert changes[0] == (collection.CHANGE_ADDED, "initial_name", response["result"])
# List
await client.send_json({"id": 3, "type": "test_item/collection/list"})
await client.send_json_auto_id({"type": "test_item/collection/list"})
response = await client.receive_json()
assert response["success"]
assert response["result"] == [
@ -496,9 +494,8 @@ async def test_storage_collection_websocket(
assert len(changes) == 1
# Update invalid data
await client.send_json(
await client.send_json_auto_id(
{
"id": 4,
"type": "test_item/collection/update",
"test_item_id": "initial_name",
"immutable_string": "no-changes",
@ -510,9 +507,8 @@ async def test_storage_collection_websocket(
assert len(changes) == 1
# Update invalid item
await client.send_json(
await client.send_json_auto_id(
{
"id": 5,
"type": "test_item/collection/update",
"test_item_id": "non-existing",
"name": "Updated name",
@ -524,9 +520,8 @@ async def test_storage_collection_websocket(
assert len(changes) == 1
# Update
await client.send_json(
await client.send_json_auto_id(
{
"id": 6,
"type": "test_item/collection/update",
"test_item_id": "initial_name",
"name": "Updated name",
@ -543,8 +538,8 @@ async def test_storage_collection_websocket(
assert changes[1] == (collection.CHANGE_UPDATED, "initial_name", response["result"])
# Delete invalid ID
await client.send_json(
{"id": 7, "type": "test_item/collection/update", "test_item_id": "non-existing"}
await client.send_json_auto_id(
{"type": "test_item/collection/update", "test_item_id": "non-existing"}
)
response = await client.receive_json()
assert not response["success"]
@ -552,8 +547,8 @@ async def test_storage_collection_websocket(
assert len(changes) == 2
# Delete
await client.send_json(
{"id": 8, "type": "test_item/collection/delete", "test_item_id": "initial_name"}
await client.send_json_auto_id(
{"type": "test_item/collection/delete", "test_item_id": "initial_name"}
)
response = await client.receive_json()
assert response["success"]