mirror of
https://github.com/jesseduffield/lazydocker.git
synced 2026-07-21 23:01:03 +00:00
19 lines
368 B
Go
19 lines
368 B
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestOnceWriter(t *testing.T) {
|
|
innerWriter := bytes.NewBuffer(nil)
|
|
counter := 0
|
|
onceWriter := NewOnceWriter(innerWriter, func() {
|
|
counter += 1
|
|
})
|
|
_, _ = onceWriter.Write([]byte("hello"))
|
|
_, _ = onceWriter.Write([]byte("hello"))
|
|
if counter != 1 {
|
|
t.Errorf("expected counter to be 1, got %d", counter)
|
|
}
|
|
}
|