Page MenuHomePhorge
Files F114213

No OneTemporary
Actions

Size
4 KB
Referenced Files
None
Subscribers
None

View Options

diff --git a/test/exile_test.exs b/test/exile_test.exs
index c8dd4ae..a75222a 100644
--- a/test/exile_test.exs
+++ b/test/exile_test.exs
@@ -1,123 +1,149 @@
defmodule ExileTest do
use ExUnit.Case
+ @write_stderr Path.join(__DIR__, "scripts/write_stderr.sh")
+
doctest Exile
test "stream with enumerable" do
proc_stream =
Exile.stream!(["cat"], input: Stream.map(1..1000, fn _ -> "a" end), stderr: :console)
stdout = proc_stream |> Enum.to_list()
assert IO.iodata_length(stdout) == 1000
end
test "stream with collectable" do
proc_stream =
Exile.stream!(["cat"], input: fn sink -> Enum.into(1..1000, sink, fn _ -> "a" end) end)
stdout = Enum.to_list(proc_stream)
assert IO.iodata_length(stdout) == 1000
end
test "stream without stdin" do
proc_stream = Exile.stream!(~w(echo hello))
stdout = Enum.to_list(proc_stream)
assert IO.iodata_to_binary(stdout) == "hello\n"
end
- test "stderr" do
- proc_stream = Exile.stream!(["sh", "-c", "echo foo >>/dev/stderr"], stderr: :consume)
+ test "stderr to console" do
+ {output, exit_status} = run_in_shell([@write_stderr, "Hello World"], stderr: :console)
+ assert output == "Hello World\n"
+ assert exit_status == 0
+ end
+
+ test "stderr disabled" do
+ {output, exit_status} = run_in_shell([@write_stderr, "Hello World"], stderr: :disable)
+ assert output == ""
+ assert exit_status == 0
+ end
+
+ test "stderr consume" do
+ proc_stream = Exile.stream!([fixture("write_stderr.sh"), "Hello World"], stderr: :consume)
assert {[], stderr} = split_stream(proc_stream)
- assert IO.iodata_to_binary(stderr) == "foo\n"
+ assert IO.iodata_to_binary(stderr) == "Hello World\n"
end
test "multiple streams" do
script = """
for i in {1..1000}; do
echo "foo ${i}"
echo "bar ${i}" >&2
done
"""
proc_stream = Exile.stream!(["sh", "-c", script], stderr: :consume)
{stdout, stderr} = split_stream(proc_stream)
stdout_lines = String.split(Enum.join(stdout), "\n", trim: true)
stderr_lines = String.split(Enum.join(stderr), "\n", trim: true)
assert length(stdout_lines) == length(stderr_lines)
assert Enum.all?(stdout_lines, &String.starts_with?(&1, "foo "))
assert Enum.all?(stderr_lines, &String.starts_with?(&1, "bar "))
end
test "environment variable" do
output =
Exile.stream!(~w(printenv FOO), env: %{"FOO" => "bar"})
|> Enum.to_list()
|> IO.iodata_to_binary()
assert output == "bar\n"
end
test "premature stream termination" do
input_stream = Stream.map(1..100_000, fn _ -> "hello" end)
assert_raise Exile.Stream.AbnormalExit,
"program exited due to :epipe error",
fn ->
Exile.stream!(~w(cat), input: input_stream)
|> Enum.take(1)
end
end
test "premature stream termination when ignore_epipe is true" do
input_stream = Stream.map(1..100_000, fn _ -> "hello" end)
assert ["hello"] ==
Exile.stream!(~w(cat), input: input_stream, ignore_epipe: true, max_chunk_size: 5)
|> Enum.take(1)
end
test "stream!/2 with exit status" do
proc_stream = Exile.stream!(["sh", "-c", "exit 10"])
assert_raise Exile.Stream.AbnormalExit, "program exited with exit status: 10", fn ->
Enum.to_list(proc_stream)
end
end
test "stream/2 with exit status" do
proc_stream = Exile.stream(["sh", "-c", "exit 10"])
stdout = Enum.to_list(proc_stream)
assert stdout == [{:exit, {:status, 10}}]
end
test "stream!/2 abnormal exit status" do
proc_stream = Exile.stream!(["sh", "-c", "exit 5"])
exit_status =
try do
proc_stream
|> Enum.to_list()
nil
rescue
e in Exile.Stream.AbnormalExit ->
e.exit_status
end
assert exit_status == 5
end
defp split_stream(stream) do
{stdout, stderr} =
Enum.reduce(stream, {[], []}, fn
{:stdout, data}, {stdout, stderr} -> {[data | stdout], stderr}
{:stderr, data}, {stdout, stderr} -> {stdout, [data | stderr]}
end)
{Enum.reverse(stdout), Enum.reverse(stderr)}
end
+
+ defp fixture(script) do
+ Path.join([__DIR__, "scripts", script])
+ end
+
+ # runs the given code in a separate mix shell and captures all the
+ # output written to the shell during the execution the output can be
+ # from the elixir or from the spawned command
+ defp run_in_shell(args, opts) do
+ expr = ~s{Exile.stream!(#{inspect(args)}, #{inspect(opts)}) |> Enum.to_list()}
+ {_output, _exit_status} = System.shell("mix run -e '#{expr}'", stderr_to_stdout: true)
+ end
end
diff --git a/test/scripts/write_stderr.sh b/test/scripts/write_stderr.sh
new file mode 100755
index 0000000..9658da0
--- /dev/null
+++ b/test/scripts/write_stderr.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+echo "$1" >> /dev/stderr

File Metadata

Mime Type
text/x-diff
Expires
Mon, Nov 25, 11:21 PM (1 d, 8 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
40109
Default Alt Text
(4 KB)

Event Timeline