1 files changed, 24 insertions, 0 deletions
diff --git a/mojo/public/tools/bindings/pylib/mojom/generate/generator_unittest.py b/mojo/public/tools/bindings/pylib/mojom/generate/generator_unittest.py
new file mode 100644
index 0000000000..9966b0b7f8
--- /dev/null
+++ b/mojo/public/tools/bindings/pylib/mojom/generate/generator_unittest.py
@@ -0,0 +1,24 @@
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import unittest
+
+import module as mojom
+import generator
+
+class TestGenerator(unittest.TestCase):
+
+  def testGetUnionsAddsOrdinals(self):
+    module = mojom.Module()
+    union = module.AddUnion('a')
+    union.AddField('a', mojom.BOOL)
+    union.AddField('b', mojom.BOOL)
+    union.AddField('c', mojom.BOOL, ordinal=10)
+    union.AddField('d', mojom.BOOL)
+
+    gen = generator.Generator(module)
+    union = gen.GetUnions()[0]
+    ordinals = [field.ordinal for field in union.fields]
+
+    self.assertEquals([0, 1, 10, 11], ordinals)