try to generate from workbook_dto

Author: Matéo Fernandez

tests/avro.rs

@@ -1,12 +1,18 @@
+use excel_gen::excel::{WorkbookDto, WorksheetDto};
+
 mod common;
 
 #[test]
 fn should_create_excel_file() {
-    let file = common::create_avro_from_json(include_str!("./input/without_format.json"));
-    println!("test {:?}", file)
-}
+    let workbook = WorkbookDto::builder()
+        .worksheets(vec![
+            WorksheetDto::builder()
+                .name("Feuille 1")
+                .autofit(true)
+                .build(),
+            WorksheetDto::builder().autofit(false).build(),
+        ])
+        .build();
 
-#[test]
-fn should_create_excel_file_with_format() {
-    // TODO
+    common::create_avro_from_workbook("excel_01", workbook);
 }

tests/common/mod.rs

@@ -1,15 +1,15 @@
 use apache_avro::Writer;
 use excel_gen::{excel::WorkbookDto, parser::avro::AvroParser};
+use std::fs::File;
 
-#[allow(dead_code)]
-pub fn create_avro_from_json(json: &str) -> Vec<u8> {
+pub fn create_avro_from_workbook(name: &str, workbook: WorkbookDto) {
     let avro_parser = AvroParser::new().unwrap();
     let schema = avro_parser.get_schema();
 
-    let workbook_dto: WorkbookDto = serde_json::from_str(json).unwrap();
+    let file = File::create(format!("tests/output/{}.avro", name)).unwrap();
+    let mut writer = Writer::new(schema, file);
 
-    let mut writer = Writer::new(schema, Vec::new());
-    writer.append_ser(workbook_dto).unwrap();
+    writer.append_ser(workbook).unwrap();
 
-    writer.into_inner().unwrap()
+    writer.flush().unwrap();
 }