mirror of
https://github.com/AppFlowy-IO/AppFlowy.git
synced 2025-07-13 12:06:03 +00:00
46a3006f4d
* chore: remove script * chore: remove script * chore: refactor test * chore: clippy * chore: fix test * chore: fix test * chore: fmt
88 lines
2.5 KiB
Rust
88 lines
2.5 KiB
Rust
use std::sync::Arc;
|
|
|
|
use crate::database::calculations_test::script::DatabaseCalculationTest;
|
|
use collab_database::fields::Field;
|
|
use flowy_database2::entities::{CalculationType, FieldType, UpdateCalculationChangesetPB};
|
|
|
|
#[tokio::test]
|
|
async fn calculations_test() {
|
|
let mut test = DatabaseCalculationTest::new().await;
|
|
|
|
let expected_sum = 25.00000;
|
|
let expected_min = 1.00000;
|
|
let expected_average = 5.00000;
|
|
let expected_max = 14.00000;
|
|
let expected_median = 3.00000;
|
|
|
|
let view_id = &test.view_id();
|
|
let number_fields = test
|
|
.fields
|
|
.clone()
|
|
.into_iter()
|
|
.filter(|field| field.field_type == FieldType::Number as i64)
|
|
.collect::<Vec<Arc<Field>>>();
|
|
let field_id = &number_fields.first().unwrap().id;
|
|
|
|
let calculation_id = "calc_id".to_owned();
|
|
|
|
// Insert Sum calculation and assert its value
|
|
test
|
|
.insert_calculation(UpdateCalculationChangesetPB {
|
|
view_id: view_id.to_owned(),
|
|
field_id: field_id.to_owned(),
|
|
calculation_id: Some(calculation_id.clone()),
|
|
calculation_type: CalculationType::Sum,
|
|
})
|
|
.await;
|
|
|
|
test.assert_calculation_value(expected_sum).await;
|
|
|
|
// Insert Min calculation and assert its value
|
|
test
|
|
.insert_calculation(UpdateCalculationChangesetPB {
|
|
view_id: view_id.to_owned(),
|
|
field_id: field_id.to_owned(),
|
|
calculation_id: Some(calculation_id.clone()),
|
|
calculation_type: CalculationType::Min,
|
|
})
|
|
.await;
|
|
|
|
test.assert_calculation_value(expected_min).await;
|
|
|
|
// Insert Average calculation and assert its value
|
|
test
|
|
.insert_calculation(UpdateCalculationChangesetPB {
|
|
view_id: view_id.to_owned(),
|
|
field_id: field_id.to_owned(),
|
|
calculation_id: Some(calculation_id.clone()),
|
|
calculation_type: CalculationType::Average,
|
|
})
|
|
.await;
|
|
|
|
test.assert_calculation_value(expected_average).await;
|
|
|
|
// Insert Max calculation and assert its value
|
|
test
|
|
.insert_calculation(UpdateCalculationChangesetPB {
|
|
view_id: view_id.to_owned(),
|
|
field_id: field_id.to_owned(),
|
|
calculation_id: Some(calculation_id.clone()),
|
|
calculation_type: CalculationType::Max,
|
|
})
|
|
.await;
|
|
|
|
test.assert_calculation_value(expected_max).await;
|
|
|
|
// Insert Median calculation and assert its value
|
|
test
|
|
.insert_calculation(UpdateCalculationChangesetPB {
|
|
view_id: view_id.to_owned(),
|
|
field_id: field_id.to_owned(),
|
|
calculation_id: Some(calculation_id.clone()),
|
|
calculation_type: CalculationType::Median,
|
|
})
|
|
.await;
|
|
|
|
test.assert_calculation_value(expected_median).await;
|
|
}
|