test: Fix s3 region in test (#1493)

This commit is contained in:
Yingwen
2023-04-27 20:25:20 +08:00
committed by GitHub
parent a77a4a4bd1
commit b1f7ad097a
3 changed files with 9 additions and 0 deletions

View File

@@ -3,6 +3,7 @@ GT_S3_BUCKET=S3 bucket
GT_S3_ACCESS_KEY_ID=S3 access key id
GT_S3_ACCESS_KEY=S3 secret access key
GT_S3_ENDPOINT_URL=S3 endpoint url
GT_S3_REGION=S3 region
# Settings for oss test
GT_OSS_BUCKET=OSS bucket
GT_OSS_ACCESS_KEY_ID=OSS access key id

View File

@@ -51,6 +51,7 @@ fn new_object_store(store_dir: &str, s3_bucket: Option<String>) -> ObjectStore {
.root(&root)
.access_key_id(&env::var("GT_S3_ACCESS_KEY_ID").unwrap())
.secret_access_key(&env::var("GT_S3_ACCESS_KEY").unwrap())
.region(&env::var("GT_S3_REGION").unwrap())
.bucket(&bucket);
return ObjectStore::new(builder).unwrap().finish();

View File

@@ -151,6 +151,13 @@ fn get_test_store_config(
.secret_access_key(&s3_config.secret_access_key)
.bucket(&s3_config.bucket);
if s3_config.endpoint.is_some() {
builder.endpoint(s3_config.endpoint.as_ref().unwrap());
}
if s3_config.region.is_some() {
builder.region(s3_config.region.as_ref().unwrap());
}
let config = ObjectStoreConfig::S3(s3_config);
let store = ObjectStore::new(builder).unwrap().finish();