use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required_list,
};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone)]
pub struct CountryRegionBatchGetRequestBuilder {
config: Config,
mdm_codes: Vec<String>,
}
impl CountryRegionBatchGetRequestBuilder {
pub fn new(config: Config) -> Self {
Self {
config,
mdm_codes: Vec::new(),
}
}
pub fn mdm_code(mut self, mdm_code: impl Into<String>) -> Self {
self.mdm_codes.push(mdm_code.into());
self
}
pub fn mdm_codes(mut self, mdm_codes: impl IntoIterator<Item = impl Into<String>>) -> Self {
self.mdm_codes.extend(mdm_codes.into_iter().map(Into::into));
self
}
pub async fn execute(self) -> SDKResult<CountryRegionBatchGetResponse> {
self.execute_with_options(RequestOption::default()).await
}
pub async fn execute_with_options(
self,
option: RequestOption,
) -> SDKResult<CountryRegionBatchGetResponse> {
let url = "/open-apis/mdm/v3/batch_country_region".to_string();
validate_required_list!(self.mdm_codes, 50, "mdm_codes 不能为空");
let req: ApiRequest<CountryRegionBatchGetResponse> = ApiRequest::get(&url);
Transport::request_typed(req, &self.config, Some(option), "Operation").await
}
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct CountryRegionBatchGetResponse {
pub items: Vec<CountryRegion>,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct CountryRegion {
#[serde(rename = "mdm_code")]
pub mdm_code: String,
pub name: String,
#[serde(rename = "i18n_name")]
pub i18n_name: Option<CountryRegionI18nName>,
#[serde(rename = "phone_code")]
pub phone_code: Option<String>,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct CountryRegionI18nName {
#[serde(rename = "zh_cn")]
pub zh_cn: Option<String>,
#[serde(rename = "en_us")]
pub en_us: Option<String>,
#[serde(rename = "ja_jp")]
pub ja_jp: Option<String>,
}
impl ApiResponseTrait for CountryRegionBatchGetResponse {}
#[deprecated(
note = "renamed to CountryRegionBatchGetRequestBuilder, will be removed in v1.0 (#271)"
)]
pub type CountryRegionBatchGetBuilder = CountryRegionBatchGetRequestBuilder;
#[cfg(test)]
mod tests {
use super::*;
#[tokio::test]
async fn test_batch_get_country_region_returns_data_on_success() {
use serde_json::json;
use wiremock::MockServer;
use wiremock::matchers::{method, path};
use wiremock::{Mock, ResponseTemplate};
let server = MockServer::start().await;
Mock::given(method("GET"))
.and(path("/open-apis/mdm/v3/batch_country_region"))
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
"code": 0,
"msg": "success",
"data": {
"items": [
{
"mdm_code": "CN",
"name": "中国",
"i18n_name": { "zh_cn": "中国", "en_us": "China" },
"phone_code": "86"
}
]
}
})))
.mount(&server)
.await;
let config = Config::builder()
.app_id("ci_app_id")
.app_secret("ci_app_secret")
.base_url(server.uri())
.enable_token_cache(false)
.build();
let resp = CountryRegionBatchGetRequestBuilder::new(config)
.mdm_code("CN")
.execute()
.await
.expect("批量查询国家/地区应成功");
assert_eq!(resp.items.len(), 1);
assert_eq!(resp.items[0].mdm_code, "CN");
assert_eq!(resp.items[0].name, "中国");
assert_eq!(resp.items[0].phone_code.as_deref(), Some("86"));
let received = server.received_requests().await.unwrap_or_default();
assert_eq!(received.len(), 1);
assert_eq!(
received[0].url.path(),
"/open-apis/mdm/v3/batch_country_region"
);
}
}