B-21337-MAIN Bulk assignment data fetch setup

pkg/factory/move_factory.go

@@ -1,6 +1,7 @@
 package factory
 
 import (
+	"fmt"
 	"log"
 	"time"
 
@@ -71,6 +72,8 @@ func BuildMove(db *pop.Connection, customs []Customization, traits []Trait) mode
 	if closeoutOfficeResult != nil {
 		move.CloseoutOffice = &closeoutOffice
 		move.CloseoutOfficeID = &closeoutOffice.ID
+		move.CounselingOffice = &closeoutOffice
+		move.CounselingOfficeID = &closeoutOffice.ID
 	}
 
 	// Overwrite values with those from assertions
@@ -181,9 +184,36 @@ func BuildMoveWithPPMShipment(db *pop.Connection, customs []Customization, trait
 	if db != nil {
 		mustSave(db, &move)
 	}
+	fmt.Println("move in factory")
+	fmt.Println(move.ID)
 
 	return move
 }
+func BuildAssignedMoveWithPPMShipment(db *pop.Connection, customs []Customization, traits []Trait) models.Move {
+	move := BuildMove(db, customs, traits)
+
+	mtoShipment := buildMTOShipmentWithBuildType(db, customs, traits, mtoShipmentPPM)
+	mtoShipment.MoveTaskOrder = move
+	mtoShipment.MoveTaskOrderID = move.ID
+
+	ppmShipment := BuildPPMShipment(db, customs, traits)
+	ppmShipment.ShipmentID = mtoShipment.ID
+
+	mtoShipment.PPMShipment = &ppmShipment
+	mtoShipment.ShipmentType = models.MTOShipmentTypePPM
+	move.MTOShipments = append(move.MTOShipments, mtoShipment)
+
+	officeUser := BuildOfficeUser(db, customs, traits)
+	move.SCAssignedUser = &officeUser
+	move.SCAssignedID = &officeUser.ID
+
+	if db != nil {
+		mustSave(db, &move)
+	}
+	fmt.Println("move")
+	fmt.Println(move.SCAssignedUser)
+	return move
+}
 
 // ------------------------
 //        TRAITS

pkg/factory/office_user_factory.go

@@ -37,19 +37,34 @@ func BuildOfficeUser(db *pop.Connection, customs []Customization, traits []Trait
 	// Find/create the user model
 	user := BuildUserAndUsersRoles(db, customs, nil)
 
+	var closeoutOffice models.TransportationOffice
+	tempCloseoutOfficeCustoms := customs
+	closeoutOfficeResult := findValidCustomization(customs, TransportationOffices.CloseoutOffice)
+	if closeoutOfficeResult != nil {
+		tempCloseoutOfficeCustoms = convertCustomizationInList(tempCloseoutOfficeCustoms, TransportationOffices.CloseoutOffice, TransportationOffice)
+		closeoutOffice = BuildTransportationOffice(db, tempCloseoutOfficeCustoms, nil)
+	}
 	// Find/create the TransportationOffice model
-	transportationOffice := BuildTransportationOffice(db, customs, nil)
+	basicTransportationOffice := BuildTransportationOffice(db, customs, nil)
 
 	// create officeuser
 	officeUser := models.OfficeUser{
-		UserID:                 &user.ID,
-		User:                   user,
-		FirstName:              "Leo",
-		LastName:               "Spaceman",
-		Email:                  "[email protected]",
-		Telephone:              "415-555-1212",
-		TransportationOffice:   transportationOffice,
-		TransportationOfficeID: transportationOffice.ID,
+		UserID:    &user.ID,
+		User:      user,
+		FirstName: "Leo",
+		LastName:  "Spaceman",
+		Email:     "[email protected]",
+		Telephone: "415-555-1212",
+		// TransportationOffice:   transportationOffice,
+		// TransportationOfficeID: transportationOffice.ID,
+	}
+
+	if closeoutOfficeResult != nil {
+		officeUser.TransportationOffice = closeoutOffice
+		officeUser.TransportationOfficeID = closeoutOffice.ID
+	} else {
+		officeUser.TransportationOffice = basicTransportationOffice
+		officeUser.TransportationOfficeID = basicTransportationOffice.ID
 	}
 	// Overwrite values with those from assertions
 	testdatagen.MergeModels(&officeUser, cOfficeUser)

pkg/factory/shared.go

@@ -252,13 +252,15 @@ type transportationOfficeGroup struct {
 	OriginDutyLocation CustomType
 	NewDutyLocation    CustomType
 	CloseoutOffice     CustomType
+	CounselingOffice   CustomType
 }
 
 // TransportationOffices is the struct to access the fields externally
 var TransportationOffices = transportationOfficeGroup{
 	OriginDutyLocation: "OriginDutyLocationTransportationOffice",
 	NewDutyLocation:    "NewDutyLocationTransportationOffice",
 	CloseoutOffice:     "CloseoutOffice",
+	CounselingOffice:   "CounselingOffice",
 }
 
 // uploadGroup is a grouping of all the upload related fields

pkg/gen/ghcapi/configure_mymove.go

@@ -221,6 +221,11 @@ func configureAPI(api *ghcoperations.MymoveAPI) http.Handler {
 			return middleware.NotImplemented("operation re_service_items.GetAllReServiceItems has not yet been implemented")
 		})
 	}
+	if api.QueuesGetBulkAssignmentDataHandler == nil {
+		api.QueuesGetBulkAssignmentDataHandler = queues.GetBulkAssignmentDataHandlerFunc(func(params queues.GetBulkAssignmentDataParams) middleware.Responder {
+			return middleware.NotImplemented("operation queues.GetBulkAssignmentData has not yet been implemented")
+		})
+	}
 	if api.CustomerGetCustomerHandler == nil {
 		api.CustomerGetCustomerHandler = customer.GetCustomerHandlerFunc(func(params customer.GetCustomerParams) middleware.Responder {
 			return middleware.NotImplemented("operation customer.GetCustomer has not yet been implemented")