use std::time::Duration;
use crate::{
error::RunnerErrorKind,
webdriver::{Locator, Webdriver},
};
use super::Command;
pub struct WaitForElementPresent {
target: Locator,
timeout: Duration,
}
impl WaitForElementPresent {
pub fn new(target: Locator, timeout: Duration) -> Self {
Self { target, timeout }
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for WaitForElementPresent {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
runner
.get_webdriver()
.wait_for_present(self.target.clone(), self.timeout)
.await
.map_err(|_| RunnerErrorKind::Timeout("WaitForElementPresent".to_owned()))?;
Ok(())
}
}
pub struct WaitForElementNotPresent {
target: Locator,
timeout: Duration,
}
impl WaitForElementNotPresent {
pub fn new(target: Locator, timeout: Duration) -> Self {
Self { target, timeout }
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for WaitForElementNotPresent {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
runner
.get_webdriver()
.wait_for_not_present(self.target.clone(), self.timeout)
.await
.map_err(|_| RunnerErrorKind::Timeout("WaitForElementPresent".to_owned()))?;
Ok(())
}
}