use super::Command;
use crate::{
error::RunnerErrorKind,
webdriver::{Element, Locator, Webdriver},
};
use serde_json::Value;
pub struct StoreValue {
target: Locator,
variable: String,
}
impl StoreValue {
pub fn new(target: Locator, variable: String) -> Self {
Self { target, variable }
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for StoreValue {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
let value = runner
.get_webdriver()
.find(self.target.clone())
.await?
.prop("value")
.await?
.unwrap_or_else(String::new);
runner.save_value(self.variable.clone(), Value::String(value));
Ok(())
}
}