testing

internal/use-case/main-use-case/unit/case.go

@@ -1,6 +1,7 @@
 package unit
 
 import (
+	"errors"
 	e "simrs-vx/internal/domain/main-entities/unit"
 	erc "simrs-vx/internal/domain/references/common"
 	esync "simrs-vx/internal/domain/sync-entities/log"
@@ -27,9 +28,22 @@ func Create(input e.CreateDto) (*d.Data, error) {
 
 	// Start log
 	pl.SetLogInfo(&event, input, "started", "create")
+
+	// validate unit_code
+	_, err := strconv.Atoi(input.Code)
+	if err != nil {
+		event.Status = "failed"
+		event.ErrInfo = pl.ErrorInfo{
+			Code:   "invalid_code_format",
+			Detail: "unit_code must be a valid integer",
+			Raw:    errors.New("invalid unit_code format"),
+		}
+		return nil, pl.SetLogError(&event, input)
+	}
+
 	mwRunner := newMiddlewareRunner(&event)
 
-	err := dg.I.Transaction(func(tx *gorm.DB) error {
+	err = dg.I.Transaction(func(tx *gorm.DB) error {
 		if resData, err := CreateData(input, &event, tx); err != nil {
 			return err
 		} else {