If-then-else support for GRIT

grit/node/misc_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 misc.GritNode'''
 
 
 import os
 import sys
@@ skipping 83 matching lines @@
     self.failUnless(hello_message not in active)
     self.failUnless(french_message not in active)
 
     grd.SetOutputLanguage('en')
     grd.SetDefines({'FORCE_FRENCH': '1', 'bingo': '1'})
     active = set(grd.ActiveDescendants())
     self.failUnless(bingo_message in active)
     self.failUnless(hello_message not in active)
     self.failUnless(french_message in active)
 
+  def testElsiness(self):
+    grd = util.ParseGrdForUnittest('''
+        <messages>
+          <if expr="True">
+            <then> <message name="IDS_YES1"></message> </then>
+            <else> <message name="IDS_NO1"></message> </else>
+          </if>
+          <if expr="True">
+            <then> <message name="IDS_YES2"></message> </then>
+            <else> </else>
+          </if>
+          <if expr="True">
+            <then> </then>
+            <else> <message name="IDS_NO2"></message> </else>
+          </if>
+          <if expr="True">
+            <then> </then>
+            <else> </else>
+          </if>
+          <if expr="False">
+            <then> <message name="IDS_NO3"></message> </then>
+            <else> <message name="IDS_YES3"></message> </else>
+          </if>
+          <if expr="False">
+            <then> <message name="IDS_NO4"></message> </then>
+            <else> </else>
+          </if>
+          <if expr="False">
+            <then> </then>
+            <else> <message name="IDS_YES4"></message> </else>
+          </if>
+          <if expr="False">
+            <then> </then>
+            <else> </else>
+          </if>
+        </messages>''')
+    included = [msg.attrs['name'] for msg in grd.ActiveDescendants()
+                                  if msg.name == 'message']
+    self.assertEqual(['IDS_YES1', 'IDS_YES2', 'IDS_YES3', 'IDS_YES4'], included)
+
   def testIffynessWithOutputNodes(self):
     grd = grd_reader.Parse(StringIO.StringIO('''
       <grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
         <outputs>
           <output filename="uncond1.rc" type="rc_data" />
           <if expr="lang == 'fr' or 'hello' in defs">
             <output filename="only_fr.adm" type="adm" />
             <output filename="only_fr.plist" type="plist" />
           </if>
           <if expr="lang == 'ru'">
@@ skipping 44 matching lines @@
     grd.SetOutputLanguage('en')
     grd.SetDefines({'bingo': '1'})
     outputs = [output.GetFilename() for output in grd.GetOutputFiles()]
     self.assertEquals(outputs, ['uncond1.rc', 'uncond2.adm', 'iftest.h'])
 
     grd.SetOutputLanguage('fr')
     grd.SetDefines({'bingo': '1'})
     outputs = [output.GetFilename() for output in grd.GetOutputFiles()]
     self.assertNotEquals(outputs, ['uncond1.rc', 'uncond2.adm', 'iftest.h'])
 
-
-class IfNodeChildrenUnittest(unittest.TestCase):
   def testChildrenAccepted(self):
     grd = grd_reader.Parse(StringIO.StringIO('''<?xml version="1.0"?>
       <grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
         <release seq="3">
           <includes>
             <if expr="'bingo' in defs">
               <include type="gif" name="ID_LOGO2" file="images/logo2.gif" />
             </if>
             <if expr="'bingo' in defs">
               <if expr="'hello' in defs">
@@ skipping 177 matching lines @@
     # a pseudo language should fail, but they do not fail and the test was
     # broken and failed to catch it. Fix this.
 
     # Should not raise an exception since pseudo is allowed
     rc.FormatMessage(bingo, 'xyz-pseudo')
     rc.FormatStructure(menu, 'xyz-pseudo', '.')
 
 
 if __name__ == '__main__':
   unittest.main()