Include chromium version number in policy templates

grit/format/policy_templates/writers/doc_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.
 
 '''Unit tests for grit.format.policy_templates.writers.doc_writer'''
 
 
 import json
 import os
@@ skipping 21 matching lines @@
 
   def setUp(self):
     # Create a writer for the tests.
     self.writer = doc_writer.GetWriter(
       config={
         'app_name': 'Chrome',
         'frame_name': 'Chrome Frame',
         'os_name': 'Chrome OS',
         'win_reg_mandatory_key_name': 'MockKey',
         'win_reg_recommended_key_name': 'MockKeyRec',
+        'build': 'test_product',
       })
     self.writer.messages = {
       'doc_back_to_top': {'text': '_test_back_to_top'},
       'doc_complex_policies_on_windows': {'text': '_test_complex_policies_win'},
       'doc_data_type': {'text': '_test_data_type'},
       'doc_description': {'text': '_test_description'},
       'doc_description_column_title': {
         'text': '_test_description_column_title'
       },
       'doc_example_value': {'text': '_test_example_value'},
@@ skipping 44 matching lines @@
                 '<td style="style_td;style_td.right;style_thead td;">'
                   '_test_description_column_title'
                 '</td>'
               '</tr></thead>'
               '<tbody/>'
             '</table>'
           '</div>'
           '<div/>'
         '</div>')
 
+  def testVersionAnnotation(self):
+    # Test if DocWriter creates the skeleton of the document correctly.
+    self.writer.config['version'] = '39.0.0.0'
+    self.writer.BeginTemplate()
+    self.assertEquals(
+        self.writer._main_div.toxml(),
+        '<div>'
+          '<!--test_product version: 39.0.0.0-->'
+          '<div>'
+            '<a name="top"/><br/>_test_intro<br/><br/><br/>'
+            '<table style="style_table;">'
+              '<thead><tr style="style_tr;">'
+                '<td style="style_td;style_td.left;style_thead td;">'
+                  '_test_name_column_title'
+                '</td>'
+                '<td style="style_td;style_td.right;style_thead td;">'
+                  '_test_description_column_title'
+                '</td>'
+              '</tr></thead>'
+              '<tbody/>'
+            '</table>'
+          '</div>'
+          '<div/>'
+        '</div>')
+
   def testGetLocalizedMessage(self):
     # Test if localized messages are retrieved correctly.
     self.writer.messages = {
       'doc_hello_world': {'text': 'hello, vilag!'}
     }
     self.assertEquals(
         self.writer._GetLocalizedMessage('hello_world'),
         'hello, vilag!')
 
   def testMapListToString(self):
@@ skipping 732 matching lines @@
             '  &lt;key&gt;True&lt;/key&gt;\n'
             '  &lt;true/&gt;\n'
             '&lt;/dict&gt;'
           '</dd>'
         '</dl>'
       '</root>')
 
 
 if __name__ == '__main__':
   unittest.main()