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/fleetupdate_add_test.go
+++ b/services/ota_update_go/handlers/fleetupdate_add_test.go
@@ -0,0 +1,188 @@
+package handlers_test
+
+import (
+	"encoding/base64"
+	"fmt"
+	"net/http"
+	"testing"
+
+	"otaupdate/handlers"
+	"otaupdate/services"
+
+	"github.com/fiskerinc/cloud-services/pkg/common"
+	dbm "github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
+	"github.com/fiskerinc/cloud-services/pkg/grpc/kafka_grpc"
+	htc "github.com/fiskerinc/cloud-services/pkg/httpclient/tester"
+	"github.com/fiskerinc/cloud-services/pkg/kafka"
+	km "github.com/fiskerinc/cloud-services/pkg/kafka/mock"
+	"github.com/fiskerinc/cloud-services/pkg/mongo"
+	"github.com/fiskerinc/cloud-services/pkg/redis"
+	rm "github.com/fiskerinc/cloud-services/pkg/redis/tester"
+	th "github.com/fiskerinc/cloud-services/pkg/testhelper"
+	"github.com/fiskerinc/cloud-services/pkg/testrunner"
+	uhelpers "github.com/fiskerinc/cloud-services/pkg/usecase_helpers"
+	"google.golang.org/protobuf/proto"
+)
+
+func TestFleetUpdateAdd(t *testing.T) {
+	client, err := services.GetMongoClient()
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	redis.MockRedisConnection()
+	vin1 := "1G1FP87S3GN100062"
+	vin2 := "TESTVIN1234567891"
+	mockDB := dbm.MockCarUpdates{}
+	mockKafka := km.KafkaMock{}
+	mockRedis := rm.MockRedis{}
+	mockMongo := mongo.NewFleetsCollection(
+		&mongo.MockCollection{
+			AggregateObject: []mongo.Fleet{
+				{
+					Name: "Grande",
+					Vehicles: []string{
+						vin1,
+						vin2,
+					},
+				},
+			},
+		},
+	)
+	client.SetFleets(mockMongo)
+	services.GetDB().SetCarUpdates(&mockDB)
+	services.SetKafkaProducer(&mockKafka)
+	services.SetRedisClientPool(rm.NewMockClientPool(&mockRedis))
+	otaUpdateKey1 := common.Service.Key(vin1)
+	otaUpdateKey2 := common.Service.Key(vin2)
+	attendentTopic := kafka.AttendantServiceGRPCKafka
+	updateMsg := &kafka_grpc.GRPC_AttendantPayload_UpdateManifest{
+		UpdateManifest: &kafka_grpc.UpdateManifest{
+			CarUpdateId: 1,
+		},
+	}
+	kafkaMSG := kafka_grpc.GRPC_AttendantPayload{
+		Handler: "send_manifest",
+		Data:    updateMsg,
+	}
+
+	binaryPayload, _ := proto.Marshal(&kafkaMSG)
+	sEnc := fmt.Sprintf(`"%s"`, base64.StdEncoding.EncodeToString(binaryPayload))
+	standardManifest := common.UpdateManifest{
+		ID:         100,
+		Name:       "TEST",
+		Version:    "10000",
+		Type:       "standard",
+		Country:    "US",
+		PowerTrain: "MD23",
+		Restraint:  "None",
+		Model:      "Ocean",
+		Trim:       "Sport",
+		Year:       2022,
+		BodyType:   "truck",
+	}
+	forcedManifest := standardManifest
+	forcedManifest.Type = "forced"
+
+	tests := []testrunner.TestCase{
+		{
+			Name: "Bad car ids",
+			HttpTestCase: &htc.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPost, "http://example.com/fleetupdate", uhelpers.JSONFleetUpdatesRequest{
+					UpdateManifestID: 1,
+				}),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"FleetNames required","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "No data",
+			HttpTestCase: &htc.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/fleetupdate", nil),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"UpdateManifestID required. FleetNames required","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "Bad package ids",
+			HttpTestCase: &htc.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPost, "http://example.com/fleetupdate", uhelpers.JSONFleetUpdatesRequest{
+					FleetNames: []string{"Slick Grande"},
+				}),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"UpdateManifestID required. FleetNames[0] fleet ","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "Good data standard manifest id",
+			HttpTestCase: &htc.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPost, "http://example.com/fleetupdate", uhelpers.JSONFleetUpdatesRequest{
+					UpdateManifestID: 1,
+					FleetNames:       []string{"Slick"},
+				}),
+				ExpectedStatus:   http.StatusOK,
+				ExpectedResponse: `{"data":["Slick"]}`,
+			},
+			DBTestCase: &dbm.DBTestCase{
+				SetupMockResponse: func() {
+					services.GetDB().SetUpdateManifests(&dbm.MockUpdateManifests{
+						LoadResponse: &standardManifest,
+					})
+				},
+			},
+			RedisTestCase: &rm.RedisTestCase{},
+			KafkaTestCase: &km.KafkaTestCase{
+				ExpectedProduceMessages: map[string]map[string]interface{}{
+					attendentTopic: {
+						otaUpdateKey1: sEnc,
+						otaUpdateKey2: sEnc,
+					},
+				},
+			},
+		},
+		{
+			Name: "Error",
+			HttpTestCase: &htc.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPost, "http://example.com/fleetupdate", uhelpers.JSONFleetUpdatesRequest{
+					UpdateManifestID: 1,
+					FleetNames:       []string{"Slick"},
+				}),
+				ExpectedStatus:   http.StatusServiceUnavailable,
+				ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
+			},
+			DBTestCase: &dbm.DBTestCase{
+				MockError: fmt.Errorf("something went wrong"),
+			},
+		},
+	}
+
+	for _, test := range tests {
+		mockRedis.Reset()
+		mockKafka.Reset()
+
+		if test.DBTestCase != nil {
+			test.DBTestCase.SetupDB(&mockDB)
+		}
+		if test.RedisTestCase != nil {
+			test.RedisTestCase.SetupRedis(&mockRedis)
+		}
+		if test.KafkaTestCase != nil {
+			test.KafkaTestCase.Setup(&mockKafka)
+		}
+
+		if test.HttpTestCase != nil {
+			w := test.HttpTestCase.Test(handlers.HandleFleetUpdatesAdd)
+			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)
+		}
+		if test.KafkaTestCase != nil {
+			test.KafkaTestCase.Validate(t, test.Name, &mockKafka)
+		}
+	}
+}