1 files changed, 72 insertions, 0 deletions
diff --git a/bindings/python/test.py b/bindings/python/test.py
new file mode 100644
index 0000000..29a6c45
--- /dev/null
+++ b/bindings/python/test.py
@@ -0,0 +1,72 @@
+
+import upb
+import unittest
+
+class TestFieldDef(unittest.TestCase):
+  def test_construction(self):
+    fielddef1 = upb.FieldDef()
+    self.assertTrue(fielddef1.number is None)
+    self.assertTrue(fielddef1.name is None)
+    self.assertTrue(fielddef1.type is None)
+    self.assertEqual(fielddef1.label, upb.LABEL_OPTIONAL)
+
+    fielddef2 = upb.FieldDef(number=5, name="field2",
+                             label=upb.LABEL_REQUIRED, type=upb.TYPE_INT32,
+                             type_name="MyType")
+
+    self.assertTrue(id(fielddef1) != id(fielddef2))
+    self.assertEqual(fielddef2.number, 5)
+    self.assertEqual(fielddef2.name, "field2")
+    self.assertEqual(fielddef2.label, upb.LABEL_REQUIRED)
+    self.assertEqual(fielddef2.type, upb.TYPE_INT32)
+    self.assertEqual(fielddef2.type_name, "MyType")
+
+    fielddef2.number = 8
+    self.assertEqual(fielddef2.number, 8)
+
+    fielddef2.name = "xxx"
+    self.assertEqual(fielddef2.name, "xxx")
+
+    fielddef2.label = upb.LABEL_REPEATED
+    self.assertEqual(fielddef2.label, upb.LABEL_REPEATED)
+
+    fielddef2.type = upb.TYPE_FLOAT
+    self.assertEqual(fielddef2.type, upb.TYPE_FLOAT)
+
+  def test_nosubclasses(self):
+    def create_subclass():
+      class MyClass(upb.FieldDef):
+        pass
+
+    self.assertRaises(TypeError, create_subclass)
+
+  # TODO: test that assigning invalid values is properly prevented.
+
+class TestMessageDef(unittest.TestCase):
+  def test_construction(self):
+    msgdef1 = upb.MessageDef()
+    self.assertTrue(msgdef1.fqname is None)
+    self.assertEqual(msgdef1.fields(), [])
+
+    fields = [upb.FieldDef(number=1, name="field1", type=upb.TYPE_INT32)]
+    msgdef2 = upb.MessageDef(fqname="Message2", fields=fields)
+
+    self.assertEqual(set(msgdef2.fields()), set(fields))
+
+    f2 = upb.FieldDef(number=2, name="field2", type=upb.TYPE_INT64)
+    msgdef2.add_field(f2)
+
+    fields.append(f2)
+    self.assertEqual(set(msgdef2.fields()), set(fields))
+
+class TestSymbolTable(unittest.TestCase):
+  def test_construction(self):
+    s = upb.SymbolTable()
+    self.assertEqual(s.defs(), []);
+
+    s.add_def(upb.MessageDef(fqname="A"))
+    self.assertTrue(s.lookup("A") is not None)
+    self.assertTrue(s.lookup("A") is s.lookup("A"))
+
+if __name__ == '__main__':
+  unittest.main()