- more tests, for "out" endpoint.

This commit is contained in:
Sebastian Zagrodzki
2017-02-22 01:35:13 +01:00
parent ea2d93bbca
commit 34c00f398b
3 changed files with 76 additions and 19 deletions

View File

@@ -17,7 +17,6 @@ package usb
import ( import (
"fmt" "fmt"
"log"
"time" "time"
) )
@@ -74,7 +73,6 @@ func (e *endpoint) transfer(buf []byte, timeout time.Duration) (int, error) {
return 0, nil return 0, nil
} }
tt := e.TransferType()
t, err := e.newUSBTransfer(buf, timeout) t, err := e.newUSBTransfer(buf, timeout)
if err != nil { if err != nil {
return 0, err return 0, err
@@ -82,17 +80,15 @@ func (e *endpoint) transfer(buf []byte, timeout time.Duration) (int, error) {
defer t.free() defer t.free()
if err := t.submit(); err != nil { if err := t.submit(); err != nil {
log.Printf("bulk: %s failed to submit: %s", tt, err)
return 0, err return 0, err
} }
n, err := t.wait() n, err := t.wait()
if err != nil { if err != nil {
log.Printf("bulk: %s failed: %s", tt, err)
return 0, err
}
return n, err return n, err
} }
return n, nil
}
func newEndpoint(d *Device) *endpoint { func newEndpoint(d *Device) *endpoint {
ep := &endpoint{ ep := &endpoint{

View File

@@ -28,3 +28,19 @@ var testBulkInSetup = InterfaceSetup{
IfClass: uint8(CLASS_VENDOR_SPEC), IfClass: uint8(CLASS_VENDOR_SPEC),
Endpoints: []EndpointInfo{testBulkInEP}, Endpoints: []EndpointInfo{testBulkInEP},
} }
// OUT iso endpoint
var testIsoOutEP = EndpointInfo{
Address: 0x06,
Attributes: uint8(TRANSFER_TYPE_ISOCHRONOUS),
MaxPacketSize: 3<<11 + 1024,
MaxIsoPacket: 3 * 1024,
PollInterval: 1,
}
var testIsoOutSetup = InterfaceSetup{
Number: 0,
Alternate: 0,
IfClass: uint8(CLASS_VENDOR_SPEC),
Endpoints: []EndpointInfo{testIsoOutEP},
}

View File

@@ -15,6 +15,8 @@
package usb package usb
import ( import (
"errors"
"reflect"
"testing" "testing"
"time" "time"
) )
@@ -30,23 +32,66 @@ func (t *fakeTransfer) wait() (int, error) { return t.ret, t.err }
func (*fakeTransfer) free() error { return nil } func (*fakeTransfer) free() error { return nil }
func TestEndpoint(t *testing.T) { func TestEndpoint(t *testing.T) {
ep := endpoint{ for _, epSetup := range []struct {
InterfaceSetup: testBulkInSetup, method string
EndpointInfo: testBulkInEP, InterfaceSetup
EndpointInfo
}{
{"Read", testBulkInSetup, testBulkInEP},
{"Write", testIsoOutSetup, testIsoOutEP},
} {
t.Run(epSetup.method, func(t *testing.T) {
for _, tc := range []struct {
buf []byte
ret int
err error
want int
wantErr bool
}{
{buf: nil, ret: 10, want: 0},
{buf: make([]byte, 128), ret: 60, want: 60},
{buf: make([]byte, 128), ret: 10, err: errors.New("some error"), want: 10, wantErr: true},
} {
ep := &endpoint{
InterfaceSetup: epSetup.InterfaceSetup,
EndpointInfo: epSetup.EndpointInfo,
newUSBTransfer: func(buf []byte, timeout time.Duration) (transferIntf, error) { newUSBTransfer: func(buf []byte, timeout time.Duration) (transferIntf, error) {
return &fakeTransfer{buf}, nil return &fakeTransfer{buf: buf, ret: tc.ret, err: tc.err}, nil
}, },
} }
b := make([]byte, 128) op, ok := reflect.TypeOf(ep).MethodByName(epSetup.method)
got, err := ep.Read(b) if !ok {
if err != nil { t.Fatalf("method %s not found in endpoint", epSetup.method)
t.Fatalf("bulkInEP.Read(): got error: %v, want nil", err)
} }
if want := len(b)/2 + 1; got != want { opv := op.Func.Interface().(func(*endpoint, []byte) (int, error))
t.Errorf("bulkInEP.Read(): got %d bytes, want half of the buffer length: %d", got, want) got, err := opv(ep, tc.buf)
if (err != nil) != tc.wantErr {
t.Errorf("bulkInEP.Read(): got err: %v, err != nil is %v, want %v", err, err != nil, tc.wantErr)
continue
} }
_, err := ep.Write(b) if got != tc.want {
t.Errorf("bulkInEP.Read(): got %d bytes, want %d", got, tc.want)
}
}
})
}
}
func TestEndpointWrongDirection(t *testing.T) {
ep := &endpoint{
InterfaceSetup: testBulkInSetup,
EndpointInfo: testBulkInEP,
}
_, err := ep.Write([]byte{1, 2, 3})
if err == nil { if err == nil {
t.Error("bulkInEP.Write(): got nil error, want non-nil") t.Error("bulkInEP.Write(): got nil error, want non-nil")
} }
ep = &endpoint{
InterfaceSetup: testIsoOutSetup,
EndpointInfo: testIsoOutEP,
}
_, err = ep.Read(make([]byte, 64))
if err == nil {
t.Error("isoOutEP.Read(): got nil error, want non-nil")
}
} }