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/pkg/mongo/fleets_test.go
+++ b/pkg/mongo/fleets_test.go
@@ -0,0 +1,201 @@
+package mongo_test
+
+import (
+	"testing"
+
+	"fiskerinc.com/modules/common"
+	"fiskerinc.com/modules/db/queries"
+	"fiskerinc.com/modules/mongo"
+	"fiskerinc.com/modules/testhelper"
+	"fiskerinc.com/modules/utils/elptr"
+)
+
+func TestNewFleetsCollection(t *testing.T) {
+	client, err := mongo.NewClient(mongo.StandardDB)
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestNewFleetsCollection", nil, err)
+		return
+	}
+
+	mongo.NewFleetsCollection(mongo.NewCollection(client.Collection("fleets")))
+}
+
+func TestFleetsAddFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.AddFleet(&mongo.Fleet{
+		Name:     "TEST-FLEET",
+		LogLevel: 2,
+		CANBus: common.CANBus{
+			Enabled:      true,
+			DataLogger:   true,
+			DTCEnabled:   elptr.ElPtr(true),
+			MemBuffSize:  1 * 1024 * 1024,
+			DiskBuffSize: 10 * 1024 * 1024,
+		},
+	})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsAddFleet", nil, err)
+	}
+}
+
+func TestFleetsFindFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.FindFleet(&mongo.Fleet{Name: "TEST-FLEET"})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsFindFleet", nil, err)
+	}
+}
+
+func TestFleetsUpdateFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	fleet := mongo.Fleet{Name: "TEST-FlEET", CANBus: common.CANBus{Enabled: true, DataLogger: true, DTCEnabled: elptr.ElPtr(true)}, DebugMask: "12", IDPSEnabled: true}
+	err := collection.UpdateFleet(&mongo.Fleet{Name: "TEST-FLEET"}, &fleet)
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsUpdateFleet", nil, err)
+	}
+}
+
+func TestFleetsDeleteFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.AddFleet(&mongo.Fleet{Name: "TEST-FLEET"})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsDeleteFleet", nil, err)
+	}
+}
+
+func TestFleetsSelectFleets(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.SelectFleets(&mongo.Fleet{Name: "TEST-FLEET"}, &queries.PageQueryOptions{Offset: 1, Limit: 1})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsSelectFleets", nil, err)
+	}
+}
+
+func TestFleetsGetFleetCount(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.GetFleetCount(&mongo.Fleet{Name: "TEST-FLEET"})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsGetFleetCount", nil, err)
+	}
+}
+
+func TestFleetsGetVehiclesForFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.GetVehiclesForFleet("TEST-FLEET", "", &queries.PageQueryOptions{Offset: 1, Limit: 1})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsGetVehiclesForFleet", nil, err)
+	}
+}
+
+func TestFleetsGetVehiclesForFleetCount(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{
+			AggregateObject: []struct{ Total int64 }{{1}},
+		},
+	}
+
+	_, err := collection.GetVehiclesForFleetCount("TEST-FLEET", "")
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsGetVehiclesForFleetCount", nil, err)
+	}
+}
+
+func TestFleetsAddVehiclesToFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.AddVehiclesToFleet("TEST-FLEET", []string{"TESTVIN123"})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsAddVehiclesToFleet", nil, err)
+	}
+}
+
+func TestFleetsRemoveVehicleFromFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.DeleteVehiclesFromFleet("TEST-FLEET", []string{"TESTVIN123"})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsRemoveVehicleFromFleet", nil, err)
+	}
+}
+
+func TestFleetsGetFiltersForFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.GetFiltersForFleet("TEST-FLEET", &queries.PageQueryOptions{Offset: 1, Limit: 1})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsGetFiltersForFleet", nil, err)
+	}
+}
+
+func TestFleetsAddFilterToFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.AddFilterToFleet("TEST-FLEET",
+		&common.CANFilter{CANID: "123", Interval: elptr.ElPtr(456)})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsAddFilterToFleet", nil, err)
+	}
+}
+
+func TestFleetsUpdateFilterForFleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.UpdateFilterForFleet("TEST-FLEET", "123",
+		&common.CANFilter{CANID: "123", Interval: elptr.ElPtr(789)})
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsUpdateFilterForFleet", nil, err)
+	}
+}
+
+func TestFleetsDeleteFilterForleet(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	err := collection.DeleteFilterFromFleet("TEST-FLEET", "123")
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsDeleteFilterForleet", nil, err)
+	}
+}
+
+func TestFleetsGetFiltersForVehicle(t *testing.T) {
+	collection := mongo.FleetsCollection{
+		&mongo.MockCollection{},
+	}
+
+	_, err := collection.GetCANBusForVehicle("TESTVIN123")
+	if err != nil {
+		t.Errorf(testhelper.TestErrorTemplate, "TestFleetsGetFiltersForVehicle", nil, err)
+	}
+}