262 lines
9.2 KiB
Go
262 lines
9.2 KiB
Go
|
package throttle
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/cyrilix/robocar-base/testtools"
|
||
|
|
"github.com/cyrilix/robocar-protobuf/go/events"
|
||
|
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
||
|
|
"google.golang.org/protobuf/proto"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDefaultThrottle(t *testing.T) {
|
||
|
|
oldRegister := registerCallbacks
|
||
|
|
oldPublish := publish
|
||
|
|
defer func() {
|
||
|
|
registerCallbacks = oldRegister
|
||
|
|
publish = oldPublish
|
||
|
|
}()
|
||
|
|
registerCallbacks = func(p *Controller) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var muEventsPublished sync.Mutex
|
||
|
|
eventsPublished := make(map[string][]byte)
|
||
|
|
publish = func(client mqtt.Client, topic string, payload []byte) {
|
||
|
|
muEventsPublished.Lock()
|
||
|
|
defer muEventsPublished.Unlock()
|
||
|
|
eventsPublished[topic] = payload
|
||
|
|
}
|
||
|
|
|
||
|
|
throttleTopic := "topic/throttle"
|
||
|
|
driveModeTopic := "topic/driveMode"
|
||
|
|
rcThrottleTopic := "topic/rcThrottle"
|
||
|
|
steeringTopic := "topic/rcThrottle"
|
||
|
|
|
||
|
|
minValue := float32(0.56)
|
||
|
|
|
||
|
|
p := New(nil, throttleTopic, driveModeTopic, rcThrottleTopic, steeringTopic, minValue, 1., 200)
|
||
|
|
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
maxThrottle float32
|
||
|
|
driveMode events.DriveModeMessage
|
||
|
|
rcThrottle events.ThrottleMessage
|
||
|
|
expectedThrottle events.ThrottleMessage
|
||
|
|
}{
|
||
|
|
{"test1", 1., events.DriveModeMessage{DriveMode: events.DriveMode_USER}, events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0}, events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0}},
|
||
|
|
{"test2", 1., events.DriveModeMessage{DriveMode: events.DriveMode_PILOT}, events.ThrottleMessage{Throttle: 0.5, Confidence: 1.0}, events.ThrottleMessage{Throttle: 1.0, Confidence: 1.0}},
|
||
|
|
{"test3", 1., events.DriveModeMessage{DriveMode: events.DriveMode_PILOT}, events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0}, events.ThrottleMessage{Throttle: 1.0, Confidence: 1.0}},
|
||
|
|
{"test4", 1., events.DriveModeMessage{DriveMode: events.DriveMode_USER}, events.ThrottleMessage{Throttle: 0.5, Confidence: 1.0}, events.ThrottleMessage{Throttle: 0.5, Confidence: 1.0}},
|
||
|
|
{"test5", 1., events.DriveModeMessage{DriveMode: events.DriveMode_USER}, events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0}, events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0}},
|
||
|
|
{"test6", 1., events.DriveModeMessage{DriveMode: events.DriveMode_USER}, events.ThrottleMessage{Throttle: 0.6, Confidence: 1.0}, events.ThrottleMessage{Throttle: 0.6, Confidence: 1.0}},
|
||
|
|
{"limit max throttle on user mode", 0.4, events.DriveModeMessage{DriveMode: events.DriveMode_USER}, events.ThrottleMessage{Throttle: 0.6, Confidence: 1.0}, events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0}},
|
||
|
|
}
|
||
|
|
|
||
|
|
go p.Start()
|
||
|
|
defer func() { close(p.cancel) }()
|
||
|
|
|
||
|
|
for _, c := range cases {
|
||
|
|
t.Run(c.name, func(t *testing.T) {
|
||
|
|
p.maxThrottle = c.maxThrottle
|
||
|
|
p.onDriveMode(nil, testtools.NewFakeMessageFromProtobuf(driveModeTopic, &c.driveMode))
|
||
|
|
p.onRCThrottle(nil, testtools.NewFakeMessageFromProtobuf(rcThrottleTopic, &c.rcThrottle))
|
||
|
|
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
|
||
|
|
for i := 3; i >= 0; i-- {
|
||
|
|
|
||
|
|
var msg events.ThrottleMessage
|
||
|
|
muEventsPublished.Lock()
|
||
|
|
err := proto.Unmarshal(eventsPublished[throttleTopic], &msg)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unable to unmarshall response: %v", err)
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
muEventsPublished.Unlock()
|
||
|
|
|
||
|
|
if msg.GetThrottle() != c.expectedThrottle.GetThrottle() {
|
||
|
|
t.Errorf("bad msg value for mode %v: %v, wants %v", c.driveMode.String(), msg.GetThrottle(), c.expectedThrottle.GetThrottle())
|
||
|
|
}
|
||
|
|
if msg.GetConfidence() != 1. {
|
||
|
|
t.Errorf("bad throtlle confidence: %v, wants %v", msg.GetConfidence(), 1.)
|
||
|
|
}
|
||
|
|
|
||
|
|
time.Sleep(1 * time.Millisecond)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func TestController_Start(t *testing.T) {
|
||
|
|
oldRegister := registerCallbacks
|
||
|
|
oldPublish := publish
|
||
|
|
defer func() {
|
||
|
|
registerCallbacks = oldRegister
|
||
|
|
publish = oldPublish
|
||
|
|
}()
|
||
|
|
registerCallbacks = func(p *Controller) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
publishPilotFrequency := 10
|
||
|
|
waitPublish := sync.WaitGroup{}
|
||
|
|
var muEventsPublished sync.Mutex
|
||
|
|
eventsPublished := make(map[string][]byte)
|
||
|
|
publish = func(client mqtt.Client, topic string, payload []byte) {
|
||
|
|
muEventsPublished.Lock()
|
||
|
|
defer muEventsPublished.Unlock()
|
||
|
|
eventsPublished[topic] = payload
|
||
|
|
waitPublish.Done()
|
||
|
|
}
|
||
|
|
|
||
|
|
throttleTopic := "topic/throttle"
|
||
|
|
steeringTopic := "topic/steering"
|
||
|
|
driveModeTopic := "topic/driveMode"
|
||
|
|
rcThrottleTopic := "topic/rcThrottle"
|
||
|
|
|
||
|
|
type fields struct {
|
||
|
|
driveMode events.DriveMode
|
||
|
|
min, max float32
|
||
|
|
publishPilotFrequency int
|
||
|
|
}
|
||
|
|
type msgEvents struct {
|
||
|
|
driveMode *events.DriveModeMessage
|
||
|
|
steering *events.SteeringMessage
|
||
|
|
rcThrottle *events.ThrottleMessage
|
||
|
|
}
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
fields fields
|
||
|
|
msgEvents msgEvents
|
||
|
|
want *events.ThrottleMessage
|
||
|
|
wantErr bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "On user drive mode, throttle from rc",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_USER,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_USER},
|
||
|
|
steering: &events.SteeringMessage{Steering: 0.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.4, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "On user drive mode, limit throttle to max allowed value",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_USER,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_USER},
|
||
|
|
steering: &events.SteeringMessage{Steering: 0.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.9, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.8, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "On user drive mode, throttle can be < to min allowed value",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_USER,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_USER},
|
||
|
|
steering: &events.SteeringMessage{Steering: 0.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.1, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.1, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "On pilot drive mode and straight steering, use max throttle allowed",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_PILOT,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_PILOT},
|
||
|
|
steering: &events.SteeringMessage{Steering: 0.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.5, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.8, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "On pilot drive mode and on left steering, use min throttle allowed",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_PILOT,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_PILOT},
|
||
|
|
steering: &events.SteeringMessage{Steering: -1.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "On pilot drive mode and on right steering, use min throttle allowed",
|
||
|
|
fields: fields{
|
||
|
|
driveMode: events.DriveMode_PILOT,
|
||
|
|
max: 0.8,
|
||
|
|
min: 0.3,
|
||
|
|
publishPilotFrequency: publishPilotFrequency,
|
||
|
|
},
|
||
|
|
msgEvents: msgEvents{
|
||
|
|
driveMode: &events.DriveModeMessage{DriveMode: events.DriveMode_PILOT},
|
||
|
|
steering: &events.SteeringMessage{Steering: 1.0, Confidence: 1.0},
|
||
|
|
rcThrottle: &events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
want: &events.ThrottleMessage{Throttle: 0.3, Confidence: 1.0},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
c := New(nil,
|
||
|
|
throttleTopic, driveModeTopic, rcThrottleTopic, steeringTopic,
|
||
|
|
tt.fields.min, tt.fields.max,
|
||
|
|
tt.fields.publishPilotFrequency,
|
||
|
|
)
|
||
|
|
|
||
|
|
go c.Start()
|
||
|
|
defer func() { close(c.cancel) }()
|
||
|
|
time.Sleep(1 * time.Millisecond)
|
||
|
|
|
||
|
|
// Publish events and wait generation of new steering message
|
||
|
|
waitPublish.Add(1)
|
||
|
|
c.onDriveMode(nil, testtools.NewFakeMessageFromProtobuf(driveModeTopic, tt.msgEvents.driveMode))
|
||
|
|
c.onRCThrottle(nil, testtools.NewFakeMessageFromProtobuf(rcThrottleTopic, tt.msgEvents.rcThrottle))
|
||
|
|
c.onSteering(nil, testtools.NewFakeMessageFromProtobuf(steeringTopic, tt.msgEvents.steering))
|
||
|
|
waitPublish.Wait()
|
||
|
|
|
||
|
|
var msg events.ThrottleMessage
|
||
|
|
muEventsPublished.Lock()
|
||
|
|
err := proto.Unmarshal(eventsPublished[throttleTopic], &msg)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unable to unmarshall response: %v", err)
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
muEventsPublished.Unlock()
|
||
|
|
|
||
|
|
if msg.GetThrottle() != tt.want.GetThrottle() {
|
||
|
|
t.Errorf("bad msg value for mode %v: %v, wants %v", c.driveMode.String(), msg.GetThrottle(), tt.want.GetThrottle())
|
||
|
|
}
|
||
|
|
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|