Add depot, attendant, jetfire, optimus, ota services with kustomize overlays

2026-01-31 15:35:07 -05:00
parent a0ec642ca1
commit 9a5cb2f547
404 changed files with 38817 additions and 16 deletions
--- a/services/ota_update_go/handlers/fleet_update_test.go
+++ b/services/ota_update_go/handlers/fleet_update_test.go
@@ -0,0 +1,144 @@
+package handlers_test
+
+import (
+	"net/http"
+	"testing"
+
+	"otaupdate/handlers"
+	"otaupdate/services"
+
+	m "github.com/fiskerinc/cloud-services/pkg/common"
+	dbtc "github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
+	htc "github.com/fiskerinc/cloud-services/pkg/httpclient/tester"
+	"github.com/fiskerinc/cloud-services/pkg/mongo"
+	"github.com/fiskerinc/cloud-services/pkg/redis"
+	rtc "github.com/fiskerinc/cloud-services/pkg/redis/tester"
+	th "github.com/fiskerinc/cloud-services/pkg/testhelper"
+	tr "github.com/fiskerinc/cloud-services/pkg/testrunner"
+	"github.com/fiskerinc/cloud-services/pkg/utils/elptr"
+)
+
+func TestFleetUpdate(t *testing.T) {
+	mockDB := dbtc.MockCars{}
+	services.GetDB().SetCars(&mockDB)
+	vin := "1G1FP87S3GN100062"
+	trexKey := m.TRex.Key(vin)
+	cacheKey := redis.CarConfigKey(vin)
+
+	client, err := services.GetMongoClient()
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	mockMongoFleets := mongo.NewFleetsCollection(&mongo.MockCollection{})
+	mockMongoVehicles := mongo.NewVehiclesCollection(&mongo.MockCollection{})
+	client.SetFleets(mockMongoFleets)
+	client.SetVehicles(mockMongoVehicles)
+
+	mockRedis := rtc.MockRedis{}
+	services.SetRedisClientPool(rtc.NewMockClientPool(&mockRedis))
+
+	tests := []tr.TestCase{
+		{
+			Name: "No data",
+			HttpTestCase: &htc.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/fleet/TESTFLEET", nil),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "Invalid data",
+			HttpTestCase: &htc.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/fleet/TESTFLEET", mongo.Fleet{Name: "TEST_FLEET"}),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "Valid data 1",
+			HttpTestCase: &htc.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPut, "http://example.com/fleet/TESTFLEET", mongo.Fleet{Name: "TESTFLEET"}),
+				ExpectedStatus:   http.StatusOK,
+				ExpectedResponse: `{"name":"TESTFLEET","log_level":"trace","canbus":{"enabled":false,"data_logger_enabled":false,"dtc_enabled":false},"idps_enabled":false,"tags":null,"vehicles":null,"vehicles_count":0}`,
+			},
+			RedisTestCase: &rtc.RedisTestCase{
+				ExpectedMessages: nil,
+				ExpectedCaches:   nil,
+			},
+		},
+		{
+			Name: "Valid data 2",
+			HttpTestCase: &htc.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPut, "http://example.com/fleet/TESTFLEET",
+					mongo.Fleet{
+						VehiclesCount: 1,
+						Vehicles:      []string{vin},
+						Name:          "TESTFLEET",
+						LogLevel:      m.Info,
+						CANBus: m.CANBus{
+							Enabled:    true,
+							DTCEnabled: elptr.ElPtr(false),
+						},
+						DebugMask:   "12",
+						IDPSEnabled: true,
+					},
+				),
+				ExpectedStatus:   http.StatusOK,
+				ExpectedResponse: `{"name":"TESTFLEET","log_level":"info","canbus":{"enabled":true,"data_logger_enabled":false,"dtc_enabled":false},"idps_enabled":true,"debug_mask":"12","tags":null,"vehicles":["1G1FP87S3GN100062"],"vehicles_count":1}`,
+				Setup: func() {
+					mockMongoFleets = mongo.NewFleetsCollection(&mongo.MockCollection{
+						AggregateObject: []mongo.Fleet{
+							{
+								Name: "US-TEST",
+								Vehicles: []string{
+									vin,
+								},
+							},
+						},
+					})
+					client.SetFleets(mockMongoFleets)
+				},
+			},
+			RedisTestCase: &rtc.RedisTestCase{
+				ExpectedMessages: map[string]string{
+					trexKey: `{"data":{"canbus":{"data_logger_enabled":false,"dtc_enabled":false,"enabled":true},"idps_enabled":true,"log_level":"info"},"handler":"config"}`,
+				},
+				ExpectedCaches: map[string]rtc.ExpiringCacheResult{
+					cacheKey: {Value: "DELETED"},
+				},
+			},
+		},
+	}
+
+	schemaTesterTRex := th.NewSchemaTestHelper(t, schemaToTRex)
+	for _, test := range tests {
+		mockRedis.Reset()
+
+		if test.DBTestCase != nil {
+			test.DBTestCase.SetupDB(&mockDB)
+		}
+		if test.RedisTestCase != nil {
+			test.RedisTestCase.SetupRedis(&mockRedis)
+		}
+
+		if test.HttpTestCase != nil {
+			if test.HttpTestCase.Setup != nil {
+				test.HttpTestCase.Setup()
+			}
+
+			w := test.HttpTestCase.TestWithParamPath(handlers.HandleFleetUpdate, "/fleet/:name")
+			test.HttpTestCase.ValidateHttp(t, test.Name, w)
+		}
+		if test.DBTestCase != nil {
+			test.DBTestCase.Validate(t, test.Name, &mockDB)
+		}
+		if test.RedisTestCase != nil {
+			test.RedisTestCase.Validate(t, test.Name, &mockRedis)
+
+			for _, mes := range test.RedisTestCase.ExpectedMessages {
+				schemaTesterTRex.ValidateSchemaObject(test.Name, []byte(mes))
+			}
+		}
+	}
+}