Mukan Erkin Törük
26b204bd04
Some checks are pending
Docs Deploy / build_and_deploy (push) Waiting to run
Generate Docs / cli (push) Waiting to run
Generate Config Doc / cli (push) Waiting to run
Go formatting / go-formatting (push) Waiting to run
Check links / markdown-link-check (push) Waiting to run
Integration / pre-test (push) Waiting to run
Integration / test on (push) Blocked by required conditions
Integration / status (push) Blocked by required conditions
Lint / Lint Go code (push) Waiting to run
Test / test (ubuntu-latest) (push) Waiting to run
40 lines
939 B
Go
40 lines
939 B
Go
package truncatedbuffer
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWriter(t *testing.T) {
|
|
ranBytes10 := make([]byte, 10)
|
|
_, err := rand.Read(ranBytes10)
|
|
require.NoError(t, err)
|
|
|
|
ranBytes1000 := make([]byte, 1000)
|
|
_, err = rand.Read(ranBytes1000)
|
|
require.NoError(t, err)
|
|
|
|
// TruncatedBuffer has a max capacity
|
|
b := NewTruncatedBuffer(100)
|
|
|
|
require.Equal(t, 100, b.GetCap())
|
|
|
|
n, err := b.Write(ranBytes10)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 10, n)
|
|
require.Equal(t, ranBytes10, b.GetBuffer().Bytes())
|
|
|
|
n, err = b.Write(ranBytes1000)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1000, n)
|
|
require.Equal(t, append(ranBytes10, ranBytes1000[:90]...), b.GetBuffer().Bytes())
|
|
|
|
// TruncatedBuffer has no max capacity
|
|
b = NewTruncatedBuffer(0)
|
|
n, err = b.Write(ranBytes1000)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1000, n)
|
|
require.Equal(t, ranBytes1000, b.GetBuffer().Bytes())
|
|
}
|