Add InputSource.FromParent (#21)

pgreze · Jun 27, 2024 · 8f833c4 · 8f833c4
1 parent 3f0fb93
commit 8f833c4
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 12 deletions.
diff --git a/src/main/kotlin/com/github/pgreze/process/InputSource.kt b/src/main/kotlin/com/github/pgreze/process/InputSource.kt
@@ -18,6 +18,12 @@ sealed class InputSource {
      */
     class FromStream(val handler: suspend (OutputStream) -> Unit) : InputSource()
 
+    /**
+     * Natively supported parent provided redirection.
+     * @see ProcessBuilder.Redirect.INHERIT
+     */
+    object FromParent : InputSource()
+
     @Suppress("BlockingMethodInNonBlockingContext")
     companion object {
         @JvmStatic
@@ -41,4 +47,5 @@ sealed class InputSource {
 internal fun InputSource.toNative() = when (this) {
     is InputSource.FromFile -> ProcessBuilder.Redirect.from(file)
     is InputSource.FromStream -> ProcessBuilder.Redirect.PIPE
+    is InputSource.FromParent -> ProcessBuilder.Redirect.INHERIT
 }
diff --git a/src/test/kotlin/com/github/pgreze/process/InputSourceTest.kt b/src/test/kotlin/com/github/pgreze/process/InputSourceTest.kt
@@ -23,40 +23,50 @@ class InputSourceTest {
     }
 
     @Test
-    fun fromString() = runSuspendTest {
+    fun fromFile(@TempDir dir: Path) = runSuspendTest {
+        val input = dir.resolve("input.txt").toFile()
+        input.writeText(STRING)
+
         val output = process(
             "cat",
-            stdin = InputSource.fromString(STRING),
-            stdout = Redirect.CAPTURE
+            stdin = InputSource.FromFile(input),
+            stdout = Redirect.CAPTURE,
         ).unwrap()
+
         output shouldBeEqualTo listOf(STRING)
     }
 
     @Test
-    fun fromFile(@TempDir dir: Path) = runSuspendTest {
-        val input = dir.resolve("input.txt").toFile()
-        input.writeText(STRING)
-
+    fun fromString() = runSuspendTest {
         val output = process(
             "cat",
-            stdin = InputSource.FromFile(input),
+            stdin = InputSource.fromString(STRING),
             stdout = Redirect.CAPTURE,
         ).unwrap()
-
         output shouldBeEqualTo listOf(STRING)
     }
 
     @Test
-    fun fromStream() = runSuspendTest {
+    fun fromInputStream() = runSuspendTest {
         val inputStream = ByteArrayInputStream(STRING.toByteArray())
         val output = process(
             "cat",
             stdin = InputSource.fromInputStream(inputStream),
-            stdout = Redirect.CAPTURE
+            stdout = Redirect.CAPTURE,
         ).unwrap()
         output shouldBeEqualTo listOf(STRING)
     }
 
+    @Test
+    fun fromParent() = runSuspendTest {
+        val output = process(
+            "cat",
+            stdin = InputSource.FromParent,
+            stdout = Redirect.CAPTURE,
+        ).unwrap()
+        output shouldBeEqualTo listOf()
+    }
+
     @Nested
     @DisplayName("ensure that input and output are concurrently processed")
     inner class AsyncStreams {
@@ -77,7 +87,7 @@ class InputSourceTest {
                         }
                     },
                     stdout = Redirect.CAPTURE,
-                    consumer = consumer::add
+                    consumer = consumer::add,
                 )
             }