use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required,
};
use serde::{Deserialize, Serialize};
use crate::ccm::wiki::v2::models::WikiSpaceNode;
use crate::common::api_endpoints::WikiApiV2;
pub struct GetWikiSpaceNodeRequest {
config: Config,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetWikiSpaceNodeParams {
pub token: String,
pub obj_type: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetWikiSpaceNodeResponse {
pub node: Option<WikiSpaceNode>,
}
impl ApiResponseTrait for GetWikiSpaceNodeResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl GetWikiSpaceNodeRequest {
pub fn new(config: Config) -> Self {
Self { config }
}
pub async fn execute(
self,
params: GetWikiSpaceNodeParams,
) -> SDKResult<GetWikiSpaceNodeResponse> {
self.execute_with_options(params, RequestOption::default())
.await
}
pub async fn execute_with_options(
self,
params: GetWikiSpaceNodeParams,
option: RequestOption,
) -> SDKResult<GetWikiSpaceNodeResponse> {
validate_required!(params.token, "token不能为空");
let api_endpoint = WikiApiV2::SpaceGetNode;
let mut api_request: ApiRequest<GetWikiSpaceNodeResponse> = api_endpoint.to_request();
api_request = api_request.query("token", ¶ms.token);
if let Some(obj_type) = params.obj_type {
api_request = api_request.query("obj_type", &obj_type);
}
Transport::request_typed(
api_request,
&self.config,
Some(option),
"获取知识空间节点信息",
)
.await
}
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::json;
use wiremock::MockServer;
use wiremock::matchers::{method, path};
use wiremock::{Mock, ResponseTemplate};
#[tokio::test]
async fn test_get_wiki_space_node_returns_data_on_success() {
let server = MockServer::start().await;
Mock::given(method("GET"))
.and(path("/open-apis/wiki/v2/spaces/get_node"))
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
"code": 0, "msg": "success", "data": {}
})))
.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 = GetWikiSpaceNodeRequest::new(config)
.execute(GetWikiSpaceNodeParams {
token: "nt001".into(),
obj_type: None,
})
.await
.expect("获取知识空间节点应成功");
assert!(resp.node.is_none());
let received = server.received_requests().await.unwrap_or_default();
assert_eq!(received.len(), 1);
assert_eq!(received[0].url.path(), "/open-apis/wiki/v2/spaces/get_node");
}
}