Generate ADMX template for Chrome OS policies

components/policy/tools/template_writers/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 writers.adml_writer."""
 
 
 import os
 import sys
 import unittest
 if __name__ == '__main__':
   sys.path.append(os.path.join(os.path.dirname(__file__), '../../../..'))
 
 
 from writers import adml_writer
 from 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'
+
+      'win_config' : {
+        'win' : {
+          'mandatory_category_path': ['test_category'],
+          'recommended_category_path': ['test_category_recommended'],
+          'category_path_strings': {
+            'test_category': 'TestCategory',
+            'test_category_recommended': 'TestCategory - recommended',
+          },
+        },
+        'chrome_os' : {
+          'mandatory_category_path': ['cros_test_category'],
+          'recommended_category_path': ['cros_test_category_recommended'],
+          'category_path_strings': {
+            'cros_test_category': 'CrOSTestCategory',
+            'cros_test_category_recommended': 'CrOSTestCategory - recommended',
+          },
+        },
       },
     }
-    self.writer = adml_writer.GetWriter(config)
+    self.writer = self._GetWriter(config)
     self.writer.messages = {
       'win_supported_winxpsp2': {
         'text': 'Supported on Test OS or higher',
         'desc': 'blah'
       },
       'doc_recommended': {
         'text': 'Recommended',
         'desc': 'bleh'
       },
     }
     self.writer.Init()
 
+  def _GetWriter(self, config):
+    return adml_writer.GetWriter(config)
+
+  def GetCategory(self):
+    return "test_category";
+
+  def GetCategoryString(self):
+    return "TestCategory";
+
   def _InitWriterForAddingPolicyGroups(self, writer):
     '''Initialize the writer for adding policy groups. This method must be
     called before the method "BeginPolicyGroup" can be called. It initializes
     attributes of the writer.
     '''
     writer.BeginTemplate()
 
   def _InitWriterForAddingPolicies(self, writer, policy):
     '''Initialize the writer for adding policies. This method must be
     called before the method "WritePolicy" can be called. It initializes
@@ skipping 14 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><string id="test_category">TestCategory'
-        '</string><string id="test_recommended_category">'
-        'TestCategory - recommended</string></stringTable><presentationTable/>'
+        ' Test OS or higher</string>'
+        '<string id="' + self.GetCategory() + '">' + \
+          self.GetCategoryString() + '</string>'
+        '<string id="' + self.GetCategory() + '_recommended">' + \
+          self.GetCategoryString() + ' - 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><string id="test_category">TestCategory'
-        '</string><string id="test_recommended_category">'
-        'TestCategory - recommended</string></stringTable><presentationTable/>'
+        ' Test OS or higher</string>'
+        '<string id="' + self.GetCategory() + '">' + \
+          self.GetCategoryString() + '</string>'
+        '<string id="' + self.GetCategory() + '_recommended">' + \
+          self.GetCategoryString() + ' - 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="' + self.GetCategory() + '">' + \
+          self.GetCategoryString() + '</string>\n'
+        '<string id="' + self.GetCategory() + '_recommended">' + \
+          self.GetCategoryString() + ' - 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 89 matching lines @@
       ],
     }
     self. _InitWriterForAddingPolicies(self.writer, enum_policy)
     self.writer.WritePolicy(enum_policy)
     # Assert generated string elements.
     output = self.GetXMLOfChildren(self.writer._string_table_elem)
     expected_output = (
         '<string id="EnumPolicyStub">Enum policy caption</string>\n'
         '<string id="EnumPolicyStub_Explain">'
         'This is a test description.</string>\n'
-        '<string id="item 1">Caption Item 1</string>\n'
-        '<string id="item 2">Caption Item 2</string>')
+        '<string id="EnumPolicyStub_item 1">Caption Item 1</string>\n'
+        '<string id="EnumPolicyStub_item 2">Caption Item 2</string>')
     self.AssertXMLEquals(output, expected_output)
     # Assert generated presentation elements.
     output = self.GetXMLOfChildren(self.writer._presentation_table_elem)
     expected_output = (
         '<presentation id="EnumPolicyStub">\n'
         '  <dropdownList refId="EnumPolicyStub">'
         'Enum policy label</dropdownList>\n'
         '</presentation>')
     self.AssertXMLEquals(output, expected_output)
 
@@ skipping 18 matching lines @@
       ],
     }
     self. _InitWriterForAddingPolicies(self.writer, enum_policy)
     self.writer.WritePolicy(enum_policy)
     # Assert generated string elements.
     output = self.GetXMLOfChildren(self.writer._string_table_elem)
     expected_output = (
         '<string id="EnumPolicyStub">Enum policy caption</string>\n'
         '<string id="EnumPolicyStub_Explain">'
         'This is a test description.</string>\n'
-        '<string id="item 1">Caption Item 1</string>\n'
-        '<string id="item 2">Caption Item 2</string>')
+        '<string id="EnumPolicyStub_item 1">Caption Item 1</string>\n'
+        '<string id="EnumPolicyStub_item 2">Caption Item 2</string>')
     self.AssertXMLEquals(output, expected_output)
     # Assert generated presentation elements.
     output = self.GetXMLOfChildren(self.writer._presentation_table_elem)
     expected_output = (
         '<presentation id="EnumPolicyStub">\n'
         '  <dropdownList refId="EnumPolicyStub">'
         'Enum policy label</dropdownList>\n'
         '</presentation>')
     self.AssertXMLEquals(output, expected_output)
 
@@ skipping 103 matching lines @@
 
   def testStringEncodings(self):
     enum_policy_a = {
       'name': 'EnumPolicy.A',
       'type': 'string-enum',
       'caption': 'Enum policy A caption',
       'label': 'Enum policy A label',
       'desc': 'This is a test description.',
       'items': [
           {
-           'name': 'tls1.2',
-           'value': 'tls1.2',
-           'caption': 'tls1.2',
+           'name': 'same_item',
+           'value': '1',
+           'caption': 'caption_a',
           }
       ],
     }
     enum_policy_b = {
       'name': 'EnumPolicy.B',
       'type': 'string-enum',
       'caption': 'Enum policy B caption',
       'label': 'Enum policy B label',
       'desc': 'This is a test description.',
       'items': [
           {
-           'name': 'tls1.2',
-           'value': 'tls1.2',
-           'caption': 'tls1.2',
+           'name': 'same_item',
+           'value': '2',
+           'caption': 'caption_b',
           }
       ],
     }
     self. _InitWriterForAddingPolicies(self.writer, enum_policy_a)
     self.writer.WritePolicy(enum_policy_a)
     self.writer.WritePolicy(enum_policy_b)
     # Assert generated string elements.
     output = self.GetXMLOfChildren(self.writer._string_table_elem)
     expected_output = (
         '<string id="EnumPolicy_A">Enum policy A caption</string>\n'
         '<string id="EnumPolicy_A_Explain">'
         'This is a test description.</string>\n'
-        '<string id="tls1_2">tls1.2</string>\n'
+        '<string id="EnumPolicy_A_same_item">caption_a</string>\n'
         '<string id="EnumPolicy_B">Enum policy B caption</string>\n'
         '<string id="EnumPolicy_B_Explain">'
-        'This is a test description.</string>\n')
+        'This is a test description.</string>\n'
+        '<string id="EnumPolicy_B_same_item">caption_b</string>\n')
     self.AssertXMLEquals(output, expected_output)
     # Assert generated presentation elements.
     output = self.GetXMLOfChildren(self.writer._presentation_table_elem)
     expected_output = (
         '<presentation id="EnumPolicy.A">\n'
         '  <dropdownList refId="EnumPolicy.A">'
         'Enum policy A label</dropdownList>\n'
         '</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()