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