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

services/ota_update_go/handlers/sub_feature_assign_test.go

@@ -0,0 +1,57 @@
+package handlers_test
+
+import (
+	"fmt"
+	"net/http"
+	"otaupdate/handlers"
+	"otaupdate/services"
+	"testing"
+
+	mo "github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
+	th "github.com/fiskerinc/cloud-services/pkg/testhelper"
+	"github.com/google/uuid"
+)
+
+func TestSubscriptionFeatureAssign(t *testing.T) {
+	mock := mo.MockSubscriptionPackages{}
+	services.GetDB().SetSubPackages(&mock)
+	testPackageID := uuid.MustParse("0557bd1d-76d3-41e5-a44e-13c479e55ab0")
+	testFeatureID := uuid.MustParse("ecfb89e0-ca03-4aa9-a43a-a9d703256edb")
+	validData := handlers.SubFeatureAssignRequest{
+		PackageID: testPackageID,
+		FeatureID: testFeatureID,
+	}
+	tests := []mo.DBHttpTest{
+		{
+			Name:             "No data",
+			Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/subscriptionpackagefeature", nil),
+			ExpectedStatus:   http.StatusBadRequest,
+			ExpectedResponse: `{"message":"PackageID required. FeatureID required","error":"Bad Request"}`,
+		},
+		{
+			Name: "Invalid data",
+			Request: th.MakeTestRequest(http.MethodPost, "http://example.com/subscriptionpackagefeature", handlers.SubFeatureAssignRequest{
+				PackageID: testPackageID,
+			}),
+			ExpectedStatus:   http.StatusBadRequest,
+			ExpectedResponse: `{"message":"FeatureID required","error":"Bad Request"}`,
+		},
+		{
+			Name:             "Valid data",
+			Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/subscriptionpackagefeature", validData),
+			ExpectedStatus:   http.StatusOK,
+			ExpectedResponse: `{"message":"OK"}`,
+		},
+		{
+			Name:             "DB error",
+			Request:          th.MakeTestRequest(http.MethodPost, "http://example.com/subscriptionpackagefeature", validData),
+			ExpectedStatus:   http.StatusServiceUnavailable,
+			ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
+			DBTestCase: mo.DBTestCase{
+				MockError: fmt.Errorf("something went wrong"),
+			},
+		},
+	}
+
+	mo.RunDBTests(t, tests, handlers.HandleSubscriptionFeatureAssign, &mock)
+}