73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
import copy
|
|
import json
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
import generate_relay_protocol as generator
|
|
|
|
|
|
class RelayProtocolGeneratorTest(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.spec = json.loads(generator.SPEC_PATH.read_text(encoding="utf-8"))
|
|
|
|
def test_supported_pattern_renders_both_targets(self) -> None:
|
|
dart_output = generator.dart_source(copy.deepcopy(self.spec))
|
|
go_output = generator.go_source(copy.deepcopy(self.spec))
|
|
|
|
self.assertIn(
|
|
f"RegExp(r{generator.SUPPORTED_ID_PATTERN!r})",
|
|
dart_output,
|
|
)
|
|
self.assertIn("func validRelayID(value string, maxLength int) bool", go_output)
|
|
|
|
def test_changed_pattern_fails_before_writing_either_target(self) -> None:
|
|
changed_spec = copy.deepcopy(self.spec)
|
|
changed_spec["idPattern"] = r"^[A-Za-z0-9_.-]+$"
|
|
|
|
for renderer in (generator.dart_source, generator.go_source):
|
|
with self.subTest(renderer=renderer.__name__):
|
|
with self.assertRaisesRegex(ValueError, "idPattern"):
|
|
renderer(changed_spec)
|
|
|
|
with tempfile.TemporaryDirectory() as temporary_directory:
|
|
root = Path(temporary_directory)
|
|
spec_path = root / "relay_protocol.json"
|
|
dart_path = root / "relay_protocol.g.dart"
|
|
go_path = root / "relay_protocol_gen.go"
|
|
spec_path.write_text(json.dumps(changed_spec), encoding="utf-8")
|
|
dart_path.write_text("dart sentinel\n", encoding="utf-8")
|
|
go_path.write_text("go sentinel\n", encoding="utf-8")
|
|
|
|
with (
|
|
mock.patch.object(generator, "SPEC_PATH", spec_path),
|
|
mock.patch.object(generator, "DART_PATH", dart_path),
|
|
mock.patch.object(generator, "GO_PATH", go_path),
|
|
):
|
|
with self.assertRaisesRegex(ValueError, "idPattern"):
|
|
generator.main()
|
|
|
|
self.assertEqual(dart_path.read_text(encoding="utf-8"), "dart sentinel\n")
|
|
self.assertEqual(go_path.read_text(encoding="utf-8"), "go sentinel\n")
|
|
|
|
def test_missing_pattern_is_rejected(self) -> None:
|
|
spec = copy.deepcopy(self.spec)
|
|
del spec["idPattern"]
|
|
|
|
with self.assertRaisesRegex(ValueError, "idPattern is required"):
|
|
generator.validated_id_pattern(spec)
|
|
|
|
def test_non_string_pattern_is_rejected(self) -> None:
|
|
for value in (None, 42, [generator.SUPPORTED_ID_PATTERN]):
|
|
with self.subTest(value=value):
|
|
spec = copy.deepcopy(self.spec)
|
|
spec["idPattern"] = value
|
|
|
|
with self.assertRaisesRegex(ValueError, "idPattern must be a string"):
|
|
generator.validated_id_pattern(spec)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|