Support string format

This commit is contained in:
Pu Chen
2025-05-06 21:51:06 +08:00
parent 7a3266e4cc
commit b52997633c
3 changed files with 245 additions and 3 deletions

View File

@@ -1,5 +1,8 @@
from jambo.parser import StringTypeParser
from pydantic import EmailStr, HttpUrl, IPvAnyAddress
from datetime import date, datetime, time
from unittest import TestCase
@@ -85,3 +88,112 @@ class TestStringTypeParser(TestCase):
with self.assertRaises(ValueError):
parser.from_properties("placeholder", properties)
def test_string_parser_with_email_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "email",
}
type_parsing, type_validator = parser.from_properties("placeholder", properties)
self.assertEqual(type_parsing, EmailStr)
def test_string_parser_with_uri_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "uri",
}
type_parsing, type_validator = parser.from_properties("placeholder", properties)
self.assertEqual(type_parsing, HttpUrl)
def test_string_parser_with_ip_formats(self):
parser = StringTypeParser()
for ip_format in ["ipv4", "ipv6"]:
properties = {
"type": "string",
"format": ip_format,
}
type_parsing, type_validator = parser.from_properties(
"placeholder", properties
)
self.assertEqual(type_parsing, IPvAnyAddress)
def test_string_parser_with_time_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "time",
}
type_parsing, type_validator = parser.from_properties("placeholder", properties)
self.assertEqual(type_parsing, time)
def test_string_parser_with_pattern_based_formats(self):
parser = StringTypeParser()
for format_type in ["hostname"]:
properties = {
"type": "string",
"format": format_type,
}
type_parsing, type_validator = parser.from_properties(
"placeholder", properties
)
self.assertEqual(type_parsing, str)
self.assertIn("pattern", type_validator)
self.assertEqual(
type_validator["pattern"], parser.format_pattern_mapping[format_type]
)
def test_string_parser_with_unsupported_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "unsupported-format",
}
with self.assertRaises(ValueError) as context:
parser.from_properties("placeholder", properties)
self.assertEqual(
str(context.exception), "Unsupported string format: unsupported-format"
)
def test_string_parser_with_date_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "date",
}
type_parsing, type_validator = parser.from_properties("placeholder", properties)
self.assertEqual(type_parsing, date)
def test_string_parser_with_datetime_format(self):
parser = StringTypeParser()
properties = {
"type": "string",
"format": "date-time",
}
type_parsing, type_validator = parser.from_properties("placeholder", properties)
self.assertEqual(type_parsing, datetime)