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, }, }, {