Martin Kavík
6 years ago
parent
2facd17653
commit
a39a278220
@ -0,0 +1,58 @@ |
|||||||
|
//! Key-value store in `*.stamps` file.
|
||||||
|
|
||||||
|
use failure::{self, ResultExt}; |
||||||
|
use std::{env, fs, path::PathBuf}; |
||||||
|
|
||||||
|
/// Get a value corresponding to the key from the JSON value.
|
||||||
|
///
|
||||||
|
/// You should use return value of function `read_stamps_file_to_json()` as `json` argument.
|
||||||
|
pub fn get_stamp_value( |
||||||
|
key: impl AsRef<str>, |
||||||
|
json: &serde_json::Value, |
||||||
|
) -> Result<String, failure::Error> { |
||||||
|
json.get(key.as_ref()) |
||||||
|
.and_then(|value| value.as_str().map(ToOwned::to_owned)) |
||||||
|
.ok_or_else(|| { |
||||||
|
failure::err_msg(format!("cannot get stamp value for key '{}'", key.as_ref())) |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
/// Save the key-value pair to the store.
|
||||||
|
pub fn save_stamp_value( |
||||||
|
key: impl Into<String>, |
||||||
|
value: impl AsRef<str>, |
||||||
|
) -> Result<(), failure::Error> { |
||||||
|
let mut json = read_stamps_file_to_json().unwrap_or_else(|_| serde_json::Map::new().into()); |
||||||
|
|
||||||
|
let stamps = json |
||||||
|
.as_object_mut() |
||||||
|
.ok_or_else(|| failure::err_msg("stamps file doesn't contain JSON object"))?; |
||||||
|
stamps.insert(key.into(), value.as_ref().into()); |
||||||
|
|
||||||
|
write_to_stamps_file(json) |
||||||
|
} |
||||||
|
|
||||||
|
/// Get the path of the `*.stamps` file that is used as the store.
|
||||||
|
pub fn get_stamps_file_path() -> Result<PathBuf, failure::Error> { |
||||||
|
let path = env::current_exe() |
||||||
|
.map(|path| path.with_extension("stamps")) |
||||||
|
.context("cannot get stamps file path")?; |
||||||
|
Ok(path) |
||||||
|
} |
||||||
|
|
||||||
|
/// Read `*.stamps` file and convert its content to the JSON value.
|
||||||
|
pub fn read_stamps_file_to_json() -> Result<serde_json::Value, failure::Error> { |
||||||
|
let stamps_file_path = get_stamps_file_path()?; |
||||||
|
let stamps_file_content = |
||||||
|
fs::read_to_string(stamps_file_path).context("cannot find or read stamps file")?; |
||||||
|
let json: serde_json::Value = serde_json::from_str(&stamps_file_content) |
||||||
|
.context("stamps file doesn't contain valid JSON")?; |
||||||
|
Ok(json) |
||||||
|
} |
||||||
|
|
||||||
|
fn write_to_stamps_file(json: serde_json::Value) -> Result<(), failure::Error> { |
||||||
|
let stamps_file_path = get_stamps_file_path()?; |
||||||
|
let pretty_json = serde_json::to_string_pretty(&json).context("JSON serialization failed")?; |
||||||
|
fs::write(stamps_file_path, pretty_json).context("cannot write to stamps file")?; |
||||||
|
Ok(()) |
||||||
|
} |
||||||
@ -0,0 +1,71 @@ |
|||||||
|
use std::{fs, panic}; |
||||||
|
use wasm_pack::stamps; |
||||||
|
|
||||||
|
fn run_test<T>(test: T) -> () |
||||||
|
where |
||||||
|
T: FnOnce() -> () + panic::UnwindSafe, |
||||||
|
{ |
||||||
|
before(); |
||||||
|
let result = panic::catch_unwind(|| test()); |
||||||
|
after(); |
||||||
|
assert!(result.is_ok()) |
||||||
|
} |
||||||
|
|
||||||
|
fn before() { |
||||||
|
remove_stamps_file() |
||||||
|
} |
||||||
|
|
||||||
|
fn after() { |
||||||
|
remove_stamps_file() |
||||||
|
} |
||||||
|
|
||||||
|
fn remove_stamps_file() { |
||||||
|
let stamps_file_path = stamps::get_stamps_file_path().unwrap(); |
||||||
|
if stamps_file_path.exists() { |
||||||
|
fs::remove_file(stamps_file_path).unwrap(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[test] |
||||||
|
#[should_panic] |
||||||
|
#[serial] |
||||||
|
fn load_stamp_from_non_existent_file() { |
||||||
|
run_test(|| { |
||||||
|
// ACT
|
||||||
|
let json = stamps::read_stamps_file_to_json().unwrap(); |
||||||
|
stamps::get_stamp_value("Foo", &json).unwrap(); |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
#[test] |
||||||
|
#[serial] |
||||||
|
fn load_stamp() { |
||||||
|
run_test(|| { |
||||||
|
// ARRANGE
|
||||||
|
stamps::save_stamp_value("Foo", "Bar").unwrap(); |
||||||
|
|
||||||
|
// ACT
|
||||||
|
let json = stamps::read_stamps_file_to_json().unwrap(); |
||||||
|
let stamp_value = stamps::get_stamp_value("Foo", &json).unwrap(); |
||||||
|
|
||||||
|
// ASSERT
|
||||||
|
assert_eq!(stamp_value, "Bar"); |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
#[test] |
||||||
|
#[serial] |
||||||
|
fn update_stamp() { |
||||||
|
run_test(|| { |
||||||
|
// ARRANGE
|
||||||
|
stamps::save_stamp_value("Foo", "Bar").unwrap(); |
||||||
|
|
||||||
|
// ACT
|
||||||
|
stamps::save_stamp_value("Foo", "John").unwrap(); |
||||||
|
|
||||||
|
// ASSERT
|
||||||
|
let json = stamps::read_stamps_file_to_json().unwrap(); |
||||||
|
let stamp_value = stamps::get_stamp_value("Foo", &json).unwrap(); |
||||||
|
assert_eq!(stamp_value, "John"); |
||||||
|
}) |
||||||
|
} |
||||||
Loading…
Reference in new issue