mithril_common/era/adapters/
file.rs1use async_trait::async_trait;
2use std::{fs, path::PathBuf};
3
4use crate::era::{EraMarker, EraReaderAdapter};
5use crate::StdResult;
6
7pub struct FileAdapter {
10 markers_file: PathBuf,
11}
12
13impl FileAdapter {
14 pub fn new(markers_file: PathBuf) -> Self {
16 Self { markers_file }
17 }
18}
19
20#[async_trait]
21impl EraReaderAdapter for FileAdapter {
22 async fn read(&self) -> StdResult<Vec<EraMarker>> {
23 Ok(serde_json::from_str(&fs::read_to_string(
24 &self.markers_file,
25 )?)?)
26 }
27}
28
29#[cfg(test)]
30mod tests {
31 use crate::entities::Epoch;
32 use crate::era::SupportedEra;
33 use crate::test_utils::TempDir;
34
35 use super::*;
36
37 fn get_temp_dir(dir_name: &str) -> PathBuf {
38 TempDir::create("era-adapter", dir_name)
39 }
40
41 #[tokio::test]
42 async fn file_adapter_output() {
43 let markers = vec![
44 EraMarker::new("one", Some(Epoch(1))),
45 EraMarker::new(&SupportedEra::dummy().to_string(), None),
46 EraMarker::new(&SupportedEra::dummy().to_string(), Some(Epoch(10))),
47 ];
48 let temp_file = get_temp_dir("era_reader_file_adapter").join("markers.json");
49 fs::write(
50 &temp_file,
51 serde_json::to_string(&markers).expect("serializing markers should not fail"),
52 )
53 .expect("writing markers to file should not fail");
54 let adapter = FileAdapter::new(temp_file);
55
56 assert_eq!(
57 markers,
58 adapter
59 .read()
60 .await
61 .expect("file adapter shall not fail reading")
62 );
63 }
64}