use super::Command;
use crate::{
error::RunnerErrorKind,
parser::SelectLocator,
webdriver::{Element, Locator, Webdriver},
};
pub struct Select {
target: Locator,
select_target: SelectLocator,
}
impl Select {
pub fn new(target: Locator, select_target: SelectLocator) -> Self {
Self {
target,
select_target,
}
}
}
#[async_trait::async_trait]
impl<D: Webdriver> Command<D> for Select {
async fn run(&self, runner: &mut crate::runner::Runner<D>) -> Result<(), RunnerErrorKind> {
let mut select = runner.get_webdriver().find(self.target.clone()).await?;
match &self.select_target {
SelectLocator::Index(index) => {
let index = runner.emit(index);
match index.parse() {
Ok(index) => {
select.select_by_index(index).await?;
}
Err(..) => {
return Err(RunnerErrorKind::MismatchedType(format!(
"expected to get int type but got {:?}",
index
)));
}
}
}
SelectLocator::Value(value) => {
let value = runner.emit(value);
select.select_by_value(&value).await?;
}
SelectLocator::Id(id) => {
let id = runner.emit(id);
let locator = format!(r#"option[id='{}']"#, id);
let mut opt = select.find(Locator::Css(locator)).await?;
let selected = opt.prop("selected").await?;
if selected.is_none() {
opt.click().await?;
}
}
SelectLocator::Label(label) => {
let label = runner.emit(label);
let locator = format!(".//*[normalize-space(.)='{}']", label);
let mut opt = select.find(Locator::XPath(locator)).await?;
if !opt.is_selected().await? {
opt.click().await?;
}
}
};
Ok(())
}
}