Codebase list golang-github-vbauerster-mpb / b65e0c2 cwriter / writer_posix_test.go
b65e0c2

Tree @b65e0c2 (Download .tar.gz)

writer_posix_test.go @b65e0c2raw · history · blame

// +build !windows

package cwriter_test

import (
	"bytes"
	"testing"

	. "github.com/vbauerster/mpb/cwriter"
)

// TestWriterPosix by writing and flushing many times. The output buffer
// must contain the clearCursor and clearLine sequences.
func TestWriterPosix(t *testing.T) {
	out := new(bytes.Buffer)
	w := New(out)

	for _, tcase := range []struct {
		input, expectedOutput string
	}{
		{input: "foo\n", expectedOutput: "foo\n"},
		{input: "bar\n", expectedOutput: "foo\n" + ClearCursorAndLine + "bar\n"},
		{input: "fizz\n", expectedOutput: "foo\n" + ClearCursorAndLine + "bar\n" + ClearCursorAndLine + "fizz\n"},
	} {
		t.Run(tcase.input, func(t *testing.T) {
			s := []byte(tcase.input)
			lc := bytes.Count(s, []byte("\n"))
			w.Write(s)
			w.Flush(lc)
			output := out.String()
			if output != tcase.expectedOutput {
				t.Fatalf("want %q, got %q", tcase.expectedOutput, output)
			}
		})
	}
}