Update some tests

Ingenjorsarbete-For-Klimatet · Mar 30, 2024 · 81dc1e0 · 81dc1e0
1 parent 59ae528
commit 81dc1e0
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 14 deletions.
diff --git a/src/smhi/metfcts.py b/src/smhi/metfcts.py
@@ -4,15 +4,15 @@
 
 from smhi.constants import METFCTS_PARAMETER_DESCRIPTIONS, METFCTS_URL
 from smhi.mesan import Mesan
-from smhi.models.metfcts import (
+from smhi.models.metfcts_model import (
     MetfctsApprovedTime,
     MetfctsMultiPointData,
     MetfctsParameters,
     MetfctsPointData,
     MetfctsPolygon,
     MetfctsValidTime,
 )
-from smhi.models.variable import (
+from smhi.models.variable_model import (
     ApprovedTime,
     MultiPointData,
     Parameters,

diff --git a/src/smhi/models/variable_model.py b/src/smhi/models/variable_model.py
@@ -2,15 +2,15 @@
 
 from typing import TypeVar
 
-from smhi.models.mesan import (
+from smhi.models.mesan_model import (
     MesanApprovedTime,
     MesanMultiPointData,
     MesanParameters,
     MesanPointData,
     MesanPolygon,
     MesanValidTime,
 )
-from smhi.models.metfcts import (
+from smhi.models.metfcts_model import (
     MetfctsApprovedTime,
     MetfctsMultiPointData,
     MetfctsParameters,

diff --git a/tests/unit/test_unit_smhi.py → tests/unit/_test_unit_smhi.py b/tests/unit/test_unit_smhi.py → tests/unit/_test_unit_smhi.py
diff --git a/tests/unit/test_unit_metobs.py b/tests/unit/test_unit_metobs.py
@@ -282,7 +282,7 @@ def test_unit_basemetobs_init(self):
         assert base.summary is None
         assert base.link is None
 
-    @patch("smhi.metobs.requests.get", return_value=MockResponse(200, None, None))
+    @patch("smhi.utils.requests.get", return_value=MockResponse(200, None, None))
     def test_unit_basemetobs_get_and_parse_request(self, mock_requests_get):
         """Unit test for BaseMetobs _get_and_parse_request method.
 
@@ -346,7 +346,7 @@ class TestUnitVersions:
     """Unit tests for Versionss class."""
 
     @pytest.mark.parametrize("data_type", [("json"), ("yaml"), ("xml"), (None)])
-    @patch("smhi.metobs.requests.get")
+    @patch("smhi.utils.requests.get")
     def test_unit_versions_init(self, mock_requests_get, data_type, setup_versions):
         """Unit test for Parameters init method.
 
@@ -391,7 +391,7 @@ class TestUnitParameters:
         ],
     )
     @patch("smhi.metobs.Versions")
-    @patch("smhi.metobs.requests.get")
+    @patch("smhi.utils.requests.get")
     def test_unit_parameters_init(
         self,
         mock_requests_get,
@@ -451,7 +451,7 @@ class TestUnitStations:
             ("1", "Lufttemperatur", None),
         ],
     )
-    @patch("smhi.metobs.requests.get")
+    @patch("smhi.utils.requests.get")
     def test_unit_stations_init(
         self,
         mock_requests_get,
@@ -516,7 +516,7 @@ class TestUnitPeriods:
             (1, "Akalla", "all", "json"),
         ],
     )
-    @patch("smhi.metobs.requests.get")
+    @patch("smhi.utils.requests.get")
     def test_unit_periods_init(
         self,
         mock_requests_get,
@@ -595,7 +595,7 @@ class TestUnitData:
             ("corrected-archive", "json"),
         ],
     )
-    @patch("smhi.metobs.requests.get")
+    @patch("smhi.utils.requests.get")
     def test_unit_data_init(
         self,
         mock_requests_get,

diff --git a/tests/unit/test_unit_strang.py b/tests/unit/test_unit_strang.py
@@ -13,7 +13,7 @@
     STRANG_PARAMETERS,
     STRANG_POINT_URL,
 )
-from smhi.models.strang import StrangParameter
+from smhi.models.strang_model import StrangParameter
 from smhi.strang import Strang
 
 INPUT_DAILY_2020_01_01_2020_01_02 = [
@@ -410,11 +410,11 @@ def test_unit_strang_build_time_multipoint_url(self, time_interval, expected_url
     @patch("smhi.strang.Strang._parse_point_data")
     @patch("smhi.strang.logging.info")
     @patch(
-        "smhi.strang.requests.get",
+        "smhi.utils.requests.get",
         return_value=type(
             "MyClass",
             (object,),
-            {"status": True, "headers": "header", "content": "content"},
+            {"status_code": 200, "headers": "header", "content": "content"},
         )(),
     )
     @patch(
@@ -447,7 +447,7 @@ def test_unit_strang_get_and_load_data(
         mock_requests_get.return_value.status_code = status_expected
 
         data, headers, status = client._get_and_load_data(client.url, "parameter")
-        mock_requests_get.assert_called_once_with(client.url)
+        mock_requests_get.assert_called_once_with(client.url, timeout=200)
 
         if status == 200:
             mock_json_loads.assert_called_once_with(