How to unit test with a stage?

[Xuanwo]

I have done some refactoring on internal stage/location related logic. Now I want to do some unit tests for them to avoid them being broken by mistake.

pub async fn parse_stage_location(
    ctx: &Arc<QueryContext>,
    location: &str,
) -> Result<(UserStageInfo, String)> {}

pub fn parse_uri_location(
    location: &str,
    credential_options: &BTreeMap<String, String>,
    encryption_options: &BTreeMap<String, String>,
) -> Result<(UserStageInfo, String)> {}

However, parse_stage_location will call get_stage() to fetch UserStageInfo:

let mgr = ctx.get_user_manager();
let stage = mgr.get_stage(&ctx.get_tenant(), names[0]).await?;

How to construct a QueryContext with a test stage? Is it ok to use ctx.get_user_manager().add_stage()?

[Xuanwo]

ctx.get_user_manager().add_stage() works.

let ctx = create_query_context().await?;
let user_stage_info = UserStageInfo {
    stage_name: "test".to_string(),
    stage_type: StageType::Internal,
    ..Default::default()
};
// Add an internal stage for testing.
ctx.get_user_manager()
    .add_stage(&ctx.get_tenant(), user_stage_info.clone(), false)
    .await?;

[sundy-li]

ctx has a determined tenant which acts as the prefix of stage.

{tenent}/stage/