Allow substitution of messages as variables in other messages.

grit/format/rc_unittest.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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.rc'''
 
 import os
 import re
 import sys
@@ skipping 69 matching lines @@
     output = buf.getvalue().strip()
     expected = u'''
 IDC_KLONKMENU MENU
 BEGIN
     POPUP "&File"
     BEGIN
         MENUITEM "E&xit",                       IDM_EXIT
         MENUITEM "This be ""Klonk"" me like",   ID_FILE_THISBE
         POPUP "gonk"
         BEGIN
-            MENUITEM "Klonk && is ""good""",           ID_GONK_KLONKIS
+            MENUITEM "Klonk && is [good]",           ID_GONK_KLONKIS
         END
     END
     POPUP "&Help"
     BEGIN
         MENUITEM "&About ...",                  IDM_ABOUT
     END
 END
 
 IDD_ABOUTBOX DIALOGEX 22, 17, 230, 75
 STYLE DS_SETFONT | DS_MODALFRAME | WS_CAPTION | WS_SYSMENU
@@ skipping 138 matching lines @@
     self.failUnless(fr_file == os.path.join(output_dir, 'fr_simple.html'))
 
     fo = file(fr_file)
     contents = fo.read()
     fo.close()
 
     self.failUnless(contents.find('<p>') != -1)  # should contain the markup
     self.failUnless(contents.find('Hello!') == -1)  # should be translated
 
 
+  def testSubstitutionHtml(self):
+    input_file = util.PathFromRoot('grit/testdata/toolbar_about.html')
+    root = grd_reader.Parse(StringIO.StringIO('''<?xml version="1.0" encoding="UTF-8"?>
+      <grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
+        <release seq="1" allow_pseudo="False">
+          <structures fallback_to_english="True">
+            <structure type="tr_html" name="IDR_HTML" file="%s" expand_variables="true"/>
+          </structures>
+        </release>
+      </grit>
+      ''' % input_file), util.PathFromRoot('.'))
+    util.FixRootForUnittest(root, '.')
+    root.SetOutputContext('ar', {})
+    # We must run the gatherers since we'll be wanting the translation of the
+    # file.  The file exists in the location pointed to.
+    root.RunGatherers(recursive=True)
+
+    output_dir = tempfile.gettempdir()
+    import grit.node.structure
+    structure = root.GetChildrenOfType(grit.node.structure.StructureNode)[0]
+    ar_file = structure.FileForLanguage('ar', output_dir)
+    self.failUnless(ar_file == os.path.join(output_dir,
+                                            'ar_toolbar_about.html'))
+
+    fo = file(ar_file)
+    contents = fo.read()
+    fo.close()
+
+    self.failUnless(contents.find('dir="RTL"') != -1)
+
+
   def testFallbackToEnglish(self):
     root = grd_reader.Parse(StringIO.StringIO('''<?xml version="1.0" encoding="UTF-8"?>
       <grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
         <release seq="1" allow_pseudo="False">
           <structures fallback_to_english="True">
             <structure type="dialog" name="IDD_ABOUTBOX" file="grit\\testdata\klonk.rc" encoding="utf-16" />
           </structures>
         </release>
       </grit>'''), util.PathFromRoot('.'))
     util.FixRootForUnittest(root)
+    root.SetOutputContext('en', {})
     root.RunGatherers(recursive = True)
 
     node = root.GetNodeById("IDD_ABOUTBOX")
     formatter = node.ItemFormatter('rc_all')
     output = formatter.Format(node, 'bingobongo')
     self.failUnless(output.strip() == '''IDD_ABOUTBOX DIALOGEX 22, 17, 230, 75
 STYLE DS_SETFONT | DS_MODALFRAME | WS_CAPTION | WS_SYSMENU
 CAPTION "About"
 FONT 8, "System", 0, 0, 0x0
 BEGIN
     ICON            IDI_KLONK,IDC_MYICON,14,9,20,20
     LTEXT           "klonk Version ""yibbee"" 1.0",IDC_STATIC,49,10,119,8,
                     SS_NOPREFIX
     LTEXT           "Copyright (C) 2005",IDC_STATIC,49,20,119,8
     DEFPUSHBUTTON   "OK",IDOK,195,6,30,11,WS_GROUP
     CONTROL         "Jack ""Black"" Daniels",IDC_RADIO1,"Button",
                     BS_AUTORADIOBUTTON,46,51,84,10
 END''')
 
 
+  def testSubstitutionRc(self):
+    root = grd_reader.Parse(StringIO.StringIO('''<?xml version="1.0" encoding="UTF-8"?>
+    <grit latest_public_release="2" source_lang_id="en-US" current_release="3"
+        base_dir=".">
+        <outputs>
+          <output lang="en" type="rc_all" filename="grit\\testdata\klonk_resources.rc"/>
+        </outputs>
+
+      <release seq="1" allow_pseudo="False">
+        <structures>
+          <structure type="menu" name="IDC_KLONKMENU"
+              file="grit\\testdata\klonk.rc" encoding="utf-16"
+              expand_variables="true" />
+        </structures>
+        <messages>
+          <message name="good" sub_variable="true">
+            excellent
+          </message>
+        </messages>
+      </release>
+    </grit>
+    '''), util.PathFromRoot('.'))
+    util.FixRootForUnittest(root)
+    root.SetOutputContext('en', {})
+    root.RunGatherers(recursive=True)
+
+    buf = StringIO.StringIO()
+    build.RcBuilder.ProcessNode(root, DummyOutput('rc_all', 'en'), buf)
+    output = buf.getvalue()
+    self.failUnless(output.strip() == '''
+// Copyright (c) Google Inc. 2012
+// All rights reserved.
+// This file is automatically generated by GRIT.  Do not edit.
+
+#include "resource.h"
+#include <winresrc.h>
+#ifdef IDC_STATIC
+#undef IDC_STATIC
+#endif
+#define IDC_STATIC (-1)
+
+LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL
+
+
+IDC_KLONKMENU MENU
+BEGIN
+    POPUP "&File"
+    BEGIN
+        MENUITEM "E&xit",                       IDM_EXIT
+        MENUITEM "This be ""Klonk"" me like",   ID_FILE_THISBE
+        POPUP "gonk"
+        BEGIN
+            MENUITEM "Klonk && is excellent",           ID_GONK_KLONKIS
+        END
+    END
+    POPUP "&Help"
+    BEGIN
+        MENUITEM "&About ...",                  IDM_ABOUT
+    END
+END
+
+STRINGTABLE
+BEGIN
+  good            "excellent"
+END
+'''.strip())
+
+
 if __name__ == '__main__':
   unittest.main()