Fix _processIncludes to take StringIO argument

armi/utils/tests/resources/lower/includeA.yaml

@@ -1,4 +1,4 @@
 full_name: Jennifer Person
 # some comment in includeA
-children:
-    !include includeB.yaml
+children: 
+  !include includeB.yaml

armi/utils/tests/resources/root.yaml

@@ -1,10 +1,10 @@
 # Behold, the Person family
 bobby: &bobby
-    full_name: Robert Person
+  full_name: Robert Person
 
 billy:
-    full_name: William Person
-    # comment
-    children:
-        - *bobby
-        - !include lower/includeA.yaml
+  full_name: William Person
+  # comment
+  children:
+  - *bobby
+  - !include lower/includeA.yaml

armi/utils/tests/test_textProcessors.py

@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Tests for functions in textProcessors.py."""
+from io import StringIO, TextIOWrapper, TextIOBase
 import os
 import pathlib
-import unittest
-
 import ruamel
+import unittest
 
 from armi.utils import textProcessors
 
@@ -60,6 +60,23 @@ def test_resolveIncludes(self):
         self.assertTrue(commentFound)
         self.assertTrue(anchorFound)
 
+    def test_resolveIncludes_StringIO(self):
+        """Tests that resolveMarkupInclusions handles StringIO input"""
+        yaml = ruamel.yaml.YAML()
+        with open(os.path.join(RES_DIR, "root.yaml")) as f:
+            loadedYaml = yaml.load(f)
+        stringIO = StringIO()
+        yaml.dump(loadedYaml, stringIO)
+        resolved = textProcessors.resolveMarkupInclusions(
+            src=stringIO, root=pathlib.Path(RES_DIR)
+        )
+        with open(os.path.join(RES_DIR, "root.yaml")) as f:
+            expected = textProcessors.resolveMarkupInclusions(
+                f, root=pathlib.Path(RES_DIR)
+            )
+        # strip it because one method gives an extra newline we don't care about
+        self.assertEqual(resolved.getvalue().strip(), expected.getvalue().strip())
+
     def test_findIncludes(self):
         includes = textProcessors.findYamlInclusions(
             pathlib.Path(RES_DIR) / "root.yaml"

armi/utils/textProcessors.py

@@ -58,10 +58,10 @@ def __str__(self):
 
 
 def _processIncludes(
-    src,
+    src: Union[TextIO, pathlib.Path],
     out,
     includes: List[Tuple[pathlib.Path, FileMark]],
-    root,
+    root: pathlib.Path,
     indentation=0,
     currentFile="<stream>",
 ):
@@ -88,7 +88,13 @@ def _beginningOfContent(line: str) -> int:
             return 0
 
     indentSpace = " " * indentation
-    for i, line in enumerate(src.readlines()):
+    if hasattr(src, "getvalue"):
+        # assume stringIO
+        lines = [l + "\n" for l in src.getvalue().split("\n")]
+    else:
+        # assume file stream or TextIOBase, and it has a readlines attr
+        lines = src.readlines()
+    for i, line in enumerate(lines):
         leadingSpace = indentSpace if i > 0 else ""
         m = _INCLUDE_RE.match(line)
         if m:
@@ -134,7 +140,7 @@ def resolveMarkupInclusions(
 
     Parameters
     ----------
-    src : TextIOBase or Path
+    src : StringIO or TextIOBase/Path
         If a Path is provided, read text from there. If is stream is provided, consume
         text from the stream. If a stream is provided, ``root`` must also be provided.
     root : Optional Path

doc/release/0.2.rst

@@ -15,6 +15,7 @@ What's new in ARMI
 
 Bug fixes
 ---------
+#. Fix _processIncludes to handle StringIO input (`PR#1333 <https://github.com/terrapower/armi/pull/1333>`_)
 #. TBD
 
 ARMI v0.2.8