Browse Source

fix(HC): Adds the postgres role override to org mapping updates (#51632)

Gabe Villalobos 1 year ago
parent
commit
58aa7cf441
1 changed files with 5 additions and 4 deletions
  1. 5 4
      src/sentry/services/hybrid_cloud/organization_mapping/impl.py

+ 5 - 4
src/sentry/services/hybrid_cloud/organization_mapping/impl.py

@@ -61,10 +61,11 @@ class DatabaseBackedOrganizationMappingService(OrganizationMappingService):
 
     def update(self, organization_id: int, update: RpcOrganizationMappingUpdate) -> None:
         # TODO: REMOVE FROM GETSENTRY!
-        try:
-            OrganizationMapping.objects.get(organization_id=organization_id).update(**update)
-        except OrganizationMapping.DoesNotExist:
-            pass
+        with in_test_psql_role_override("postgres"):
+            try:
+                OrganizationMapping.objects.get(organization_id=organization_id).update(**update)
+            except OrganizationMapping.DoesNotExist:
+                pass
 
     def upsert(
         self, organization_id: int, update: RpcOrganizationMappingUpdate