Function test_vector_update

src/grpc/vectors/tests.rs:120–156

()

#[tokio::test]
async fn test_vector_update() {
    let result = setup_with_timeout().await;
    assert!(result.is_ok(), "Setup timed out");
    let (service, _context, collection_id) = result.unwrap();

    timeout(Duration::from_secs(5), async {
        let create_req = CreateVectorRequest {
            collection_id: collection_id.clone(),
            vector: Some(Vector {
                id: 1,
                values: vec![0.1, 0.2, 0.3, 0.4],
            }),
        };
        service
            .create_vector(Request::new(create_req))
            .await
            .unwrap();
    })
    .await
    .unwrap();

    let update_result = timeout(Duration::from_secs(5), async {
        let update_req = UpdateVectorRequest {
            collection_id: collection_id.clone(),
            vector_id: 1,
            values: vec![0.5, 0.6, 0.7, 0.8],
        };
        service.update_vector(Request::new(update_req)).await
    })
    .await;

    assert!(update_result.is_ok(), "Update operation timed out");
    let update_response = update_result.unwrap().unwrap();
    assert_eq!(update_response.get_ref().values, vec![0.5, 0.6, 0.7, 0.8]);

    cleanup();
}

#[tokio::test]
async fn test_find_similar_vectors() {