From b47be7a41cd6d0e93cb8dc99e59269aa25731fc6 Mon Sep 17 00:00:00 2001
From: everpcpc <git@everpcpc.com>
Date: Mon, 8 Jul 2024 15:30:12 +0800
Subject: [PATCH] chore(ci): add license for integration test

---
 .github/workflows/ci.yml  | 15 +++++++++++++++
 tests/docker-compose.yaml |  9 +++------
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d3b29348..0679f377 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -50,12 +50,27 @@ jobs:
 
   integration:
     needs: check
+    permissions:
+      contents: read
+      id-token: write
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
       - uses: ./.github/actions/setup
         with:
           cache-key: integration
+      - name: Configure AWS credentials
+        uses: aws-actions/configure-aws-credentials@v4
+        with:
+          role-duration-seconds: 600
+          role-to-assume: arn:aws:iam::034080348588:role/databend-ci
+          aws-region: us-east-2
+      - name: Get License from S3
+        run: |
+          aws s3 cp s3://databend-ci/misc/license.key license.key
+          aws s3 cp s3://databend-ci/misc/license.json license.json
+          cat license.json
+          echo "QUERY_DATABEND_ENTERPRISE_LICENSE=$(cat license.key)" >> $GITHUB_ENV
       - run: make -C tests test-core
       - run: make -C tests test-driver
       - run: make -C tests test-bendsql
diff --git a/tests/docker-compose.yaml b/tests/docker-compose.yaml
index 69a10bc1..dc576854 100644
--- a/tests/docker-compose.yaml
+++ b/tests/docker-compose.yaml
@@ -1,4 +1,4 @@
-version: '3'
+version: "3"
 services:
   minio:
     image: docker.io/minio/minio
@@ -7,13 +7,10 @@ services:
     volumes:
       - ./data:/data
   databend:
-    # Use nightly image for testing
-    #
-    # Please remove the nightly tag once https://github.com/datafuselabs/databend/pull/15559 included
-    # in stable.
-    image: docker.io/datafuselabs/databend:nightly
+    image: docker.io/datafuselabs/databend
     environment:
       - QUERY_STORAGE_TYPE=s3
+      - QUERY_DATABEND_ENTERPRISE_LICENSE
       - AWS_S3_ENDPOINT=http://localhost:9000
       - AWS_ACCESS_KEY_ID=minioadmin
       - AWS_SECRET_ACCESS_KEY=minioadmin