Add and restore context in recorder (#15859)

2018-08-10 18:09:01 +02:00 · 2018-08-10 18:09:01 +02:00 · 9512bb9587
commit 9512bb9587
parent da916d7b27
7 changed files with 73 additions and 14 deletions
--- a/tests/test_core.py
+++ b/tests/test_core.py
@ -246,8 +246,9 @@ class TestEvent(unittest.TestCase):
        """Test events."""
        now = dt_util.utcnow()
        data = {'some': 'attr'}
+        context = ha.Context()
        event1, event2 = [
-            ha.Event('some_type', data, time_fired=now)
+            ha.Event('some_type', data, time_fired=now, context=context)
            for _ in range(2)
        ]