text: # 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)
  