Fix nabweatherd test

nabaztag2018 · Apr 25, 2021 · bda4645 · bda4645
1 parent 2886354
commit bda4645
Showing 1 changed file with 38 additions and 7 deletions.
diff --git a/nabweatherd/tests/nabweatherd_test.py b/nabweatherd/tests/nabweatherd_test.py
@@ -24,14 +24,22 @@ def test_aliases(self):
 @pytest.mark.django_db(transaction=True)
 class TestNabWeatherdDB(unittest.TestCase):
     RENNES_LOCATION_JSON = '{"insee":"35238","name":"Rennes","lat":48.11417,"lon":-1.68083,"country":"FR","admin":"Bretagne","admin2":"35","postCode":"35000"}'
+
     def tearDown(self):
         close_old_async_connections()
 
     def test_fetch_info_data(self):
         service = NabWeatherd()
-        
+
         data = async_to_sync(service.fetch_info_data)(
-            (TestNabWeatherdDB.RENNES_LOCATION_JSON, NabWeatherd.UNIT_CELSIUS, "weather_and_rain")
+            (
+                TestNabWeatherdDB.RENNES_LOCATION_JSON,
+                NabWeatherd.UNIT_CELSIUS,
+                "weather_and_rain",
+                3,
+                None,
+                False,
+            )
         )
         self.assertTrue("current_weather_class" in data)
         self.assertTrue("today_forecast_weather_class" in data)
@@ -41,12 +49,18 @@ def test_fetch_info_data(self):
         self.assertTrue("next_rain" in data)
         self.assertTrue("weather_animation_type" in data)
 
-
     def test_perform_both(self):
         service = NabWeatherd()
         writer = MockWriter()
         service.writer = writer
-        config_t = (TestNabWeatherdDB.RENNES_LOCATION_JSON, NabWeatherd.UNIT_CELSIUS, "weather_and_rain")
+        config_t = (
+            TestNabWeatherdDB.RENNES_LOCATION_JSON,
+            NabWeatherd.UNIT_CELSIUS,
+            "weather_and_rain",
+            3,
+            None,
+            False,
+        )
         expiration = datetime.datetime(2019, 4, 22, 0, 0, 0)
         async_to_sync(service.perform)(expiration, "today", config_t)
         self.assertEqual(len(writer.written), 3)
@@ -69,7 +83,14 @@ def test_perform_rain(self):
         service = NabWeatherd()
         writer = MockWriter()
         service.writer = writer
-        config_t = (TestNabWeatherdDB.RENNES_LOCATION_JSON, NabWeatherd.UNIT_CELSIUS, "rain_only")
+        config_t = (
+            TestNabWeatherdDB.RENNES_LOCATION_JSON,
+            NabWeatherd.UNIT_CELSIUS,
+            "rain_only",
+            3,
+            None,
+            False,
+        )
         expiration = datetime.datetime(2019, 4, 22, 0, 0, 0)
         async_to_sync(service.perform)(expiration, "today", config_t)
         self.assertEqual(len(writer.written), 3)
@@ -92,7 +113,14 @@ def test_perform(self):
         service = NabWeatherd()
         writer = MockWriter()
         service.writer = writer
-        config_t = (TestNabWeatherdDB.RENNES_LOCATION_JSON, NabWeatherd.UNIT_CELSIUS, "weather_only")
+        config_t = (
+            TestNabWeatherdDB.RENNES_LOCATION_JSON,
+            NabWeatherd.UNIT_CELSIUS,
+            "weather_only",
+            3,
+            None,
+            False,
+        )
         expiration = datetime.datetime(2019, 4, 22, 0, 0, 0)
         async_to_sync(service.perform)(expiration, "today", config_t)
         self.assertEqual(len(writer.written), 3)
@@ -120,7 +148,10 @@ def test_asr(self):
         service = NabWeatherd()
         writer = MockWriter()
         service.writer = writer
-        packet = {"type": "asr_event", "nlu": {"intent": "nabweatherd/forecast"}}
+        packet = {
+            "type": "asr_event",
+            "nlu": {"intent": "nabweatherd/forecast"},
+        }
         async_to_sync(service.process_nabd_packet)(packet)
         self.assertEqual(len(writer.written), 3)
         packet = writer.written[0]