From 4399f2bae67c3c2c0b242af015ddeee4ac638bfe Mon Sep 17 00:00:00 2001
From: Michael Carroll <mjcarroll@intrinsic.ai>
Date: Wed, 15 Jan 2025 16:25:10 +0000
Subject: [PATCH] Add test for --root-dir argument

Signed-off-by: Michael Carroll <mjcarroll@intrinsic.ai>
---
 test/test_xacro.py | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/test/test_xacro.py b/test/test_xacro.py
index 2a7c89c..5f0ea41 100644
--- a/test/test_xacro.py
+++ b/test/test_xacro.py
@@ -1141,6 +1141,46 @@ def test_create_subdirs(self):
 
         self.assertTrue(output_file_created)
 
+    def test_set_root_directory(self):
+        # Run xacro in one directory, but specify the directory to resolve
+        # filenames in.
+        tmp_dir_name = tempfile.mkdtemp()  # create directory we can trash
+
+        output_path = os.path.join(tmp_dir_name, "out")
+
+        # Generate a pair of files to be parsed by xacro, outside of the
+        # test directory, to ensure the root-dir argument works
+        file_foo = os.path.join(tmp_dir_name, 'foo.xml.xacro')
+        file_bar = os.path.join(tmp_dir_name, 'bar.xml.xacro')
+        with open(file_foo, 'w') as f:
+            f.write('''<?xml version="1.0"?>
+<robot xmlns:xacro="http://ros.org/wiki/xacro">
+	<xacro:include filename="bar.xml.xacro"/>
+</robot>
+''')
+        with open(file_bar, 'w') as f:
+            f.write('''<?xml version="1.0"?>
+<robot xmlns:xacro="http://ros.org/wiki/xacro">
+    <link name="my_link"/>
+</robot>
+''')
+
+        # Run xacro with no --root-dir arg, which will then use the
+        # current path as the path to resolveto
+        self.run_xacro('foo.xml.xacro', '-o', output_path)
+        output_file_created = os.path.isfile(output_path)
+        self.assertFalse(output_file_created)
+
+        # Run xacro with --root-dir arg set to the new temp directory
+        self.run_xacro('foo.xml.xacro',
+                       '--root-dir', tmp_dir_name,
+                       '-o', output_path)
+
+        output_file_created = os.path.isfile(output_path)
+        shutil.rmtree(tmp_dir_name)  # clean up after ourselves
+
+        self.assertTrue(output_file_created)
+
     def test_iterable_literals_plain(self):
         self.assert_matches(self.quick_xacro('''
 <a xmlns:xacro="http://www.ros.org/wiki/xacro">