Split off 'Google' category into separate ADMX/ADML files

tools/grit/grit/format/policy_templates/writers/adml_writer_unittest.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 
 """Unittests for grit.format.policy_templates.writers.adml_writer."""
 
 
 import os
 import sys
 import unittest
 if __name__ == '__main__':
   sys.path.append(os.path.join(os.path.dirname(__file__), '../../../..'))
 
 
 from grit.format.policy_templates.writers import adml_writer
 from grit.format.policy_templates.writers import xml_writer_base_unittest
 
 
 class AdmlWriterUnittest(xml_writer_base_unittest.XmlWriterBaseTest):
 
   def setUp(self):
     config = {
       'app_name': 'test',
       'build': 'test',
       'win_supported_os': 'SUPPORTED_TESTOS',
+      'win_mandatory_category_path': ['test_category'],
+      'win_recommended_category_path': ['test_recommended_category'],
+      'win_category_path_strings': {
+        'test_category': 'TestCategory',
+        'test_recommended_category': 'TestCategory - recommended'
+      },
     }
     self.writer = adml_writer.GetWriter(config)
     self.writer.messages = {
       'win_supported_winxpsp2': {
         'text': 'Supported on Test OS or higher',
         'desc': 'blah'
       },
       'doc_recommended': {
         'text': 'Recommended',
         'desc': 'bleh'
@@ skipping 28 matching lines @@
       self.writer._string_table_elem.removeChild(elem)
 
   def testEmpty(self):
     self.writer.BeginTemplate()
     self.writer.EndTemplate()
     output = self.writer.GetTemplateText()
     expected_output = (
         '<?xml version="1.0" ?><policyDefinitionResources'
         ' revision="1.0" schemaVersion="1.0"><displayName/><description/>'
         '<resources><stringTable><string id="SUPPORTED_TESTOS">Supported on'
-        ' Test OS or higher</string></stringTable><presentationTable/>'
+        ' Test OS or higher</string><string id="test_category">TestCategory'
+        '</string><string id="test_recommended_category">'
+        'TestCategory - recommended</string></stringTable><presentationTable/>'
         '</resources></policyDefinitionResources>')
     self.AssertXMLEquals(output, expected_output)
 
   def testVersionAnnotation(self):
     self.writer.config['version'] = '39.0.0.0'
     self.writer.BeginTemplate()
     self.writer.EndTemplate()
     output = self.writer.GetTemplateText()
     expected_output = (
         '<?xml version="1.0" ?><policyDefinitionResources'
         ' revision="1.0" schemaVersion="1.0"><!--test version: 39.0.0.0-->'
         '<displayName/><description/><resources><stringTable>'
         '<string id="SUPPORTED_TESTOS">Supported on'
-        ' Test OS or higher</string></stringTable><presentationTable/>'
+        ' Test OS or higher</string><string id="test_category">TestCategory'
+        '</string><string id="test_recommended_category">'
+        'TestCategory - recommended</string></stringTable><presentationTable/>'
         '</resources></policyDefinitionResources>')
     self.AssertXMLEquals(output, expected_output)
 
   def testPolicyGroup(self):
     empty_policy_group = {
       'name': 'PolicyGroup',
       'caption': 'Test Group Caption',
       'desc': 'This is the test description of the test policy group.',
       'policies': [
           {'name': 'PolicyStub2',
            'type': 'main'},
           {'name': 'PolicyStub1',
            'type': 'main'},
       ],
     }
     self._InitWriterForAddingPolicyGroups(self.writer)
     self.writer.BeginPolicyGroup(empty_policy_group)
     self.writer.EndPolicyGroup
     # Assert generated string elements.
     output = self.GetXMLOfChildren(self.writer._string_table_elem)
     expected_output = (
         '<string id="SUPPORTED_TESTOS">'
         'Supported on Test OS or higher</string>\n'
+        '<string id="test_category">TestCategory</string>\n'
+        '<string id="test_recommended_category">'
+        'TestCategory - recommended</string>\n'
         '<string id="PolicyGroup_group">Test Group Caption</string>')
     self.AssertXMLEquals(output, expected_output)
     # Assert generated presentation elements.
     output = self.GetXMLOfChildren(self.writer._presentation_table_elem)
     expected_output = ''
     self.AssertXMLEquals(output, expected_output)
 
   def testMainPolicy(self):
     main_policy = {
       'name': 'DummyMainPolicy',
@@ skipping 304 matching lines @@
         '</presentation>\n'
         '<presentation id="EnumPolicy.B">\n'
         '  <dropdownList refId="EnumPolicy.B">'
         'Enum policy B label</dropdownList>\n'
         '</presentation>')
     self.AssertXMLEquals(output, expected_output)
 
 
 if __name__ == '__main__':
   unittest.main()