Initial cloud-services repo - gateway service + pkg modules

2026-01-30 23:14:52 -05:00
commit fbb820d7b3
1037 changed files with 171318 additions and 0 deletions
--- a/services/gateway/websocket/session_trex_test.go
+++ b/services/gateway/websocket/session_trex_test.go
@@ -0,0 +1,191 @@
+package websocket
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"fiskerinc.com/modules/common"
+	"fiskerinc.com/modules/grpc/kafka_grpc"
+	kafka "fiskerinc.com/modules/kafka/mock"
+	"fiskerinc.com/modules/testhelper"
+
+	"github.com/gobwas/ws"
+	"github.com/gobwas/ws/wsutil"
+)
+
+func TestSessionTRex(t *testing.T) {
+	ws, _ := net.Pipe()
+	s := &SessionTRex{
+		Session: &Session{
+			Websocket: ws,
+		},
+	}
+
+	if fmt.Sprintf("%T", s) != "*websocket.SessionTRex" {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRex", "*websocket.SessionTRex", fmt.Sprintf("%T", s))
+	}
+}
+
+func TestNewTRexSession(t *testing.T) {
+	createNewSession := func(w http.ResponseWriter, r *http.Request) {
+		s, err := NewTRexSession(w, r, "1F15K3R45N1234567", "1.2.3.4", "12345678912346789123456789")
+		if err != nil {
+			t.Errorf(testhelper.TestErrorTemplate, "TestNewSessionTRex", nil, err)
+		}
+		defer s.Close()
+
+		if fmt.Sprintf("%T", s) != "*websocket.SessionTRex" {
+			t.Errorf(testhelper.TestErrorTemplate, "TestNewSessionTRex", "*websocket.SessionTRex", fmt.Sprintf("%T", s))
+		}
+	}
+
+	server := httptest.NewServer(http.HandlerFunc(createNewSession))
+	defer server.Close()
+
+	conn := createMockWebsocketClient(server.URL, "")
+	defer conn.Close()
+}
+
+func TestSessionTRexAuthenticate(t *testing.T) {
+	userAgent := "Fisker Ocean T.Rex 1.2.3.4 abc123"
+
+	createNewSession := func(w http.ResponseWriter, r *http.Request) {
+		s, err := NewSecureSession(w, r)
+		if err != nil {
+			t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexAuthenticate", nil, err)
+		}
+		defer s.Close()
+
+		err = s.Authenticate()
+		if err != nil {
+			t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexAuthenticate", nil, err)
+		}
+	}
+
+	server := httptest.NewServer(http.HandlerFunc(createNewSession))
+	defer server.Close()
+
+	conn := createMockWebsocketClient(server.URL, userAgent)
+	defer conn.Close()
+
+	am := common.TRexSessionMessage{
+		Handler: "verify",
+		Data: common.TRexSessionData{
+			VIN: "1HD1CGP134K410769",
+		},
+	}
+
+	msg, err := json.Marshal(am)
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexAuthenticate", nil, err)
+	}
+
+	wsutil.WriteClientMessage(conn, ws.OpText, msg)
+}
+
+func TestSessionTRexListen(t *testing.T) {
+	userAgent := "Fisker Ocean T.Rex 1.2.3.4 abc123"
+	payload := "hello fisker!"
+
+	createNewSession := func(w http.ResponseWriter, r *http.Request) {
+		s, err := NewSecureSession(w, r)
+		if err != nil {
+			t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexListen", nil, err)
+		}
+		defer s.Close()
+
+		ctx := context.Background()
+		err = s.Listen(ctx, kafka.GetKafkaMock(nil))
+		if err.Error() != "EOF" {
+			t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexListen", nil, err)
+		}
+	}
+
+	server := httptest.NewServer(http.HandlerFunc(createNewSession))
+	defer server.Close()
+
+	conn := createMockWebsocketClient(server.URL, userAgent)
+	defer conn.Close()
+
+	err := wsutil.WriteClientMessage(conn, ws.OpText, []byte(payload))
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexListen", nil, err)
+	}
+}
+
+func TestSessionTRexRoute(t *testing.T) {
+	ws, _ := net.Pipe()
+	s := &SessionTRex{
+		Session: &Session{
+			Websocket: ws,
+		},
+	}
+
+	msg := common.Message{
+		Handler: "canbus",
+		Data: &kafka_grpc.GRPC_CANData{
+			EpochUsec: 1653255445,
+			Dropped:   10,
+			Filtered:  20,
+			Frames: []*kafka_grpc.GRPC_CANFrame{
+				{
+					Epoch: 1642455023642165,
+					ID:    832,
+					Value: []byte("AAAAAAAAIAE="),
+				},
+			},
+		},
+	}
+
+	data, err := json.Marshal(msg)
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexRoute", nil, err)
+	}
+
+	err = s.Route(kafka.GetKafkaMock(nil), data)
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexRoute", nil, err)
+	}
+}
+
+func TestSessionTRexLoad(t *testing.T) {
+	ws, _ := net.Pipe()
+	s := &SessionTRex{
+		Session: &Session{
+			Websocket: ws,
+			Type:      common.TRex,
+			ID:        "FISKER123",
+		},
+	}
+
+	err := s.Load(kafka.GetKafkaMock(nil))
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexLoad", nil, err)
+	}
+}
+
+func TestSessionTRexTeardown(t *testing.T) {
+	ws, _ := net.Pipe()
+	s := &SessionTRex{
+		Session: &Session{
+			Websocket: ws,
+			Type:      common.TRex,
+			ID:        "FISKER123",
+		},
+	}
+
+	err := s.Load(kafka.GetKafkaMock(nil))
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexTeardown", nil, err)
+	}
+
+	err = s.Teardown(kafka.GetKafkaMock(nil))
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestSessionTRexTeardown", nil, err)
+	}
+}