chore: add test for integer type

meltano · Oct 21, 2022 · 8eb0026 · 8eb0026
1 parent 372fc34
commit 8eb0026
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/tests/core/test_jsonschema_helpers.py b/tests/core/test_jsonschema_helpers.py
@@ -14,6 +14,8 @@
     is_boolean_type,
     is_date_or_datetime_type,
     is_datetime_type,
+    is_integer_type,
+    is_object_type,
     is_secret_type,
     is_string_array_type,
     is_string_type,
@@ -53,6 +55,7 @@
     is_boolean_type,
     is_date_or_datetime_type,
     is_datetime_type,
+    is_integer_type,
     is_secret_type,
     is_string_array_type,
     is_string_type,
@@ -330,6 +333,38 @@ def test_inbuilt_type(json_type: JSONTypeHelper, expected_json_schema: dict):
             },
             {is_array_type, is_string_array_type},
         ),
+        (
+            Property(
+                "my_prop7",
+                ObjectType(
+                    Property("not_a_secret", StringType),
+                    Property("is_a_secret", StringType, secret=True),
+                ),
+            ),
+            {
+                "my_prop7": {
+                    "type": ["object", "null"],
+                    "properties": {
+                        "not_a_secret": {"type": ["string", "null"]},
+                        "is_a_secret": {
+                            "type": ["string", "null"],
+                            "secret": True,
+                            "writeOnly": True,
+                        },
+                    },
+                }
+            },
+            {is_object_type, is_secret_type},
+        ),
+        (
+            Property("my_prop8", IntegerType),
+            {
+                "my_prop8": {
+                    "type": ["integer", "null"],
+                }
+            },
+            {is_integer_type},
+        ),
     ],
 )
 def test_property_creation(