Index: tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py |
diff --git a/tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py b/tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py |
index 03a43992473a3822fa10f25ee76e2c1ff571c395..0ac74a620529438239c1a93726b3069b2b5e05ee 100644 |
--- a/tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py |
+++ b/tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py |
@@ -1,5 +1,5 @@ |
#!/usr/bin/python2.4 |
-# Copyright (c) 2010 The Chromium Authors. All rights reserved. |
+# Copyright (c) 2011 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. |
@@ -55,7 +55,7 @@ class AdmlWriterTest(xml_writer_base_unittest.XmlWriterBaseTest): |
} |
writer.BeginPolicyGroup(policy_group) |
- string_elements =\ |
+ string_elements = \ |
self.writer._string_table_elem.getElementsByTagName('string') |
for elem in string_elements: |
self.writer._string_table_elem.removeChild(elem) |
@@ -158,6 +158,38 @@ class AdmlWriterTest(xml_writer_base_unittest.XmlWriterBaseTest): |
'</presentation>') |
self.AssertXMLEquals(output, expected_output) |
+ def testIntPolicy(self): |
+ int_policy = { |
+ 'name': 'IntPolicyStub', |
+ 'type': 'int', |
+ 'caption': 'Int policy caption', |
+ 'label': 'Int policy label', |
+ 'desc': 'This is a test description.', |
+ } |
+ self. _InitWriterForAddingPolicies(self.writer, int_policy) |
+ self.writer.WritePolicy(int_policy) |
+ # Assert generated string elements. |
+ output = self.GetXMLOfChildren(self.writer._string_table_elem) |
+ expected_output = ( |
+ '<string id="IntPolicyStub">\n' |
+ ' Int policy caption\n' |
+ '</string>\n' |
+ '<string id="IntPolicyStub_Explain">\n' |
+ ' This is a test description.\n' |
+ '</string>') |
+ self.AssertXMLEquals(output, expected_output) |
+ # Assert generated presentation elements. |
+ output = self.GetXMLOfChildren(self.writer._presentation_table_elem) |
+ expected_output = ( |
+ '<presentation id="IntPolicyStub">\n' |
+ ' <decimalTextBox refId="IntPolicyStub">\n' |
+ ' <label>\n' |
+ ' Int policy label\n' |
+ ' </label>\n' |
+ ' </decimalTextBox>\n' |
+ '</presentation>') |
+ self.AssertXMLEquals(output, expected_output) |
+ |
def testIntEnumPolicy(self): |
enum_policy = { |
'name': 'EnumPolicyStub', |