Include chromium version number in policy templates

grit/format/policy_templates/writers/json_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.json_writer'''
 
 
 import os
 import sys
 if __name__ == '__main__':
   sys.path.append(os.path.join(os.path.dirname(__file__), '../../../..'))
 
 import unittest
 
 from grit.format.policy_templates.writers import writer_unittest_common
 
 
 TEMPLATE_HEADER="""\
 // Policy template for Linux.
 // Uncomment the policies you wish to activate and change their values to
 // something useful for your case. The provided values are for reference only
 // and do not provide meaningful defaults!
 {
 """
 
+TEMPLATE_HEADER_WITH_VERSION="""\
+// chromium version: 39.0.0.0
+// Policy template for Linux.
+// Uncomment the policies you wish to activate and change their values to
+// something useful for your case. The provided values are for reference only
+// and do not provide meaningful defaults!
+{
+"""
+
 
 HEADER_DELIMETER="""\
   //-------------------------------------------------------------------------
 """
 
 
 class JsonWriterUnittest(writer_unittest_common.WriterUnittestCommon):
   '''Unit tests for JsonWriter.'''
 
   def CompareOutputs(self, output, expected_output):
@@ skipping 11 matching lines @@
         expected_output.strip())
 
   def testEmpty(self):
     # Test the handling of an empty policy list.
     grd = self.PrepareTest(
         '{'
         '  "policy_definitions": [],'
         '  "placeholders": [],'
         '  "messages": {},'
         '}')
-    output = self.GetOutput(grd, 'fr', {'_chromium': '1',}, 'json', 'en')
+    output = self.GetOutput(grd, 'fr', {'_chromium': '1'}, 'json', 'en')
     expected_output = TEMPLATE_HEADER + '}'
     self.CompareOutputs(output, expected_output)
 
+  def testEmptyWithVersion(self):
+    # Test the handling of an empty policy list.
+    grd = self.PrepareTest(
+        '{'
+        '  "policy_definitions": [],'
+        '  "placeholders": [],'
+        '  "messages": {},'
+        '}')
+    output = self.GetOutput(
+        grd, 'fr', {'_chromium': '1', 'version':'39.0.0.0'}, 'json', 'en')
+    expected_output = TEMPLATE_HEADER_WITH_VERSION + '}'
+    self.CompareOutputs(output, expected_output)
+
   def testMainPolicy(self):
     # Tests a policy group with a single policy of type 'main'.
     grd = self.PrepareTest(
         '{'
         '  "policy_definitions": ['
         '    {'
         '      "name": "MainPolicy",'
         '      "type": "main",'
         '      "caption": "Example Main Policy",'
         '      "desc": "Example Main Policy",'
@@ skipping 326 matching lines @@
         '  // Policy Two\n' +
         HEADER_DELIMETER +
         '  // Policy Two\n\n'
         '  //"Policy2": "c"\n\n'
         '}')
     self.CompareOutputs(output, expected_output)
 
 
 if __name__ == '__main__':
   unittest.main()