Spaces:
Build error
Build error
File size: 1,951 Bytes
84d2a97 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
use segment::common::operation_error::{OperationError, SegmentFailedState};
use segment::data_types::vectors::only_default_vector;
use segment::entry::entry_point::SegmentEntry;
use serde_json::json;
use tempfile::Builder;
use crate::fixtures::segment::empty_segment;
#[test]
fn test_insert_fail_recovery() {
let dir = Builder::new().prefix("segment_dir").tempdir().unwrap();
let vec1 = vec![1.0, 0.0, 1.0, 1.0];
let mut segment = empty_segment(dir.path());
segment
.upsert_point(1, 1.into(), only_default_vector(&vec1))
.unwrap();
segment
.upsert_point(1, 2.into(), only_default_vector(&vec1))
.unwrap();
segment.error_status = Some(SegmentFailedState {
version: 2,
point_id: Some(1.into()),
error: OperationError::service_error("test error"),
});
// op_num is greater than errored. Skip because not recovered yet
let fail_res = segment.set_payload(
3,
1.into(),
&json!({ "color": vec!["red".to_string()] }).into(),
&None,
);
assert!(fail_res.is_err());
// Also skip even with another point operation
let fail_res = segment.set_payload(
3,
2.into(),
&json!({ "color": vec!["red".to_string()] }).into(),
&None,
);
assert!(fail_res.is_err());
// Perform operation, but keep error status: operation is not fully recovered yet
let ok_res = segment.set_payload(
2,
2.into(),
&json!({ "color": vec!["red".to_string()] }).into(),
&None,
);
assert!(ok_res.is_ok());
assert!(segment.error_status.is_some());
// Perform operation and recover the error - operation is fixed now
let recover_res = segment.set_payload(
2,
1.into(),
&json!({ "color": vec!["red".to_string()] }).into(),
&None,
);
assert!(recover_res.is_ok());
assert!(segment.error_status.is_none());
}
|