test: Fix testing infrastructure for Neptune

.github/workflows/cfn-nag.yml

@@ -42,11 +42,20 @@ jobs:
           npm install -g aws-cdk
           cdk --version
       - uses: actions/checkout@v4
-      - name: Set up Python ${{ matrix.python-version }}
+      - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: 3.11
       - name: Install Requirements
+        run: |
+          cd test_infra
+          python -m pip install --upgrade pip
+          python -m pip install poetry
+          poetry env use python
+          poetry env info
+          source $(poetry env info --path)/bin/activate
+          poetry install -vvv
+      - name: Set up cdk.json
         run: |
           cd test_infra
           cat <<EOT >> cdk.context.json
@@ -61,12 +70,8 @@ jobs:
             ]
           }
           EOT
-          python -m pip install --upgrade pip
-          python -m pip install poetry
-          poetry env use python
-          poetry env info
-          source $(poetry env info --path)/bin/activate
-          poetry install -vvv
+          cat cdk.json | jq -r '.context.databases.neptune = true' | jq -r '.context.databases.oracle = true' | jq -r '.context.databases.sqlserver = true' > overwrite.cdk.json
+          rm cdk.json && mv overwrite.cdk.json cdk.json
       - name: CDK Synth
         run: |
           cd test_infra

test_infra/pyproject.toml

@@ -9,4 +9,5 @@ license = "Apache License 2.0"
 python = ">=3.8, <4.0"
 "aws-cdk-lib" = "^2.128.0"
 "aws-cdk.aws-glue-alpha" = "^2.128.0a0"
+"aws-cdk.aws-neptune-alpha" = "^2.128.0a0"
 "aws-cdk.aws-redshift-alpha" = "^2.128.0a0"

test_infra/stacks/databases_stack.py

@@ -5,7 +5,7 @@
 from aws_cdk import aws_glue_alpha as glue
 from aws_cdk import aws_iam as iam
 from aws_cdk import aws_kms as kms
-from aws_cdk import aws_neptune as neptune
+from aws_cdk import aws_neptune_alpha as neptune
 from aws_cdk import aws_rds as rds
 from aws_cdk import aws_redshift_alpha as redshift
 from aws_cdk import aws_redshiftserverless as redshiftserverless