imag/lib/entry/libimagentrydatetime/src/range.rs

92 lines
2.6 KiB
Rust
Raw Normal View History

2017-06-06 17:28:00 +00:00
//
// imag - the personal information management suite for the commandline
// Copyright (C) 2015, 2016 Matthias Beyer <mail@beyermatthias.de> and contributors
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; version
// 2.1 of the License.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
//
use chrono::naive::NaiveDateTime;
2017-09-03 11:53:55 +00:00
use error::DateErrorKind as DEK;
use error::DateError as DE;
use error::Result;
2017-09-03 11:53:55 +00:00
2017-06-06 17:28:00 +00:00
/// A Range between two dates
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct DateTimeRange(NaiveDateTime, NaiveDateTime);
impl DateTimeRange {
/// Create a new DateTimeRange object
///
/// # Return value
///
/// Ok(DateTimeRange) if start is before end,
/// else Err(DateTimeRangeError)
///
pub fn new(start: NaiveDateTime, end: NaiveDateTime) -> Result<DateTimeRange> {
if start < end {
Ok(DateTimeRange(start, end))
} else {
Err(DE::from_kind(DEK::EndDateTimeBeforeStartDateTime))
2017-06-06 17:28:00 +00:00
}
}
}
2017-06-06 17:36:10 +00:00
#[cfg(test)]
mod tests {
2017-07-15 18:14:18 +00:00
use chrono::naive::NaiveDateTime;
use chrono::naive::NaiveDate;
use chrono::naive::NaiveTime;
2017-06-06 17:36:10 +00:00
use super::DateTimeRange;
#[test]
fn test_new_returns_error_if_start_after_end_date() {
let start = NaiveDateTime::new(
NaiveDate::from_ymd(2000, 02, 02),
NaiveTime::from_hms(12, 00, 02)
);
let end = NaiveDateTime::new(
NaiveDate::from_ymd(2000, 02, 02),
NaiveTime::from_hms(12, 00, 01)
);
let res = DateTimeRange::new(start, end);
assert!(res.is_err());
}
#[test]
fn test_new_returns_ok_if_start_is_before_end() {
let start = NaiveDateTime::new(
NaiveDate::from_ymd(2000, 02, 02),
NaiveTime::from_hms(12, 00, 01)
);
let end = NaiveDateTime::new(
NaiveDate::from_ymd(2000, 02, 02),
NaiveTime::from_hms(12, 00, 02)
);
let res = DateTimeRange::new(start, end);
assert!(res.is_ok());
}
}