From 954b823fb8e785ce5aca7a9f25ee3f06d951fb15 Mon Sep 17 00:00:00 2001
From: Craig Barratt <19445341+craigbarratt@users.noreply.github.com>
Date: Thu, 4 Mar 2021 21:03:43 +1100
Subject: [PATCH] try again to fix testing error with new service description

---
 tests/test_init.py | 29 +++++++++++------------------
 1 file changed, 11 insertions(+), 18 deletions(-)

diff --git a/tests/test_init.py b/tests/test_init.py
index 768bf9c..721f361 100644
--- a/tests/test_init.py
+++ b/tests/test_init.py
@@ -200,28 +200,21 @@ def func_yaml_doc_string(param2=None, param3=None):
     with patch(
         "homeassistant.loader.async_get_custom_components", return_value={"pyscript": integration},
     ):
-        descriptions = await async_get_all_descriptions(hass)
+        descriptions = (await async_get_all_descriptions(hass))[DOMAIN]
 
-    descriptions[DOMAIN].pop("name", None)
-    assert descriptions[DOMAIN]["func_no_doc_string"] == {
-        "description": "pyscript function func_no_doc_string()",
-        "fields": {"param1": {"description": "argument param1"}},
-    }
+    assert descriptions["func_no_doc_string"]["description"] == "pyscript function func_no_doc_string()"
+    assert descriptions["func_no_doc_string"]["fields"] == {"param1": {"description": "argument param1"}}
 
-    assert descriptions[DOMAIN]["func_simple_doc_string"] == {
-        "description": "This is func2_simple_doc_string.",
-        "fields": {
-            "param2": {"description": "argument param2"},
-            "param3": {"description": "argument param3"},
-        },
+    assert descriptions["func_simple_doc_string"]["description"] == "This is func2_simple_doc_string."
+    assert descriptions["func_simple_doc_string"]["fields"] == {
+        "param2": {"description": "argument param2"},
+        "param3": {"description": "argument param3"},
     }
 
-    assert descriptions[DOMAIN]["func_yaml_doc_string"] == {
-        "description": "This is func_yaml_doc_string.",
-        "fields": {
-            "param1": {"description": "first argument", "example": "12"},
-            "param2": {"description": "second argument", "example": "34"},
-        },
+    assert descriptions["func_yaml_doc_string"]["description"] == "This is func_yaml_doc_string."
+    assert descriptions["func_yaml_doc_string"]["fields"] == {
+        "param1": {"description": "first argument", "example": "12"},
+        "param2": {"description": "second argument", "example": "34"},
     }