156 lines
3.5 KiB
Go
156 lines
3.5 KiB
Go
package db
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/kschamplin/gotelem/internal/logparsers"
|
|
"github.com/kschamplin/gotelem/skylab"
|
|
)
|
|
|
|
// helper func to get a random bus event with random data.
|
|
func GetRandomBusEvent() skylab.BusEvent {
|
|
data := skylab.WsrVelocity{
|
|
MotorVelocity: 1.0,
|
|
VehicleVelocity: 4.0,
|
|
}
|
|
ev := skylab.BusEvent{
|
|
Timestamp: time.Now(),
|
|
Data: &data,
|
|
}
|
|
|
|
return ev
|
|
}
|
|
|
|
// exampleData is a telemetry log data snippet that
|
|
// we use to seed the database.
|
|
const exampleData = `1698013005.164 1455ED8FDBDFF4FC3BD
|
|
1698013005.168 1460000000000000000
|
|
1698013005.170 1470000000000000000
|
|
1698013005.172 1610000000000000000
|
|
1698013005.175 1210000000000000000
|
|
1698013005.177 157FFFFC74200000000
|
|
1698013005.181 1030000000000000000
|
|
1698013005.184 1430000000000000000
|
|
1698013005.187 04020D281405EA8FB41
|
|
1698013005.210 0413BDF81406AF70042
|
|
1698013005.212 042569F81408EF0FF41
|
|
1698013005.215 04358A8814041060242
|
|
1698013005.219 04481958140D2A40342
|
|
1698013005.221 0452DB2814042990442
|
|
1698013005.224 047AF948140C031FD41
|
|
1698013005.226 04B27A081401ACD0B42
|
|
1698013005.229 04DCEAA81403C8C0A42
|
|
1698013005.283 04E0378814024580142
|
|
1698013005.286 04F97908140BFBC0142
|
|
1698013005.289 050098A81402F0F0A42
|
|
1698013005.293 051E6AE81402AF20842
|
|
1698013005.297 0521AC081403A970742
|
|
1698013005.300 0535BB181403CEB0542
|
|
1698013005.304 054ECC0814088FE0142
|
|
1698013005.307 0554ED181401F44F341
|
|
1698013005.309 05726E48140D42BEB41
|
|
1698013005.312 059EFC98140EC400142
|
|
`
|
|
|
|
// MakeMockDatabase creates a new dummy database.
|
|
func MakeMockDatabase(name string) *TelemDb {
|
|
fstring := fmt.Sprintf("file:%s?mode=memory&cache=shared", name)
|
|
tdb, err := openRawDb(fstring)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// seed the database now.
|
|
scanner := bufio.NewScanner(strings.NewReader(exampleData))
|
|
|
|
for scanner.Scan() {
|
|
str := scanner.Text()
|
|
|
|
bev, err := logparsers.ParsersMap["telem"](str)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err = tdb.AddEvents(bev)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
return tdb
|
|
}
|
|
|
|
func TestTelemDb(t *testing.T) {
|
|
|
|
|
|
t.Run("test opening database", func(t *testing.T) {
|
|
// create our mock
|
|
tdb := MakeMockDatabase(t.Name())
|
|
tdb.db.Ping()
|
|
})
|
|
|
|
t.Run("test inserting bus event", func(t *testing.T) {
|
|
tdb := MakeMockDatabase(t.Name())
|
|
type args struct {
|
|
events []skylab.BusEvent
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "add no packet",
|
|
args: args{
|
|
events: []skylab.BusEvent{},
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "add single packet",
|
|
args: args{
|
|
events: []skylab.BusEvent{GetRandomBusEvent()},
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "add multiple packet",
|
|
args: args{
|
|
events: []skylab.BusEvent{GetRandomBusEvent(), GetRandomBusEvent()},
|
|
},
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if _, err := tdb.AddEvents(tt.args.events...); (err != nil) != tt.wantErr {
|
|
t.Errorf("TelemDb.AddEvents() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
|
|
})
|
|
|
|
t.Run("test getting packets", func(t *testing.T) {
|
|
tdb := MakeMockDatabase(t.Name())
|
|
|
|
ctx := context.Background()
|
|
f := BusEventFilter{}
|
|
limitMod := LimitOffsetModifier{Limit: 1}
|
|
pkt, err := tdb.GetPackets(ctx, f, limitMod)
|
|
if err != nil {
|
|
t.Fatalf("error getting packets: %v", err)
|
|
}
|
|
if len(pkt) != 1 {
|
|
t.Fatalf("expected exactly one response, got %d", len(pkt))
|
|
}
|
|
})
|
|
|
|
t.Run("test read-write packet", func(t *testing.T) {
|
|
|
|
})
|
|
}
|