Tests

Author: bogdandm

rest_client_gen/dynamic_typing/string_datetime.py

@@ -49,8 +49,8 @@ def is_date(s: str) -> Optional[date]:
 
 
 _check_values_time = (
-    datetime(2018, 10, 11, 0, 4, 5, 678, tzinfo=None),
-    datetime(2018, 12, 30, 0, 4, 5, 678, tzinfo=None)
+    datetime(2018, 10, 11),
+    datetime(2018, 12, 30)
 )
 
 

test/test_dynamic_typing/test_string_datetime.py

@@ -0,0 +1,94 @@
+import datetime
+
+import pytest
+
+from rest_client_gen.dynamic_typing import (
+    FloatString, IntString, IsoDateString, IsoDatetimeString, IsoTimeString, register_datetime_classes
+)
+from rest_client_gen.generator import MetadataGenerator
+
+register_datetime_classes()
+
+test_detect_type_data = [
+    # Check that string datetime doesn't break default string types
+    pytest.param(
+        "1",
+        IntString,
+        id="default_check_int"
+    ),
+    pytest.param(
+        "1.5",
+        FloatString,
+        id="default_check_float"
+    ),
+    pytest.param(
+        "2018-12-31",
+        IsoDateString,
+        id="date"
+    ),
+    pytest.param(
+        "12:58",
+        IsoTimeString,
+        id="time"
+    ),
+    pytest.param(
+        "2018-12-31T12:58:12Z",
+        IsoDatetimeString,
+        id="datetime"
+    )
+]
+
+
+@pytest.mark.parametrize("value,expected", test_detect_type_data)
+def test_detect_type(models_generator: MetadataGenerator, value, expected):
+    result = models_generator._detect_type(value)
+    assert result == expected
+
+
+test_parse_data = [
+    pytest.param(
+        "2018-12-31",
+        IsoDateString(2018, 12, 31),
+        id="date"
+    ),
+    pytest.param(
+        "12:13",
+        IsoTimeString(12, 13),
+        id="time"
+    ),
+    pytest.param(
+        "04:15:34",
+        IsoTimeString(4, 15, 34),
+        id="time_seconds"
+    ),
+    pytest.param(
+        "04:15:34.034",
+        IsoTimeString(4, 15, 34, 34000),
+        id="time_ms"
+    ),
+    pytest.param(
+        "2018-12-04T04:15:34.034000+00:00",
+        IsoDatetimeString(2018, 12, 4, 4, 15, 34, 34000, tzinfo=datetime.timezone.utc),
+        id="datetime_full"
+    ),
+    pytest.param(
+        "2018-12-04T04:15",
+        IsoDatetimeString(2018, 12, 4, 4, 15),
+        id="datetime_partial"
+    )
+]
+
+
+@pytest.mark.parametrize("value,expected", test_parse_data)
+def test_parse(models_generator: MetadataGenerator, value, expected):
+    cls = models_generator._detect_type(value)
+    result = cls.to_internal_value(value)
+    assert result == expected
+    assert value in result.to_representation()
+
+
+def test_replace():
+    assert IsoTimeString(14, 12, 57).replace(minute=58, second=32) == IsoTimeString(14, 58, 32)
+    assert IsoDateString(2014, 12, 5).replace(day=4, month=5) == IsoDateString(2014, 5, 4)
+    assert IsoDatetimeString(2014, 12, 5, 14, 12, 57).replace(minute=58, second=32, day=4, month=5) \
+           == IsoDatetimeString(2014, 5, 4, 14, 58, 32)