Move files from gousb/usb to gousb
This commit is contained in:
Sebastian Zagrodzki
249
transfer_stream_test.go
Normal file
249
transfer_stream_test.go
Normal file
@@ -0,0 +1,249 @@
|
||||
// Copyright 2017 the gousb Authors. All rights reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gousb
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var fakeTransferBuf = make([]byte, 1500)
|
||||
|
||||
type fakeStreamResult struct {
|
||||
n int
|
||||
waitErr error
|
||||
submitErr error
|
||||
}
|
||||
|
||||
type fakeStreamTransfer struct {
|
||||
res []fakeStreamResult
|
||||
inFlight bool
|
||||
released bool
|
||||
}
|
||||
|
||||
func (f *fakeStreamTransfer) submit() error {
|
||||
if f.released {
|
||||
return errors.New("submit() called on a free()d transfer")
|
||||
}
|
||||
if f.inFlight {
|
||||
return errors.New("submit() called twice")
|
||||
}
|
||||
if len(f.res) == 0 {
|
||||
return errors.New("submit() called but fake result missing")
|
||||
}
|
||||
f.inFlight = true
|
||||
res := f.res[0]
|
||||
if res.submitErr != nil {
|
||||
f.res = nil
|
||||
return res.submitErr
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *fakeStreamTransfer) wait() (int, error) {
|
||||
if f.released {
|
||||
return 0, errors.New("wait() called on a free()d transfer")
|
||||
}
|
||||
if !f.inFlight {
|
||||
return 0, errors.New("wait() called without submit()")
|
||||
}
|
||||
if len(f.res) == 0 {
|
||||
return 0, errors.New("wait() called but fake result missing")
|
||||
}
|
||||
f.inFlight = false
|
||||
res := f.res[0]
|
||||
if res.waitErr == nil {
|
||||
f.res = f.res[1:]
|
||||
} else {
|
||||
f.res = nil
|
||||
}
|
||||
return res.n, res.waitErr
|
||||
}
|
||||
|
||||
func (f *fakeStreamTransfer) free() error {
|
||||
if f.released {
|
||||
return errors.New("free() called twice")
|
||||
}
|
||||
f.released = true
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *fakeStreamTransfer) cancel() error { return nil }
|
||||
func (f *fakeStreamTransfer) data() []byte { return fakeTransferBuf }
|
||||
|
||||
var sentinelError = errors.New("sentinel error")
|
||||
|
||||
type readRes struct {
|
||||
n int
|
||||
err error
|
||||
}
|
||||
|
||||
func (r readRes) String() string {
|
||||
var buf bytes.Buffer
|
||||
fmt.Fprintf(&buf, "<%d bytes", r.n)
|
||||
if r.err != nil {
|
||||
fmt.Fprintf(&buf, ", error: %s", r.err.Error())
|
||||
}
|
||||
buf.WriteString(">")
|
||||
return buf.String()
|
||||
}
|
||||
|
||||
func TestTransferReadStream(t *testing.T) {
|
||||
for tcNum, tc := range []struct {
|
||||
desc string
|
||||
closeBefore int
|
||||
// transfers is a list of allocated transfers, each transfers
|
||||
// carries a list of results for subsequent submits/waits.
|
||||
transfers [][]fakeStreamResult
|
||||
want []readRes
|
||||
}{
|
||||
{
|
||||
desc: "two transfers submitted, close, read returns both and EOF",
|
||||
closeBefore: 1,
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 400}},
|
||||
{{n: 400}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{err: io.EOF},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "two transfers, two and a half cycles through transfer queue",
|
||||
closeBefore: 4,
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 400}, {n: 400}, {n: 400}, {waitErr: errors.New("fake wait error")}},
|
||||
{{n: 400}, {n: 400}, {waitErr: errors.New("fake wait error")}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{err: io.EOF},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "4 transfers submitted, two return, third fails on wait",
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 500}},
|
||||
{{n: 500}},
|
||||
{{n: 123, waitErr: sentinelError}},
|
||||
{{n: 500}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 400},
|
||||
{n: 100},
|
||||
{n: 400},
|
||||
{n: 100},
|
||||
{n: 123, err: sentinelError},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "2 transfers, second submit fails initialization but error overshadowed by wait error",
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 123, waitErr: sentinelError}},
|
||||
{{submitErr: errors.New("fake submit error")}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 123, err: sentinelError},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "2 transfers, second submit fails during initialization",
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 400}},
|
||||
{{submitErr: sentinelError}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 400},
|
||||
{err: sentinelError},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "2 transfers, 3rd submit fails during second round",
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{n: 400}, {submitErr: sentinelError}},
|
||||
{{n: 400}},
|
||||
},
|
||||
want: []readRes{
|
||||
{n: 400},
|
||||
{n: 400},
|
||||
{err: sentinelError},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "fail quickly",
|
||||
transfers: [][]fakeStreamResult{
|
||||
{{waitErr: sentinelError}},
|
||||
{{n: 500}},
|
||||
{{n: 500}},
|
||||
},
|
||||
want: []readRes{
|
||||
{err: sentinelError},
|
||||
{err: io.ErrClosedPipe},
|
||||
},
|
||||
},
|
||||
} {
|
||||
t.Run(strconv.Itoa(tcNum), func(t *testing.T) {
|
||||
t.Logf("Case %d: %s", tcNum, tc.desc)
|
||||
ftt := make([]*fakeStreamTransfer, len(tc.transfers))
|
||||
tt := make([]transferIntf, len(tc.transfers))
|
||||
for i := range tc.transfers {
|
||||
ftt[i] = &fakeStreamTransfer{
|
||||
res: tc.transfers[i],
|
||||
}
|
||||
tt[i] = ftt[i]
|
||||
}
|
||||
s := ReadStream{newStream(tt, true)}
|
||||
buf := make([]byte, 400)
|
||||
got := make([]readRes, len(tc.want))
|
||||
for i := range tc.want {
|
||||
if i == tc.closeBefore-1 {
|
||||
t.Log("Close()")
|
||||
s.Close()
|
||||
}
|
||||
n, err := s.Read(buf)
|
||||
t.Logf("Read(): got %d, %v", n, err)
|
||||
got[i] = readRes{
|
||||
n: n,
|
||||
err: err,
|
||||
}
|
||||
}
|
||||
if !reflect.DeepEqual(got, tc.want) {
|
||||
t.Errorf("Got Read() results:\n%v\nwant Read() results:\n%v\n", got, tc.want)
|
||||
}
|
||||
for i := range ftt {
|
||||
if !ftt[i].released {
|
||||
t.Errorf("Transfer #%d was not freed after stream completed", i)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user