Migrate reviewDecision of Airlock Reviews

api_app/api/routes/migrations.py

@@ -75,6 +75,10 @@ async def migrate_database(resources_repo=Depends(get_repository(ResourceReposit
         num_updated = await airlock_migration.change_review_resources_to_dict()
         migrations.append(Migration(issueNumber="XXXX", status=f'Updated {num_updated} airlock requests with new reviewUserResources format'))
 
+        logging.info("PR 3152 - Migrate reviewDecision of Airlock Reviews")
+        num_updated = await airlock_migration.update_review_decision_values()
+        migrations.append(Migration(issueNumber="3121", status=f'Updated {num_updated} airlock requests with new reviewDecision value'))
+
         return MigrationOutList(migrations=migrations)
     except Exception as e:
         logging.exception("Failed to migrate database")

api_app/db/migrations/airlock.py

@@ -1,4 +1,5 @@
 from azure.cosmos.aio import CosmosClient
+from resources import strings
 from db.repositories.airlock_requests import AirlockRequestRepository
 
 
@@ -52,3 +53,28 @@ async def change_review_resources_to_dict(self) -> int:
                     num_updated += 1
 
         return num_updated
+
+    async def update_review_decision_values(self) -> int:
+        num_updated = 0
+        for request in await self.query('SELECT * FROM c WHERE ARRAY_LENGTH(c.reviews) > 0'):
+            request_changed = False
+
+            for review in request['reviews']:
+                old_decision = review['reviewDecision']
+                new_decision = old_decision
+
+                if old_decision == 'approval_in_progress':
+                    new_decision = strings.AIRLOCK_REVIEW_DECISION_APPROVED
+
+                if old_decision == 'rejection_in_progress':
+                    new_decision = strings.AIRLOCK_REVIEW_DECISION_REJECTED
+
+                if new_decision != old_decision:
+                    request_changed = True
+                    review['reviewDecision'] = new_decision
+
+            if request_changed:
+                await self.update_item_dict(request)
+                num_updated += 1
+
+        return num_updated

api_app/tests_ma/test_api/test_routes/test_migrations.py

@@ -44,7 +44,8 @@ def _prepare(self, app, admin_user):
     @ patch("api.routes.migrations.AirlockMigration.add_created_by_and_rename_in_history")
     @ patch("api.routes.migrations.AirlockMigration.rename_field_name")
     @ patch("api.routes.migrations.AirlockMigration.change_review_resources_to_dict")
-    async def test_post_migrations_returns_202_on_successful(self, change_review_resources_to_dict, airlock_rename_field, add_created_by_and_rename_in_history,
+    @ patch("api.routes.migrations.AirlockMigration.update_review_decision_values")
+    async def test_post_migrations_returns_202_on_successful(self, update_review_decision_values, change_review_resources_to_dict, airlock_rename_field, add_created_by_and_rename_in_history,
                                                              check_min_firewall_version, workspace_migration, shared_services_migration,
                                                              rename_field, add_deployment_field, archive_history, _, logging, client, app):
         response = await client.post(app.url_path_for(strings.API_MIGRATE_DATABASE))
@@ -57,6 +58,7 @@ async def test_post_migrations_returns_202_on_successful(self, change_review_res
         add_created_by_and_rename_in_history.assert_called_once()
         airlock_rename_field.assert_called()
         change_review_resources_to_dict.assert_called_once()
+        update_review_decision_values.assert_called_once()
         archive_history.assert_called_once()
         logging.assert_called()
         assert response.status_code == status.HTTP_202_ACCEPTED

devops/scripts/migrate_state_store.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # This script migrates the Cosmos database based on any breaking changes that have occurred.
-# Cosmos is behind a private network, so we call the /migrate endpoint of the API
+# Cosmos is behind a private network, so we call the /migrations endpoint of the API
 
 set -o errexit
 set -o pipefail