test(rust_actix_codegen): add multipart tests

This commit is contained in:
Tanish2002
2024-03-12 18:06:03 +05:30
parent f6c8c9addf
commit cdc86b57fb
2 changed files with 489 additions and 5 deletions

View File

@ -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] =

View File

@ -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', () {