From 63366cd64d68565769bf528561820c6ca0d636cd Mon Sep 17 00:00:00 2001 From: Alex Lusk Date: Tue, 4 Mar 2025 22:45:29 +0000 Subject: [PATCH] primev3 move_task_order handler test fixes --- pkg/handlers/primeapiv3/move_task_order_test.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/pkg/handlers/primeapiv3/move_task_order_test.go b/pkg/handlers/primeapiv3/move_task_order_test.go index 06880f84230..e118c808c4b 100644 --- a/pkg/handlers/primeapiv3/move_task_order_test.go +++ b/pkg/handlers/primeapiv3/move_task_order_test.go @@ -32,8 +32,12 @@ func (suite *HandlerSuite) TestGetMoveTaskOrder() { verifyAddressFields := func(address *models.Address, payload *primev3messages.Address) { suite.Equal(address.ID.String(), payload.ID.String()) suite.Equal(address.StreetAddress1, *payload.StreetAddress1) - suite.Equal(*address.StreetAddress2, *payload.StreetAddress2) - suite.Equal(*address.StreetAddress3, *payload.StreetAddress3) + if address.StreetAddress2 != nil { + suite.Equal(*address.StreetAddress2, *payload.StreetAddress2) + } + if address.StreetAddress3 != nil { + suite.Equal(*address.StreetAddress3, *payload.StreetAddress3) + } suite.Equal(address.City, *payload.City) suite.Equal(address.State, *payload.State) suite.Equal(address.PostalCode, *payload.PostalCode) @@ -1484,6 +1488,7 @@ func (suite *HandlerSuite) TestGetMoveTaskOrder() { PrimeEstimatedWeightRecordedDate: &aWeekAgo, RequiredDeliveryDate: &nowDate, ScheduledDeliveryDate: &nowDate, + MarketCode: models.MarketCodeInternational, }, }, {