From d478517c51c8428c5606bfa9dfd6f734aff5df13 Mon Sep 17 00:00:00 2001 From: Sean Dague Date: Mon, 22 Jan 2018 09:21:56 -0500 Subject: [PATCH] Fix races on recorder test (#11857) --- tests/components/recorder/test_purge.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/components/recorder/test_purge.py b/tests/components/recorder/test_purge.py index bbb87fb5016..c429ee2fbbb 100644 --- a/tests/components/recorder/test_purge.py +++ b/tests/components/recorder/test_purge.py @@ -1,7 +1,6 @@ """Test data purging.""" import json from datetime import datetime, timedelta -from time import sleep import unittest from homeassistant.components import recorder @@ -162,7 +161,7 @@ class TestRecorderPurge(unittest.TestCase): self.hass.async_block_till_done() # Small wait for recorder thread - sleep(0.1) + self.hass.data[DATA_INSTANCE].block_till_done() # we should still have everything from before self.assertEqual(states.count(), 6) @@ -174,7 +173,7 @@ class TestRecorderPurge(unittest.TestCase): self.hass.async_block_till_done() # Small wait for recorder thread - sleep(0.1) + self.hass.data[DATA_INSTANCE].block_till_done() # we should only have 3 states left after purging self.assertEqual(states.count(), 3)