Fix serverless output draining regressions

1 files changed, 14 insertions, 0 deletions

diff --git a/internal/clients/session_spec_test.go b/internal/clients/session_spec_test.go
index aa3c45d..8133bc9 100644
--- a/internal/clients/session_spec_test.go
+++ b/internal/clients/session_spec_test.go
@@ -131,3 +131,17 @@ func TestNewSessionSpecSplitsFiles(t *testing.T) {
 		t.Fatalf("unexpected timeout: %d", spec.Timeout)
 	}
 }
+
+func TestNewSessionSpecUsesPipeSentinelForServerlessStdin(t *testing.T) {
+	t.Parallel()
+
+	spec := NewSessionSpec(config.Args{
+		Mode:       omode.GrepClient,
+		Serverless: true,
+		RegexStr:   "ERROR",
+	})
+
+	if len(spec.Files) != 1 || spec.Files[0] != "-" {
+		t.Fatalf("unexpected files for serverless stdin: %#v", spec.Files)
+	}
+}