147 lines
4.2 KiB
Go
147 lines
4.2 KiB
Go
package history
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"strings"
|
|
"testing"
|
|
|
|
"lingniu-vehicle-ingest/go/vehicle-gateway/internal/envelope"
|
|
)
|
|
|
|
func TestSchemaStatementsCreateCoreStables(t *testing.T) {
|
|
statements := strings.Join(SchemaStatements("test_ts"), "\n")
|
|
for _, want := range []string{"CREATE DATABASE IF NOT EXISTS test_ts", "raw_frames", "vehicle_locations", "vehicle_mileage_points"} {
|
|
if !strings.Contains(statements, want) {
|
|
t.Fatalf("schema missing %q:\n%s", want, statements)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestWriterAppendsRawLocationAndMileage(t *testing.T) {
|
|
exec := &recordingExec{}
|
|
writer := NewWriter(exec)
|
|
env := sampleEnvelope()
|
|
|
|
if err := writer.AppendAll(context.Background(), env); err != nil {
|
|
t.Fatalf("AppendAll() error = %v", err)
|
|
}
|
|
|
|
if got := countSQL(exec.calls, "USING raw_frames"); got != 1 {
|
|
t.Fatalf("raw child create count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "USING vehicle_locations"); got != 1 {
|
|
t.Fatalf("location child create count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "USING vehicle_mileage_points"); got != 1 {
|
|
t.Fatalf("mileage child create count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "INSERT INTO raw_"); got != 1 {
|
|
t.Fatalf("raw insert count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "INSERT INTO loc_"); got != 1 {
|
|
t.Fatalf("location insert count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "INSERT INTO mil_"); got != 1 {
|
|
t.Fatalf("mileage insert count = %d", got)
|
|
}
|
|
rawInsert := findSQL(exec.calls, "INSERT INTO raw_")
|
|
if !strings.Contains(rawInsert, "'go_") || !strings.Contains(rawInsert, "'2e") && !strings.Contains(rawInsert, "'") {
|
|
t.Fatalf("raw insert should quote string literals: %s", rawInsert)
|
|
}
|
|
if strings.Contains(rawInsert, "VALUES (?,") {
|
|
t.Fatalf("raw insert should not use placeholders for tdengine websocket plain insert: %s", rawInsert)
|
|
}
|
|
}
|
|
|
|
func TestWriterSkipsSparseDerivedRows(t *testing.T) {
|
|
exec := &recordingExec{}
|
|
writer := NewWriter(exec)
|
|
env := sampleEnvelope()
|
|
env.Fields = map[string]any{}
|
|
|
|
if err := writer.AppendAll(context.Background(), env); err != nil {
|
|
t.Fatalf("AppendAll() error = %v", err)
|
|
}
|
|
|
|
if got := countSQL(exec.calls, "INSERT INTO raw_"); got != 1 {
|
|
t.Fatalf("raw insert count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "INSERT INTO loc_"); got != 0 {
|
|
t.Fatalf("location insert count = %d", got)
|
|
}
|
|
if got := countSQL(exec.calls, "INSERT INTO mil_"); got != 0 {
|
|
t.Fatalf("mileage insert count = %d", got)
|
|
}
|
|
}
|
|
|
|
func TestRawSizeBytesUsesRawTextWhenHexIsEmpty(t *testing.T) {
|
|
env := envelope.FrameEnvelope{RawText: `{"code":"0F80","data":{"speed":12}}`}
|
|
if got, want := rawSizeBytes(env), len([]byte(env.RawText)); got != want {
|
|
t.Fatalf("raw text size = %d, want %d", got, want)
|
|
}
|
|
|
|
env.RawHex = "7E0200"
|
|
if got, want := rawSizeBytes(env), 3; got != want {
|
|
t.Fatalf("raw hex size = %d, want %d", got, want)
|
|
}
|
|
}
|
|
|
|
func sampleEnvelope() envelope.FrameEnvelope {
|
|
return envelope.FrameEnvelope{
|
|
Protocol: envelope.ProtocolJT808,
|
|
MessageID: "0x0200",
|
|
Sequence: 1,
|
|
VIN: "LNBVIN00000000001",
|
|
Phone: "013307795425",
|
|
SourceEndpoint: "127.0.0.1:18080",
|
|
EventTimeMS: 1782745114000,
|
|
ReceivedAtMS: 1782745114999,
|
|
RawHex: "7E0200",
|
|
Parsed: map[string]any{"message": "location"},
|
|
Fields: map[string]any{
|
|
envelope.FieldLongitude: 121.069881,
|
|
envelope.FieldLatitude: 30.590151,
|
|
envelope.FieldSpeedKMH: 23.0,
|
|
envelope.FieldTotalMileageKM: 10241.2,
|
|
"direction_deg": uint16(79),
|
|
"alarm_flag": uint32(0),
|
|
"status_flag": uint32(72),
|
|
},
|
|
ParseStatus: envelope.ParseOK,
|
|
}
|
|
}
|
|
|
|
func countSQL(calls []execCall, pattern string) int {
|
|
var count int
|
|
for _, call := range calls {
|
|
if strings.Contains(call.query, pattern) {
|
|
count++
|
|
}
|
|
}
|
|
return count
|
|
}
|
|
|
|
func findSQL(calls []execCall, pattern string) string {
|
|
for _, call := range calls {
|
|
if strings.Contains(call.query, pattern) {
|
|
return call.query
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
type execCall struct {
|
|
query string
|
|
args []any
|
|
}
|
|
|
|
type recordingExec struct {
|
|
calls []execCall
|
|
}
|
|
|
|
func (e *recordingExec) ExecContext(_ context.Context, query string, args ...any) (sql.Result, error) {
|
|
e.calls = append(e.calls, execCall{query: query, args: args})
|
|
return nil, nil
|
|
}
|