From 48ac70de95cffdda0c4197ce54437a17632dbc3c Mon Sep 17 00:00:00 2001
From: Mikhail Epifanov <mikhail.epifanov@gmail.com>
Date: Sat, 13 Jan 2024 11:56:51 +0100
Subject: [PATCH] make the tests check for any error message

---
 .../tests/edits/test_edits_home_assistant_config.py  |  9 +++++++--
 cookbook/tests/edits/test_edits_storage.py           | 12 +++++++++---
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/cookbook/tests/edits/test_edits_home_assistant_config.py b/cookbook/tests/edits/test_edits_home_assistant_config.py
index 7f2e19a9..0df507de 100644
--- a/cookbook/tests/edits/test_edits_home_assistant_config.py
+++ b/cookbook/tests/edits/test_edits_home_assistant_config.py
@@ -1,5 +1,7 @@
 import pytest
 from django.contrib import auth
+from django.contrib import messages
+from django.contrib.messages import get_messages
 from django.urls import reverse
 
 from cookbook.models import HomeAssistantConfig
@@ -12,7 +14,7 @@ def home_assistant_config_obj(a1_s1, space_1):
     return HomeAssistantConfig.objects.create(
         name='HomeAssistant 1',
         token='token',
-        url='url',
+        url='http://localhost:8123/api',
         todo_entity='todo.shopping_list',
         enabled=True,
         created_by=auth.get_user(a1_s1),
@@ -28,12 +30,15 @@ def test_edit_home_assistant_config(home_assistant_config_obj: HomeAssistantConf
         {
             'name': home_assistant_config_obj.name,
             'url': home_assistant_config_obj.url,
-            'todo_entity': home_assistant_config_obj.todo_entity,
             'token': new_token,
+            'todo_entity': home_assistant_config_obj.todo_entity,
             'enabled': home_assistant_config_obj.enabled,
         }
     )
     assert r.status_code == 200
+    r_messages = [m for m in get_messages(r.wsgi_request)]
+    assert not any(m.level > messages.SUCCESS for m in r_messages)
+
     home_assistant_config_obj.refresh_from_db()
     assert home_assistant_config_obj.token == new_token
 
diff --git a/cookbook/tests/edits/test_edits_storage.py b/cookbook/tests/edits/test_edits_storage.py
index 125445e1..9c4e08e1 100644
--- a/cookbook/tests/edits/test_edits_storage.py
+++ b/cookbook/tests/edits/test_edits_storage.py
@@ -1,7 +1,10 @@
-from cookbook.models import Storage
-from django.contrib import auth
-from django.urls import reverse
 import pytest
+from django.contrib import auth
+from django.contrib import messages
+from django.contrib.messages import get_messages
+from django.urls import reverse
+
+from cookbook.models import Storage
 
 
 @pytest.fixture
@@ -29,6 +32,9 @@ def test_edit_storage(storage_obj, a1_s1, a1_s2):
     )
     storage_obj.refresh_from_db()
     assert r.status_code == 200
+    r_messages = [m for m in get_messages(r.wsgi_request)]
+    assert not any(m.level > messages.SUCCESS for m in r_messages)
+
     assert storage_obj.password == '1234_pw'
     assert storage_obj.token == '1234_token'