use super::super::{comment::shouldbespace, value::value, IResult};
use crate::Start;
use nom::{
bytes::complete::tag_no_case,
combinator::{cut, opt},
sequence::terminated,
};
pub fn start(i: &str) -> IResult<&str, Start> {
let (i, _) = tag_no_case("START")(i)?;
let (i, _) = shouldbespace(i)?;
cut(|i| {
let (i, _) = opt(terminated(tag_no_case("AT"), shouldbespace))(i)?;
let (i, v) = value(i)?;
Ok((i, Start(v)))
})(i)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::Value;
#[test]
fn start_statement() {
let sql = "START 100";
let res = start(sql);
let out = res.unwrap().1;
assert_eq!(out, Start(Value::from(100)));
assert_eq!("START 100", format!("{}", out));
}
#[test]
fn start_statement_at() {
let sql = "START AT 100";
let res = start(sql);
let out = res.unwrap().1;
assert_eq!(out, Start(Value::from(100)));
assert_eq!("START 100", format!("{}", out));
}
}