blob: 9966b0b7f8f53bdb2660d2cf001e817146e2eefa [file] [log] [blame]
# 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)