Include chromium version number in policy templates

grit/format/policy_templates/writers/admx_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.admx_writer."""
 
 
 import os
@@ skipping 19 matching lines @@
     # Writer configuration. This dictionary contains parameter used by the ADMX
     # Writer
     config = {
       'win_group_policy_class': 'TestClass',
       'win_supported_os': 'SUPPORTED_TESTOS',
       'win_reg_mandatory_key_name': 'Software\\Policies\\Test',
       'win_reg_recommended_key_name': 'Software\\Policies\\Test\\Recommended',
       'win_mandatory_category_path': ['test_category'],
       'win_recommended_category_path': ['test_recommended_category'],
       'admx_namespace': 'ADMXWriter.Test.Namespace',
-      'admx_prefix': 'test_prefix'
+      'admx_prefix': 'test_prefix',
+      'build': 'test_product',
     }
     self.writer = admx_writer.GetWriter(config)
     self.writer.Init()
 
   def _GetPoliciesElement(self, doc):
     node_list = doc.getElementsByTagName('policies')
     self.assertTrue(node_list.length == 1)
     return node_list.item(0)
 
   def _GetCategoriesElement(self, doc):
@@ skipping 24 matching lines @@
         '  <categories>\n'
         '    <category displayName="$(string.test_category)"'
         ' name="test_category"/>\n'
         '    <category displayName="$(string.test_recommended_category)"'
         ' name="test_recommended_category"/>\n'
         '  </categories>\n'
         '  <policies/>\n'
         '</policyDefinitions>')
     self.AssertXMLEquals(output, expected_output)
 
+  def testEmptyVersion(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" ?>\n'
+        '<policyDefinitions revision="1.0" schemaVersion="1.0">\n'
+        '  <!--test_product version: 39.0.0.0-->\n'
+        '  <policyNamespaces>\n'
+        '    <target namespace="ADMXWriter.Test.Namespace"'
+        ' prefix="test_prefix"/>\n'
+        '    <using namespace="Microsoft.Policies.Windows" prefix="windows"/>\n'
+        '  </policyNamespaces>\n'
+        '  <resources minRequiredRevision="1.0"/>\n'
+        '  <supportedOn>\n'
+        '    <definitions>\n'
+        '      <definition displayName="'
+        '$(string.SUPPORTED_TESTOS)" name="SUPPORTED_TESTOS"/>\n'
+        '    </definitions>\n'
+        '  </supportedOn>\n'
+        '  <categories>\n'
+        '    <category displayName="$(string.test_category)"'
+        ' name="test_category"/>\n'
+        '    <category displayName="$(string.test_recommended_category)"'
+        ' name="test_recommended_category"/>\n'
+        '  </categories>\n'
+        '  <policies/>\n'
+        '</policyDefinitions>')
+    self.AssertXMLEquals(output, expected_output)
+
   def testEmptyPolicyGroup(self):
     empty_policy_group = {
       'name': 'PolicyGroup',
       'policies': []
     }
     # Initialize writer to write a policy group.
     self.writer.BeginTemplate()
     # Write policy group
     self.writer.BeginPolicyGroup(empty_policy_group)
     self.writer.EndPolicyGroup()
@@ skipping 374 matching lines @@
     }))
     self.assertFalse(self.writer.IsPolicySupported({
       'supported_on': [
         {'platforms': ['mac', 'linux']}, {'platforms': ['aaa']}
       ]
     }))
 
 
 if __name__ == '__main__':
   unittest.main()