use openlark_core::{
SDKResult,
api::{ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
};
use crate::common::{api_endpoints::DriveApi, api_utils::*};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize)]
pub struct CreateFolderRequest {
#[serde(skip)]
config: Config,
pub name: String,
pub folder_token: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateFolderResponse {
pub token: String,
pub url: String,
}
impl ApiResponseTrait for CreateFolderResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl CreateFolderRequest {
pub fn new(config: Config, name: impl Into<String>, folder_token: impl Into<String>) -> Self {
Self {
config,
name: name.into(),
folder_token: folder_token.into(),
}
}
pub async fn execute(self) -> SDKResult<CreateFolderResponse> {
self.execute_with_options(openlark_core::req_option::RequestOption::default())
.await
}
pub async fn execute_with_options(
self,
option: openlark_core::req_option::RequestOption,
) -> SDKResult<CreateFolderResponse> {
let name_len = self.name.len();
if name_len == 0 || name_len > 256 {
return Err(openlark_core::error::validation_error(
"name",
"name 长度必须在 1~256 字节之间",
));
}
let api_endpoint = DriveApi::CreateFolder;
let request = api_endpoint
.to_request::<CreateFolderResponse>()
.body(serialize_params(&self, "新建文件夹")?);
let response = Transport::request(request, &self.config, Some(option)).await?;
extract_response_data(response, "新建文件夹")
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_create_folder_request() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "test_folder", "parent_token");
assert_eq!(request.name, "test_folder");
assert_eq!(request.folder_token, "parent_token");
}
#[test]
fn test_create_folder_with_empty_name() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "", "parent_token");
assert_eq!(request.name, "");
}
#[test]
fn test_create_folder_with_too_long_name() {
let config = Config::default();
let long_name = "a".repeat(257); let request = CreateFolderRequest::new(config, long_name, "parent_token");
assert!(request.name.len() > 256);
}
#[test]
fn test_create_folder_with_max_length_name() {
let config = Config::default();
let max_name = "a".repeat(256); let request = CreateFolderRequest::new(config, max_name.clone(), "parent_token");
assert_eq!(request.name, max_name);
assert_eq!(request.name.len(), 256);
}
#[test]
fn test_create_folder_in_root() {
let config = Config::default();
let request = CreateFolderRequest::new(config, "test_folder", "");
assert_eq!(request.folder_token, "");
}
#[tokio::test]
async fn test_create_folder_returns_data_on_success_440() {
use crate::common::test_utils::tenant_test_transport;
use serde_json::json;
use wiremock::matchers::{header, method, path};
use wiremock::{Mock, ResponseTemplate};
let (server, config, option) = tenant_test_transport().await;
Mock::given(method("POST"))
.and(path("/open-apis/drive/v1/files/create_folder"))
.and(header("Authorization", "Bearer test-tenant-token"))
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
"code": 0,
"msg": "success",
"data": { "token": "new_folder_token", "url": "https://example.com/folder" }
})))
.mount(&server)
.await;
let resp = CreateFolderRequest::new(config, "测试文件夹", "parent_token_001")
.execute_with_options(option)
.await
.expect("新建文件夹应成功");
assert_eq!(resp.token, "new_folder_token");
let received = server.received_requests().await.unwrap_or_default();
assert_eq!(received.len(), 1);
assert_eq!(received[0].method, "POST");
assert_eq!(
received[0].url.path(),
"/open-apis/drive/v1/files/create_folder"
);
assert_eq!(
received[0]
.headers
.get("authorization")
.and_then(|h| h.to_str().ok()),
Some("Bearer test-tenant-token")
);
let body: serde_json::Value =
serde_json::from_slice(&received[0].body).expect("请求体应为合法 JSON");
assert_eq!(body["name"], "测试文件夹");
assert_eq!(body["folder_token"], "parent_token_001");
}
#[test]
fn test_create_folder_url() {
let ep = DriveApi::CreateFolder;
assert!(ep.to_url().contains("create_folder"));
}
}