Fix flakey test_update_record test (round 2) (#528)

bdraco · web-flow · commit 14542bd2bd32 · 2021-06-12T10:18:10.000-10:00
diff --git a/zeroconf/test.py b/zeroconf/test.py
@@ -2156,6 +2156,8 @@ def mock_incoming_msg(
                 zeroconf,
                 mock_incoming_msg(r.ServiceStateChange.Added, service_types[1], service_names[1], 120),
             )
+            zeroconf.wait(100)
+
             called_with_refresh_time_check = False
 
             def _mock_get_expiration_time(self, percent):
@@ -2171,7 +2173,6 @@ def _mock_get_expiration_time(self, percent):
                     zeroconf,
                     mock_incoming_msg(r.ServiceStateChange.Added, service_types[0], service_names[0], 120),
                 )
-                zeroconf.wait(100)
                 # Add the last record after updating the first one
                 # to ensure the service_add_event only gets set
                 # after the update
