| // Copyright 2022 The Go Authors. All rights reserved. | |
| // Use of this source code is governed by a BSD-style | |
| // license that can be found in the LICENSE file. | |
| package buffer | |
| import ( | |
| "internal/race" | |
| "internal/testenv" | |
| "testing" | |
| ) | |
| func Test(t *testing.T) { | |
| b := New() | |
| defer b.Free() | |
| b.WriteString("hello") | |
| b.WriteByte(',') | |
| b.Write([]byte(" world")) | |
| got := b.String() | |
| want := "hello, world" | |
| if got != want { | |
| t.Errorf("got %q, want %q", got, want) | |
| } | |
| } | |
| func TestAlloc(t *testing.T) { | |
| if race.Enabled { | |
| t.Skip("skipping test in race mode") | |
| } | |
| testenv.SkipIfOptimizationOff(t) | |
| got := int(testing.AllocsPerRun(5, func() { | |
| b := New() | |
| defer b.Free() | |
| b.WriteString("not 1K worth of bytes") | |
| })) | |
| if got != 0 { | |
| t.Errorf("got %d allocs, want 0", got) | |
| } | |
| } | |