B-22056 - fix tests after api change.

transcom · Jan 28, 2025 · 55767df · 55767df
1 parent 39cf19d
commit 55767df
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/handlers/routing/ghcapi_test/uploads_test.go b/pkg/handlers/routing/ghcapi_test/uploads_test.go
@@ -6,6 +6,7 @@ import (
 
 	"github.com/transcom/mymove/pkg/factory"
 	"github.com/transcom/mymove/pkg/models"
+	"github.com/transcom/mymove/pkg/models/roles"
 	storageTest "github.com/transcom/mymove/pkg/storage/test"
 	"github.com/transcom/mymove/pkg/uploader"
 )
@@ -31,7 +32,9 @@ func (suite *GhcAPISuite) TestUploads() {
 		_, err := suite.HandlerConfig().FileStorer().Store(uploadUser1.Upload.StorageKey, file.Data, "somehash", nil)
 		suite.NoError(err)
 
-		req := suite.NewAuthenticatedMilRequest("GET", "/internal/uploads/"+uploadUser1.Upload.ID.String()+"/status", nil, orders.ServiceMember)
+		officeUser := factory.BuildOfficeUserWithRoles(suite.DB(), factory.GetTraitActiveOfficeUser(),
+			[]roles.RoleType{roles.RoleTypeTOO})
+		req := suite.NewAuthenticatedOfficeRequest("GET", "/ghc/v1/uploads/"+uploadUser1.Upload.ID.String()+"/status", nil, officeUser)
 		rr := httptest.NewRecorder()
 
 		suite.SetupSiteHandler().ServeHTTP(rr, req)
@@ -60,7 +63,9 @@ func (suite *GhcAPISuite) TestUploads() {
 		_, err := suite.HandlerConfig().FileStorer().Store(uploadUser1.Upload.StorageKey, file.Data, "somehash", nil)
 		suite.NoError(err)
 
-		req := suite.NewAuthenticatedMilRequest("GET", "/internal/uploads/"+uploadUser1.Upload.ID.String()+"/status", nil, orders.ServiceMember)
+		officeUser := factory.BuildOfficeUserWithRoles(suite.DB(), factory.GetTraitActiveOfficeUser(),
+			[]roles.RoleType{roles.RoleTypeTOO})
+		req := suite.NewAuthenticatedOfficeRequest("GET", "/ghc/v1/uploads/"+uploadUser1.Upload.ID.String()+"/status", nil, officeUser)
 		rr := httptest.NewRecorder()
 
 		fakeS3, ok := suite.HandlerConfig().FileStorer().(*storageTest.FakeS3Storage)