Merge pull request #60 from godatadriven/add-custom-name-test

Added test for a custom name
godatadriven · Apr 18, 2023 · 705dc7c · 705dc7c
2 parents 872b556 + 5897989
commit 705dc7c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_to_avro.py b/tests/test_to_avro.py
@@ -400,3 +400,20 @@ def test_int():
         "name": "IntModel",
         "fields": [{"type": "int", "name": "c1"}],
     }
+
+
+class CustomNameModel(AvroBase):
+    c1: int
+
+    class Config:
+        title = "some_other_name"
+
+
+def test_custom_name():
+    result = CustomNameModel.avro_schema()
+    assert result == {
+        "type": "record",
+        "namespace": "some_other_name",
+        "name": "some_other_name",
+        "fields": [{"type": "long", "name": "c1"}],
+    }