Make first_ids_file an optional attribute of the .grd file.

grit/grd_reader_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 grd_reader package'''
 
 import os
 import sys
 if __name__ == '__main__':
@@ skipping 80 matching lines @@
   </release>
 </grit>'''
     pseudo_file = StringIO.StringIO(input)
     tree = grd_reader.Parse(pseudo_file, '.')
 
     greeting = tree.GetNodeById('IDS_GREETING')
     self.failUnless(greeting.GetCliques()[0].GetMessage().GetRealContent() ==
                     'This is a very long line with no linebreaks yes yes it '
                     'stretches on and on and on!')
 
-  def testAssignFirstIds(self):
+  def doTestAssignFirstIds(self, first_ids_path):
     input = u'''<?xml version="1.0" encoding="UTF-8"?>
-<grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
+<grit latest_public_release="2" source_lang_id="en-US" current_release="3"
+      base_dir="." first_ids_file="%s">
   <release seq="3">
     <messages>
       <message name="IDS_TEST" desc="test">
         test
       </message>
     </messages>
   </release>
-</grit>'''
+</grit>''' % first_ids_path
     pseudo_file = StringIO.StringIO(input)
-    root = grd_reader.Parse(pseudo_file, '.')
     grit_root_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)),
                                  '..')
-    root.AssignFirstIds(
-        os.path.join(grit_root_dir,
-                     "grit/testdata/chrome/app/generated_resources.grd"),
-        os.path.join(grit_root_dir, "grit/testdata/tools/grit/resource_ids"),
-        {})
+    fake_input_path = os.path.join(
+        grit_root_dir, "grit/testdata/chrome/app/generated_resources.grd")
+    root = grd_reader.Parse(pseudo_file, os.path.split(fake_input_path)[0])
+    root.AssignFirstIds(fake_input_path, {})
     messages_node = root.children[0].children[0]
     self.failUnless(isinstance(messages_node, empty.MessagesNode))
     self.failUnless(messages_node.attrs["first_id"] !=
         empty.MessagesNode().DefaultAttributes()["first_id"])
 
+  def testAssignFirstIds(self):
+    self.doTestAssignFirstIds("../../tools/grit/resource_ids")
+
+  def testAssignFirstIdsUseGritDir(self):
+    self.doTestAssignFirstIds("GRIT_DIR/grit/testdata/tools/grit/resource_ids")
+
   def testAssignFirstIdsMultipleMessages(self):
     """If there are multiple messages sections, the resource_ids file
     needs to list multiple first_id values."""
     input = u'''<?xml version="1.0" encoding="UTF-8"?>
-<grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
+<grit latest_public_release="2" source_lang_id="en-US" current_release="3"
+      base_dir="." first_ids_file="resource_ids">
   <release seq="3">
     <messages>
       <message name="IDS_TEST" desc="test">
         test
       </message>
     </messages>
     <messages>
       <message name="IDS_TEST2" desc="test">
         test2
       </message>
     </messages>
   </release>
 </grit>'''
     pseudo_file = StringIO.StringIO(input)
-    root = grd_reader.Parse(pseudo_file, '.')
     grit_root_dir = os.path.join(os.path.abspath(os.path.dirname(__file__)),
                                  '..')
-    root.AssignFirstIds(
-        # This file does not actually exist, but must use absolute
-        # path as resource_ids has '' for SRCDIR (see special handling
-        # in grit.node.misc).
-        os.path.join(grit_root_dir, "grit/testdata/test.grd"),
-        os.path.join(grit_root_dir, "grit/testdata/resource_ids"),
-        {})
+    fake_input_path = os.path.join(grit_root_dir, "grit/testdata/test.grd")
+
+    root = grd_reader.Parse(pseudo_file, os.path.split(fake_input_path)[0])
+    root.AssignFirstIds(fake_input_path, {})
     messages_node = root.children[0].children[0]
     self.assertTrue(isinstance(messages_node, empty.MessagesNode))
     self.assertEqual('100', messages_node.attrs["first_id"])
     messages_node = root.children[0].children[1]
     self.assertTrue(isinstance(messages_node, empty.MessagesNode))
     self.assertEqual('10000', messages_node.attrs["first_id"])
 
   def testUseNameForIdAndPpIfdef(self):
     input = u'''<?xml version="1.0" encoding="UTF-8"?>
 <grit latest_public_release="2" source_lang_id="en-US" current_release="3" base_dir=".">
@@ skipping 10 matching lines @@
     pseudo_file = StringIO.StringIO(input)
     root = grd_reader.Parse(pseudo_file, '.', defines={'hello': '1'})
 
     # Check if the ID is set to the name. In the past, there was a bug
     # that caused the ID to be a generated number.
     hello = root.GetNodeById('IDS_HELLO')
     self.failUnless(hello.GetCliques()[0].GetId() == 'IDS_HELLO')
 
 if __name__ == '__main__':
   unittest.main()