37 lines
681 B
Go
37 lines
681 B
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"golang.org/x/sync/errgroup"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPrefixWriter(t *testing.T) {
|
||
|
|
root := NewPrefixWriter("", ioutil.Discard)
|
||
|
|
alpha := root.Prefixed("alpha")
|
||
|
|
beta := root.Prefixed("beta")
|
||
|
|
|
||
|
|
var group errgroup.Group
|
||
|
|
defer func() {
|
||
|
|
require.NoError(t, group.Wait())
|
||
|
|
}()
|
||
|
|
|
||
|
|
group.Go(func() error {
|
||
|
|
_, err := alpha.Write([]byte{1, 2, 3})
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
group.Go(func() error {
|
||
|
|
_, err := alpha.Write([]byte{3, 2, 1})
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
group.Go(func() error {
|
||
|
|
_, err := beta.Write([]byte{1, 2, 3})
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
}
|