Function TestLoggingProcessAdapterTrailingChunk

pkg/logging/logging_test.go:151–195

TestLoggingProcessAdapterTrailingChunk verifies that the logger forwards all of the container's output, including a final chunk that has no trailing newline, rather than holding that chunk back until something closes the stream. The container's stdio FIFOs are modelled with os.Pipe; closing the writ

(t *testing.T)

// write end models the container exiting and containerd closing the FIFO.
// Regression test for https://github.com/containerd/nerdctl/issues/5006
func TestLoggingProcessAdapterTrailingChunk(t *testing.T) {
	const expected = "'Hello World!\nThere is no newline'"

	stdoutR, stdoutW, err := os.Pipe()
	if err != nil {
		t.Fatal(err)
	}
	defer stdoutR.Close()
	stderrR, stderrW, err := os.Pipe()
	if err != nil {
		t.Fatal(err)
	}
	defer stderrR.Close()

	driver := &MockDriver{}
	config := &logging.Config{
		Stdout: stdoutR,
		Stderr: stderrR,
	}

	// Write the container's output, including a trailing chunk without a newline,
	// then close the write ends to model the container exiting.
	if _, err := stdoutW.WriteString(expected); err != nil {
		t.Fatal(err)
	}
	stdoutW.Close()
	stderrW.Close()

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	// getContainerWait never reports an exit here: completion is driven by the
	// FIFOs reaching EOF, as it usually is in practice.
	var getContainerWaitMock ContainerWaitFunc = func(ctx context.Context, address string, config *logging.Config, outputSeen func() bool) (<-chan containerd.ExitStatus, error) {
		return make(chan containerd.ExitStatus), nil
	}

	if err := loggingProcessAdapter(ctx, driver, "testDataStore", "", getContainerWaitMock, config); err != nil {
		t.Fatal(err)
	}

	if actual := strings.Join(driver.receivedStdout, ""); actual != expected {
		t.Fatalf("stdout is %q, expected %q", actual, expected)
	}
}

// TestLoggingProcessAdapterSyncTrailingChunk verifies the same trailing-chunk
// behaviour for a driver that writes synchronously (SyncDriver), which is the