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/supplier_update_test.go
+++ b/services/ota_update_go/handlers/supplier_update_test.go
@@ -0,0 +1,105 @@
+package handlers_test
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+
+	"otaupdate/handlers"
+	"otaupdate/services"
+
+	"github.com/fiskerinc/cloud-services/pkg/common"
+	"github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
+	"github.com/fiskerinc/cloud-services/pkg/httpclient/tester"
+	th "github.com/fiskerinc/cloud-services/pkg/testhelper"
+	"github.com/fiskerinc/cloud-services/pkg/testrunner"
+)
+
+const err404NotFound = "404 page not found\n"
+
+func TestHandleSupplierUpdate(t *testing.T) {
+	route := "/supplier/:email"
+	mock := mocks.MockSupplierAccounts{}
+	reqGoodData := common.SupplierAccount{
+		SupplierOrganizationID: 14,
+		Company:                "TEST",
+		Address:                "100 Main",
+		Contact:                "John Doe",
+		Telephone:              "+1-555-555-5555",
+		Program:                "Ocean",
+		ECUs:                   []string{"ADAS"},
+	}
+
+	tests := []testrunner.TestCase{
+		{
+			Name: "No data",
+			HttpTestCase: &tester.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPut, "http://example.com/supplier", nil),
+				ExpectedStatus:   http.StatusNotFound,
+				ExpectedResponse: err404NotFound,
+			},
+		},
+		{
+			Name: "Bad Data",
+			HttpTestCase: &tester.HttpTestCase{
+				Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test", common.SupplierAccount{
+					Company:   "TEST",
+					Address:   "100 Main",
+					Contact:   "John Doe",
+					Telephone: "+1-555-555-5555",
+					Program:   "Ocean",
+					ECUs:      []string{},
+				}),
+				ExpectedStatus:   http.StatusBadRequest,
+				ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
+			},
+		},
+		{
+			Name: "Good data",
+			HttpTestCase: &tester.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
+				ExpectedStatus:   http.StatusOK,
+				ExpectedResponse: `{"supplier_organization_id":"14","email":"test@supplier.com","company":"TEST","address":"100 Main","contact":"John Doe","telephone":"+1-555-555-5555","program":"Ocean","ecus":["ADAS"]}`,
+			},
+		},
+		{
+			Name: "Error unregistered email",
+			HttpTestCase: &tester.HttpTestCase{
+				Request:          th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
+				ExpectedStatus:   http.StatusServiceUnavailable,
+				ExpectedResponse: `{"message":"account does not exist","error":"Service Unavailable"}`,
+				Setup: func() {
+					mock.ORMResponse = &mocks.MockORMResults{AffectedRows: 0}
+					services.GetDB().SetSupplierAccount(&mock)
+				},
+			},
+		},
+		{
+			Name: "Error",
+			HttpTestCase: &tester.HttpTestCase{Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
+				ExpectedStatus:   http.StatusServiceUnavailable,
+				ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
+			},
+			DBTestCase: &mocks.DBTestCase{
+				MockError: fmt.Errorf("something went wrong"),
+			},
+		},
+	}
+
+	for _, test := range tests {
+		if test.DBTestCase != nil {
+			test.DBTestCase.SetupDB(&mock)
+		}
+
+		mock.ORMResponse = &mocks.MockORMResults{AffectedRows: 1}
+		services.GetDB().SetSupplierAccount(&mock)
+
+		if test.HttpTestCase != nil && test.HttpTestCase.Setup != nil {
+			test.HttpTestCase.Setup()
+		}
+
+		w := test.HttpTestCase.TestWithParamPath(handlers.HandleSupplierUpdate, route)
+
+		test.HttpTestCase.ValidateHttp(t, test.Name, w)
+	}
+}