use super::Command;
use crate::{
error::RunnerErrorKind,
webdriver::{Element, Locator, Webdriver},
};
pub struct Check {
target: Locator,
}
impl Check {
pub fn new(target: Locator) -> Self {
Self { target }
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for Check {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
let mut e = runner.get_webdriver().find(self.target.clone()).await?;
let selected = e.prop("selected").await?;
if selected.is_none() {
e.click().await?;
}
Ok(())
}
}
pub struct UnCheck {
target: Locator,
}
impl UnCheck {
pub fn new(target: Locator) -> Self {
Self { target }
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for UnCheck {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
let mut e = runner.get_webdriver().find(self.target.clone()).await?;
let selected = e.prop("selected").await?;
if selected.is_some() {
e.click().await?;
}
Ok(())
}
}