mirror of
https://github.com/foss42/apidash.git
synced 2025-06-01 23:45:19 +08:00
test(rust_actix_codegen): add multipart tests
This commit is contained in:
@ -53,11 +53,11 @@ multipart/form-data; boundary={{boundary}}''';
|
|||||||
}
|
}
|
||||||
|
|
||||||
let form_data_items: Vec<FormDataItem> = vec![
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
{%- for formitem in fields_list %}
|
{%- for formitem in fields_list %}
|
||||||
FormDataItem {
|
FormDataItem {
|
||||||
{%- for key, val in formitem %}
|
{%- for key, val in formitem %}
|
||||||
{% if key == "type" %}field_type: "{{ val }}".to_string(),{% else %}{{ key }}: "{{ val }}".to_string(),{% endif %}
|
{% if key == "type" %}field_type: "{{ val }}".to_string(),{% else %}{{ key }}: "{{ val }}".to_string(),{% endif %}
|
||||||
{%- endfor %}
|
{%- endfor %}
|
||||||
},
|
},
|
||||||
{%- endfor %}
|
{%- endfor %}
|
||||||
];
|
];
|
||||||
@ -116,6 +116,7 @@ multipart/form-data; boundary={{boundary}}''';
|
|||||||
String? boundary,
|
String? boundary,
|
||||||
}) {
|
}) {
|
||||||
try {
|
try {
|
||||||
|
String uuid = getNewUuid();
|
||||||
String result = "";
|
String result = "";
|
||||||
bool hasBody = false;
|
bool hasBody = false;
|
||||||
bool hasJsonBody = false;
|
bool hasJsonBody = false;
|
||||||
@ -157,7 +158,7 @@ multipart/form-data; boundary={{boundary}}''';
|
|||||||
result += formDataBodyData.render(
|
result += formDataBodyData.render(
|
||||||
{
|
{
|
||||||
"fields_list": requestModel.formDataMapList,
|
"fields_list": requestModel.formDataMapList,
|
||||||
"boundary": boundary ?? getNewUuid(),
|
"boundary": boundary ?? uuid,
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@ -179,13 +180,13 @@ multipart/form-data; boundary={{boundary}}''';
|
|||||||
}
|
}
|
||||||
|
|
||||||
var headersList = requestModel.enabledRequestHeaders;
|
var headersList = requestModel.enabledRequestHeaders;
|
||||||
if (headersList != null || hasBody || requestModel.hasTextData) {
|
if (headersList != null || hasBody || requestModel.hasFormData) {
|
||||||
var headers = requestModel.enabledHeadersMap;
|
var headers = requestModel.enabledHeadersMap;
|
||||||
if (requestModel.hasFormData) {
|
if (requestModel.hasFormData) {
|
||||||
var formHeaderTemplate =
|
var formHeaderTemplate =
|
||||||
jj.Template(kTemplateFormHeaderContentType);
|
jj.Template(kTemplateFormHeaderContentType);
|
||||||
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
|
headers[HttpHeaders.contentTypeHeader] = formHeaderTemplate.render({
|
||||||
"boundary": boundary ?? getNewUuid(),
|
"boundary": boundary ?? uuid,
|
||||||
});
|
});
|
||||||
} else if (hasBody) {
|
} else if (hasBody) {
|
||||||
headers[HttpHeaders.contentTypeHeader] =
|
headers[HttpHeaders.contentTypeHeader] =
|
||||||
|
@ -483,6 +483,489 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|||||||
CodegenLanguage.rustActix, requestModelPost3, "https"),
|
CodegenLanguage.rustActix, requestModelPost3, "https"),
|
||||||
expectedCode);
|
expectedCode);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
test('POST 4', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/form";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "text".to_string(),
|
||||||
|
value: "API".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "sep".to_string(),
|
||||||
|
value: "|".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "times".to_string(),
|
||||||
|
value: "3".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(
|
||||||
|
CodegenLanguage.rustActix,
|
||||||
|
requestModelPost4,
|
||||||
|
"https",
|
||||||
|
boundary: "test",
|
||||||
|
),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
|
test('POST 5', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/form";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "text".to_string(),
|
||||||
|
value: "API".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "sep".to_string(),
|
||||||
|
value: "|".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "times".to_string(),
|
||||||
|
value: "3".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.insert_header(("User-Agent", "Test Agent"))
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(CodegenLanguage.rustActix, requestModelPost5, "https",
|
||||||
|
boundary: "test"),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
|
test('POST 6', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/img";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "token".to_string(),
|
||||||
|
value: "xyz".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "imfile".to_string(),
|
||||||
|
value: "/Documents/up/1.png".to_string(),
|
||||||
|
field_type: "file".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(CodegenLanguage.rustActix, requestModelPost6, "https",
|
||||||
|
boundary: "test"),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
|
test('POST 7', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/img";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "token".to_string(),
|
||||||
|
value: "xyz".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "imfile".to_string(),
|
||||||
|
value: "/Documents/up/1.png".to_string(),
|
||||||
|
field_type: "file".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(CodegenLanguage.rustActix, requestModelPost7, "https",
|
||||||
|
boundary: "test"),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
|
test('POST 8', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/form";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "text".to_string(),
|
||||||
|
value: "API".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "sep".to_string(),
|
||||||
|
value: "|".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "times".to_string(),
|
||||||
|
value: "3".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.query(&[("size", "2"), ("len", "3")])
|
||||||
|
.unwrap()
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(CodegenLanguage.rustActix, requestModelPost8, "https",
|
||||||
|
boundary: "test"),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
|
test('POST 9', () {
|
||||||
|
const expectedCode = r"""use std::io::Read;
|
||||||
|
#[actix_rt::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
let url = "https://api.apidash.dev/io/img";
|
||||||
|
let client = awc::Client::default();
|
||||||
|
|
||||||
|
struct FormDataItem {
|
||||||
|
name: String,
|
||||||
|
value: String,
|
||||||
|
field_type: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
let form_data_items: Vec<FormDataItem> = vec![
|
||||||
|
FormDataItem {
|
||||||
|
name: "token".to_string(),
|
||||||
|
value: "xyz".to_string(),
|
||||||
|
field_type: "text".to_string(),
|
||||||
|
},
|
||||||
|
FormDataItem {
|
||||||
|
name: "imfile".to_string(),
|
||||||
|
value: "/Documents/up/1.png".to_string(),
|
||||||
|
field_type: "file".to_string(),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
fn build_data_list(fields: Vec<FormDataItem>) -> Vec<u8> {
|
||||||
|
let mut data_list = Vec::new();
|
||||||
|
|
||||||
|
for field in fields {
|
||||||
|
data_list.extend_from_slice(b"--test\r\n");
|
||||||
|
|
||||||
|
if field.field_type == "text" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"\r\n", field.name).as_bytes());
|
||||||
|
data_list.extend_from_slice(b"Content-Type: text/plain\r\n\r\n");
|
||||||
|
data_list.extend_from_slice(field.value.as_bytes());
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
} else if field.field_type == "file" {
|
||||||
|
data_list.extend_from_slice(format!("Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n", field.name, field.value).as_bytes());
|
||||||
|
|
||||||
|
let mime_type = mime_guess::from_path(&field.value).first_or(mime_guess::mime::APPLICATION_OCTET_STREAM);
|
||||||
|
data_list.extend_from_slice(format!("Content-Type: {}\r\n\r\n", mime_type).as_bytes());
|
||||||
|
|
||||||
|
let mut file = std::fs::File::open(&field.value).unwrap();
|
||||||
|
let mut file_contents = Vec::new();
|
||||||
|
file.read_to_end(&mut file_contents).unwrap();
|
||||||
|
data_list.extend_from_slice(&file_contents);
|
||||||
|
data_list.extend_from_slice(b"\r\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data_list.extend_from_slice(b"--test--\r\n");
|
||||||
|
data_list
|
||||||
|
}
|
||||||
|
|
||||||
|
let payload = build_data_list(form_data_items);
|
||||||
|
let mut response = client
|
||||||
|
.post(url)
|
||||||
|
.query(&[("size", "2"), ("len", "3")])
|
||||||
|
.unwrap()
|
||||||
|
.insert_header(("User-Agent", "Test Agent"))
|
||||||
|
.insert_header(("Keep-Alive", "true"))
|
||||||
|
.insert_header(("content-type", "multipart/form-data; boundary=test"))
|
||||||
|
.send_body(payload)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let body_bytes = response.body().await.unwrap();
|
||||||
|
let body = std::str::from_utf8(&body_bytes).unwrap();
|
||||||
|
println!("Response Status: {}", response.status());
|
||||||
|
println!("Response: {:?}", body);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
""";
|
||||||
|
expect(
|
||||||
|
codeGen.getCode(CodegenLanguage.rustActix, requestModelPost9, "https",
|
||||||
|
boundary: "test"),
|
||||||
|
expectedCode);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
group('PUT Request', () {
|
group('PUT Request', () {
|
||||||
|
Reference in New Issue
Block a user