OLD | NEW |
| (Empty) |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 import imp | |
6 import os.path | |
7 import sys | |
8 import unittest | |
9 | |
10 | |
11 def _GetDirAbove(dirname): | |
12 """Returns the directory "above" this file containing |dirname| (which must | |
13 also be "above" this file).""" | |
14 path = os.path.abspath(__file__) | |
15 while True: | |
16 path, tail = os.path.split(path) | |
17 assert tail | |
18 if tail == dirname: | |
19 return path | |
20 | |
21 | |
22 try: | |
23 imp.find_module("mojom") | |
24 except ImportError: | |
25 sys.path.append(os.path.join(_GetDirAbove("pylib"), "pylib")) | |
26 from mojom.generate import pack | |
27 from mojom.generate import module as mojom | |
28 | |
29 | |
30 # TODO(yzshen): Move tests in pack_tests.py here. | |
31 class PackTest(unittest.TestCase): | |
32 def _CheckPackSequence(self, kinds, fields, offsets): | |
33 """Checks the pack order and offsets of a sequence of mojom.Kinds. | |
34 | |
35 Args: | |
36 kinds: A sequence of mojom.Kinds that specify the fields that are to be | |
37 created. | |
38 fields: The expected order of the resulting fields, with the integer "1" | |
39 first. | |
40 offsets: The expected order of offsets, with the integer "0" first. | |
41 """ | |
42 struct = mojom.Struct('test') | |
43 index = 1 | |
44 for kind in kinds: | |
45 struct.AddField('%d' % index, kind) | |
46 index += 1 | |
47 ps = pack.PackedStruct(struct) | |
48 num_fields = len(ps.packed_fields) | |
49 self.assertEquals(len(kinds), num_fields) | |
50 for i in xrange(num_fields): | |
51 self.assertEquals('%d' % fields[i], ps.packed_fields[i].field.name) | |
52 self.assertEquals(offsets[i], ps.packed_fields[i].offset) | |
53 | |
54 def testMinVersion(self): | |
55 """Tests that |min_version| is properly set for packed fields.""" | |
56 struct = mojom.Struct('test') | |
57 struct.AddField('field_2', mojom.BOOL, 2) | |
58 struct.AddField('field_0', mojom.INT32, 0) | |
59 struct.AddField('field_1', mojom.INT64, 1) | |
60 ps = pack.PackedStruct(struct) | |
61 | |
62 self.assertEquals('field_0', ps.packed_fields[0].field.name) | |
63 self.assertEquals('field_2', ps.packed_fields[1].field.name) | |
64 self.assertEquals('field_1', ps.packed_fields[2].field.name) | |
65 | |
66 self.assertEquals(0, ps.packed_fields[0].min_version) | |
67 self.assertEquals(0, ps.packed_fields[1].min_version) | |
68 self.assertEquals(0, ps.packed_fields[2].min_version) | |
69 | |
70 struct.fields[0].attributes = {'MinVersion': 1} | |
71 ps = pack.PackedStruct(struct) | |
72 | |
73 self.assertEquals(0, ps.packed_fields[0].min_version) | |
74 self.assertEquals(1, ps.packed_fields[1].min_version) | |
75 self.assertEquals(0, ps.packed_fields[2].min_version) | |
76 | |
77 def testGetVersionInfoEmptyStruct(self): | |
78 """Tests that pack.GetVersionInfo() never returns an empty list, even for | |
79 empty structs. | |
80 """ | |
81 struct = mojom.Struct('test') | |
82 ps = pack.PackedStruct(struct) | |
83 | |
84 versions = pack.GetVersionInfo(ps) | |
85 self.assertEquals(1, len(versions)) | |
86 self.assertEquals(0, versions[0].version) | |
87 self.assertEquals(0, versions[0].num_fields) | |
88 self.assertEquals(8, versions[0].num_bytes) | |
89 | |
90 def testGetVersionInfoComplexOrder(self): | |
91 """Tests pack.GetVersionInfo() using a struct whose definition order, | |
92 ordinal order and pack order for fields are all different. | |
93 """ | |
94 struct = mojom.Struct('test') | |
95 struct.AddField('field_3', mojom.BOOL, ordinal=3, | |
96 attributes={'MinVersion': 3}) | |
97 struct.AddField('field_0', mojom.INT32, ordinal=0) | |
98 struct.AddField('field_1', mojom.INT64, ordinal=1, | |
99 attributes={'MinVersion': 2}) | |
100 struct.AddField('field_2', mojom.INT64, ordinal=2, | |
101 attributes={'MinVersion': 3}) | |
102 ps = pack.PackedStruct(struct) | |
103 | |
104 versions = pack.GetVersionInfo(ps) | |
105 self.assertEquals(3, len(versions)) | |
106 | |
107 self.assertEquals(0, versions[0].version) | |
108 self.assertEquals(1, versions[0].num_fields) | |
109 self.assertEquals(16, versions[0].num_bytes) | |
110 | |
111 self.assertEquals(2, versions[1].version) | |
112 self.assertEquals(2, versions[1].num_fields) | |
113 self.assertEquals(24, versions[1].num_bytes) | |
114 | |
115 self.assertEquals(3, versions[2].version) | |
116 self.assertEquals(4, versions[2].num_fields) | |
117 self.assertEquals(32, versions[2].num_bytes) | |
118 | |
119 def testInterfaceAlignment(self): | |
120 """Tests that interfaces are aligned on 4-byte boundaries, although the size | |
121 of an interface is 8 bytes. | |
122 """ | |
123 kinds = (mojom.INT32, mojom.Interface('test_interface')) | |
124 fields = (1, 2) | |
125 offsets = (0, 4) | |
126 self._CheckPackSequence(kinds, fields, offsets) | |
OLD | NEW |