OLD | NEW |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import unittest | 5 import unittest |
6 | 6 |
7 import fieldtrial_to_struct | 7 import fieldtrial_to_struct |
8 import os | 8 import os |
9 | 9 |
10 | 10 |
(...skipping 30 matching lines...) Expand all Loading... |
41 { | 41 { |
42 'platforms': ['win'], | 42 'platforms': ['win'], |
43 'experiments': [{'name': 'OtherGroup'}] | 43 'experiments': [{'name': 'OtherGroup'}] |
44 } | 44 } |
45 ] | 45 ] |
46 } | 46 } |
47 result = fieldtrial_to_struct._FieldTrialConfigToDescription(config, 'win') | 47 result = fieldtrial_to_struct._FieldTrialConfigToDescription(config, 'win') |
48 expected = { | 48 expected = { |
49 'elements': { | 49 'elements': { |
50 'kFieldTrialConfig': { | 50 'kFieldTrialConfig': { |
51 'trials': [ | 51 'studies': [ |
52 { | 52 { |
53 'name': 'Trial1', | 53 'name': 'Trial1', |
54 'groups': [ | 54 'experiments': [ |
55 { | 55 { |
56 'name': 'Group1', | 56 'name': 'Group1', |
57 'params': [ | 57 'params': [ |
58 {'key': 'x', 'value': '1'}, | 58 {'key': 'x', 'value': '1'}, |
59 {'key': 'y', 'value': '2'} | 59 {'key': 'y', 'value': '2'} |
60 ], | 60 ], |
61 'enable_features': ['A', 'B'], | 61 'enable_features': ['A', 'B'], |
62 'disable_features': ['C'] | 62 'disable_features': ['C'] |
63 }, | 63 }, |
64 { | 64 { |
65 'name': 'Group2', | 65 'name': 'Group2', |
66 'params': [ | 66 'params': [ |
67 {'key': 'x', 'value': '3'}, | 67 {'key': 'x', 'value': '3'}, |
68 {'key': 'y', 'value': '4'} | 68 {'key': 'y', 'value': '4'} |
69 ], | 69 ], |
70 'enable_features': ['D', 'E'], | 70 'enable_features': ['D', 'E'], |
71 'disable_features': ['F'] | 71 'disable_features': ['F'] |
72 }, | 72 }, |
73 ], | 73 ], |
74 }, | 74 }, |
75 { | 75 { |
76 'name': 'Trial2', | 76 'name': 'Trial2', |
77 'groups': [{'name': 'OtherGroup'}] | 77 'experiments': [{'name': 'OtherGroup'}] |
78 }, | 78 }, |
79 ] | 79 ] |
80 } | 80 } |
81 } | 81 } |
82 } | 82 } |
83 self.maxDiff = None | 83 self.maxDiff = None |
84 self.assertEqual(expected, result) | 84 self.assertEqual(expected, result) |
85 | 85 |
86 _MULTIPLE_PLATFORM_CONFIG = { | 86 _MULTIPLE_PLATFORM_CONFIG = { |
87 'Trial1': [ | 87 'Trial1': [ |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
124 } | 124 } |
125 ] | 125 ] |
126 } | 126 } |
127 | 127 |
128 def test_FieldTrialToDescriptionMultipleSinglePlatformMultipleTrial(self): | 128 def test_FieldTrialToDescriptionMultipleSinglePlatformMultipleTrial(self): |
129 result = fieldtrial_to_struct._FieldTrialConfigToDescription( | 129 result = fieldtrial_to_struct._FieldTrialConfigToDescription( |
130 self._MULTIPLE_PLATFORM_CONFIG, 'ios') | 130 self._MULTIPLE_PLATFORM_CONFIG, 'ios') |
131 expected = { | 131 expected = { |
132 'elements': { | 132 'elements': { |
133 'kFieldTrialConfig': { | 133 'kFieldTrialConfig': { |
134 'trials': [ | 134 'studies': [ |
135 { | 135 { |
136 'name': 'Trial1', | 136 'name': 'Trial1', |
137 'groups': [ | 137 'experiments': [ |
138 { | 138 { |
139 'name': 'Group1', | 139 'name': 'Group1', |
140 'params': [ | 140 'params': [ |
141 {'key': 'x', 'value': '1'}, | 141 {'key': 'x', 'value': '1'}, |
142 {'key': 'y', 'value': '2'} | 142 {'key': 'y', 'value': '2'} |
143 ], | 143 ], |
144 'enable_features': ['A', 'B'], | 144 'enable_features': ['A', 'B'], |
145 'disable_features': ['C'] | 145 'disable_features': ['C'] |
146 }, | 146 }, |
147 { | 147 { |
(...skipping 16 matching lines...) Expand all Loading... |
164 } | 164 } |
165 self.maxDiff = None | 165 self.maxDiff = None |
166 self.assertEqual(expected, result) | 166 self.assertEqual(expected, result) |
167 | 167 |
168 def test_FieldTrialToDescriptionMultipleSinglePlatformSingleTrial(self): | 168 def test_FieldTrialToDescriptionMultipleSinglePlatformSingleTrial(self): |
169 result = fieldtrial_to_struct._FieldTrialConfigToDescription( | 169 result = fieldtrial_to_struct._FieldTrialConfigToDescription( |
170 self._MULTIPLE_PLATFORM_CONFIG, 'mac') | 170 self._MULTIPLE_PLATFORM_CONFIG, 'mac') |
171 expected = { | 171 expected = { |
172 'elements': { | 172 'elements': { |
173 'kFieldTrialConfig': { | 173 'kFieldTrialConfig': { |
174 'trials': [ | 174 'studies': [ |
175 { | 175 { |
176 'name': 'Trial2', | 176 'name': 'Trial2', |
177 'groups': [ | 177 'experiments': [ |
178 { | 178 { |
179 'name': 'OtherGroup', | 179 'name': 'OtherGroup', |
180 }, | 180 }, |
181 ], | 181 ], |
182 }, | 182 }, |
183 ] | 183 ] |
184 } | 184 } |
185 } | 185 } |
186 } | 186 } |
187 self.maxDiff = None | 187 self.maxDiff = None |
(...skipping 21 matching lines...) Expand all Loading... |
209 cc_filename = test_output_filename + '.cc' | 209 cc_filename = test_output_filename + '.cc' |
210 with open(cc_filename, 'r') as cc: | 210 with open(cc_filename, 'r') as cc: |
211 test_cc = cc.read() | 211 test_cc = cc.read() |
212 with open('unittest_data/expected_output.cc', 'r') as expected: | 212 with open('unittest_data/expected_output.cc', 'r') as expected: |
213 expected_cc = expected.read() | 213 expected_cc = expected.read() |
214 self.assertEqual(expected_cc, test_cc) | 214 self.assertEqual(expected_cc, test_cc) |
215 os.unlink(cc_filename) | 215 os.unlink(cc_filename) |
216 | 216 |
217 if __name__ == '__main__': | 217 if __name__ == '__main__': |
218 unittest.main() | 218 unittest.main() |
OLD | NEW |