millipede-storage-memory 0.1.0

In-memory StorageClient implementation for the Millipede web crawler.
Documentation 
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

//! Integration tests for persisted typed state.

use millipede_core::storage::{AutoSaved, KeyValueStore};
use millipede_storage_memory::MemoryKeyValueStore;
use serde::{Deserialize, Serialize};
use std::sync::Arc;

#[derive(Clone, Debug, Default, Deserialize, PartialEq, Serialize)]
struct State {
    count: u32,
}

#[tokio::test]
async fn persisted_value_is_observed_by_a_second_wrapper() {
    let store: Arc<dyn KeyValueStore> = Arc::new(MemoryKeyValueStore::new("state"));
    let saved = AutoSaved::open(store.clone(), "counter", State::default())
        .await
        .unwrap();
    saved.update(|value| value.count += 3).await;
    saved.persist().await.unwrap();

    let reopened = AutoSaved::open(store.clone(), "counter", State::default())
        .await
        .unwrap();
    assert_eq!(reopened.get().await, State { count: 3 });
    assert_eq!(
        store
            .get_bytes("counter")
            .await
            .unwrap()
            .unwrap()
            .content_type,
        "application/json"
    );
}