diff --git a/.github/workflows/develop.yml b/.github/workflows/develop.yml index f2003ccf2c..50f1972dc6 100644 --- a/.github/workflows/develop.yml +++ b/.github/workflows/develop.yml @@ -224,6 +224,7 @@ jobs: GT_S3_BUCKET: ${{ secrets.S3_BUCKET }} GT_S3_ACCESS_KEY_ID: ${{ secrets.S3_ACCESS_KEY_ID }} GT_S3_ACCESS_KEY: ${{ secrets.S3_ACCESS_KEY }} + GT_S3_REGION: ${{ secrets.S3_REGION }} UNITTEST_LOG_DIR: "__unittest_logs" - name: Codecov upload uses: codecov/codecov-action@v2 diff --git a/src/frontend/src/tests/instance_test.rs b/src/frontend/src/tests/instance_test.rs index 5d1551e66e..fa7325544b 100644 --- a/src/frontend/src/tests/instance_test.rs +++ b/src/frontend/src/tests/instance_test.rs @@ -1123,11 +1123,12 @@ async fn test_execute_copy_to_s3(instance: Arc) { assert!(matches!(output, Output::AffectedRows(2))); let key_id = env::var("GT_S3_ACCESS_KEY_ID").unwrap(); let key = env::var("GT_S3_ACCESS_KEY").unwrap(); + let region = env::var("GT_S3_REGION").unwrap(); let root = uuid::Uuid::new_v4().to_string(); // exports - let copy_to_stmt = format!("Copy demo TO 's3://{}/{}/export/demo.parquet' CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}')", bucket, root, key_id, key); + let copy_to_stmt = format!("Copy demo TO 's3://{}/{}/export/demo.parquet' CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}',REGION='{}')", bucket, root, key_id, key, region); let output = execute_sql(&instance, ©_to_stmt).await; assert!(matches!(output, Output::AffectedRows(2))); @@ -1163,8 +1164,9 @@ async fn test_execute_copy_from_s3(instance: Arc) { let root = uuid::Uuid::new_v4().to_string(); let key_id = env::var("GT_S3_ACCESS_KEY_ID").unwrap(); let key = env::var("GT_S3_ACCESS_KEY").unwrap(); + let region = env::var("GT_S3_REGION").unwrap(); - let copy_to_stmt = format!("Copy demo TO 's3://{}/{}/export/demo.parquet' CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}')", bucket, root, key_id, key); + let copy_to_stmt = format!("Copy demo TO 's3://{}/{}/export/demo.parquet' CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}',REGION='{}')", bucket, root, key_id, key, region); let output = execute_sql(&instance, ©_to_stmt).await; assert!(matches!(output, Output::AffectedRows(2))); @@ -1205,8 +1207,8 @@ async fn test_execute_copy_from_s3(instance: Arc) { ) .await; let sql = format!( - "{} CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}')", - test.sql, key_id, key + "{} CONNECTION (ACCESS_KEY_ID='{}',SECRET_ACCESS_KEY='{}',REGION='{}')", + test.sql, key_id, key, region, ); logging::info!("Running sql: {}", sql);