1
0
mirror of https://github.com/hybridgroup/gobot.git synced 2025-04-24 13:48:49 +08:00

neurosky: eliminate race conditions introduced by tests

Signed-off-by: deadprogram <ron@hybridgroup.com>
This commit is contained in:
deadprogram 2017-04-02 22:08:10 +02:00
parent fe19c2f167
commit c11b0d3b3c
2 changed files with 51 additions and 15 deletions

View File

@ -3,6 +3,7 @@ package neurosky
import (
"errors"
"io"
"sync"
"testing"
"gobot.io/x/gobot"
@ -11,22 +12,44 @@ import (
var _ gobot.Adaptor = (*Adaptor)(nil)
type NullReadWriteCloser struct{}
type NullReadWriteCloser struct {
mtx sync.Mutex
readError error
closeError error
}
func (NullReadWriteCloser) Write(p []byte) (int, error) {
// func NewNullReadWriteCloser() *NullReadWriteCloser {
// return NullReadWriteCloser{
//
// }
// }
func (n *NullReadWriteCloser) ReadError(e error) {
n.mtx.Lock()
defer n.mtx.Unlock()
n.readError = e
}
func (n *NullReadWriteCloser) CloseError(e error) {
n.mtx.Lock()
defer n.mtx.Unlock()
n.closeError = e
}
func (n *NullReadWriteCloser) Write(p []byte) (int, error) {
return len(p), nil
}
var readError error = nil
func (NullReadWriteCloser) Read(b []byte) (int, error) {
return len(b), readError
func (n *NullReadWriteCloser) Read(b []byte) (int, error) {
n.mtx.Lock()
defer n.mtx.Unlock()
return len(b), n.readError
}
var closeError error
func (NullReadWriteCloser) Close() error {
return closeError
func (n *NullReadWriteCloser) Close() error {
n.mtx.Lock()
defer n.mtx.Unlock()
return n.closeError
}
func initTestNeuroskyAdaptor() *Adaptor {
@ -53,11 +76,15 @@ func TestNeuroskyAdaptorConnect(t *testing.T) {
}
func TestNeuroskyAdaptorFinalize(t *testing.T) {
a := initTestNeuroskyAdaptor()
rwc := &NullReadWriteCloser{}
a := NewAdaptor("/dev/null")
a.connect = func(n *Adaptor) (io.ReadWriteCloser, error) {
return rwc, nil
}
a.Connect()
gobottest.Assert(t, a.Finalize(), nil)
closeError = errors.New("close error")
rwc.CloseError(errors.New("close error"))
a.Connect()
gobottest.Assert(t, a.Finalize(), errors.New("close error"))
}

View File

@ -28,16 +28,25 @@ func TestNeuroskyDriver(t *testing.T) {
}
func TestNeuroskyDriverStart(t *testing.T) {
sem := make(chan bool, 0)
d := initTestNeuroskyDriver()
rwc := &NullReadWriteCloser{}
a := NewAdaptor("/dev/null")
a.connect = func(n *Adaptor) (io.ReadWriteCloser, error) {
return rwc, nil
}
a.Connect()
d := NewDriver(a)
e := errors.New("read error")
d.Once(d.Event(Error), func(data interface{}) {
gobottest.Assert(t, data.(error), errors.New("read error"))
gobottest.Assert(t, data.(error), e)
sem <- true
})
gobottest.Assert(t, d.Start(), nil)
time.Sleep(50 * time.Millisecond)
readError = errors.New("read error")
rwc.ReadError(e)
select {
case <-sem: