Add optional mandatory policy setting for template generation

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
@@ skipping 68 matching lines @@
     output = self.GetOutput(grd, 'fr', {'_google_chrome' : '1'}, 'json', 'en')
     expected_output = (
         TEMPLATE_HEADER +
         '  // Example Main Policy\n' +
         HEADER_DELIMETER +
         '  // Example Main Policy\n\n'
         '  //"MainPolicy": true\n\n'
         '}')
     self.CompareOutputs(output, expected_output)
 
+  def testRecommendedOnlyPolicy(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",'
+        '      "features": {'
+        '        "can_be_recommended": True,'
+        '        "can_be_mandatory": False'
+        '      },'
+        '      "supported_on": ["chrome.linux:8-"],'
+        '      "example_value": True'
+        '    },'
+        '  ],'
+        '  "placeholders": [],'
+        '  "messages": {},'
+        '}')
+    output = self.GetOutput(grd, 'fr', {'_google_chrome' : '1'}, 'json', 'en')
+    expected_output = (
+        TEMPLATE_HEADER +
+        '  // Note: this policy is supported only in recommended mode.\n' +
+        '  // The JSON file should be placed in' +
+        ' /etc/opt/chrome/policies/recommended.\n' +
+        '  // Example Main Policy\n' +
+        HEADER_DELIMETER +
+        '  // Example Main Policy\n\n'
+        '  //"MainPolicy": true\n\n'
+        '}')
+    self.CompareOutputs(output, expected_output)
+
   def testStringPolicy(self):
     # Tests a policy group with a single policy of type 'string'.
     grd = self.PrepareTest(
         '{'
         '  "policy_definitions": ['
         '    {'
         '      "name": "StringPolicy",'
         '      "type": "string",'
         '      "caption": "Example String Policy",'
         '      "desc": "Example String Policy",'
@@ skipping 265 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()