42 lines
844 B
Go
42 lines
844 B
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information
|
||
|
|
|
||
|
|
package sync2_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"crypto/rand"
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"storj.io/storj/internal/memory"
|
||
|
|
"storj.io/storj/internal/sync2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCopy(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
r := io.LimitReader(rand.Reader, 32*memory.KB.Int64())
|
||
|
|
|
||
|
|
n, err := sync2.Copy(ctx, ioutil.Discard, r)
|
||
|
|
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, n, 32*memory.KB.Int64())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCopy_Cancel(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
cancel()
|
||
|
|
|
||
|
|
r := io.LimitReader(rand.Reader, 32*memory.KB.Int64())
|
||
|
|
|
||
|
|
n, err := sync2.Copy(ctx, ioutil.Discard, r)
|
||
|
|
|
||
|
|
assert.EqualError(t, err, context.Canceled.Error())
|
||
|
|
assert.EqualValues(t, n, 0)
|
||
|
|
}
|