use crate::Currency;
use std::ops::{Add, Div, Mul, Sub};
impl Add for Currency {
type Output = Currency;
#[inline]
fn add(self, rhs: Currency) -> Currency {
if self.symbol == rhs.symbol || self.symbol.is_none() {
Currency {
symbol: rhs.symbol,
value: self.value + rhs.value,
}
} else {
panic!(
"Cannot add two different types of currency!\n{:?} vs {:?}",
self.symbol, rhs.symbol
);
}
}
}
impl Sub for Currency {
type Output = Currency;
#[inline]
fn sub(self, rhs: Currency) -> Currency {
if self.symbol == rhs.symbol || self.symbol.is_none() {
Currency {
symbol: rhs.symbol,
value: self.value - rhs.value,
}
} else {
panic!("Cannot subtract two different types of currency!");
}
}
}
impl Mul<i64> for Currency {
type Output = Currency;
#[inline]
fn mul(self, rhs: i64) -> Currency {
Currency {
symbol: self.symbol,
value: self.value * rhs,
}
}
}
impl Mul<Currency> for i64 {
type Output = Currency;
#[inline]
fn mul(self, rhs: Currency) -> Currency {
Currency {
symbol: rhs.symbol,
value: rhs.value * self,
}
}
}
impl Mul<f64> for Currency {
type Output = Currency;
#[inline]
fn mul(self, rhs: f64) -> Currency {
Currency {
symbol: self.symbol,
value: (self.value as f64 * rhs).round() as i64,
}
}
}
impl Mul<Currency> for f64 {
type Output = Currency;
#[inline]
fn mul(self, rhs: Currency) -> Currency {
rhs * self
}
}
impl Div<i64> for Currency {
type Output = Currency;
#[inline]
fn div(self, rhs: i64) -> Currency {
Currency {
symbol: self.symbol,
value: self.value / rhs,
}
}
}
#[cfg(test)]
mod test_arithmetic {
use crate::Currency;
fn dollars(value: i64) -> Currency {
Currency {
symbol: Some('$'),
value,
}
}
fn bare(value: i64) -> Currency {
Currency::from_value(value)
}
#[test]
fn sub_same_symbol() {
assert_eq!(dollars(100) - dollars(50), dollars(50));
}
#[test]
fn add_same_symbol() {
assert_eq!(dollars(100) + dollars(50), dollars(150));
}
#[test]
fn bare_lhs_adopts_rhs_symbol_on_add() {
assert_eq!(bare(100) + dollars(50), dollars(150));
}
#[test]
fn bare_lhs_adopts_rhs_symbol_on_sub() {
assert_eq!(bare(100) - dollars(50), dollars(50));
}
#[test]
#[should_panic]
fn symbolled_lhs_plus_bare_rhs_panics() {
let _ = dollars(100) + bare(50);
}
#[test]
#[should_panic]
fn symbolled_lhs_minus_bare_rhs_panics() {
let _ = dollars(100) - bare(50);
}
#[test]
#[should_panic]
fn mixed_symbols_add_panics() {
let euro = Currency {
symbol: Some('€'),
value: 100,
};
let _ = dollars(100) + euro;
}
#[test]
#[should_panic]
fn mixed_symbols_sub_panics() {
let euro = Currency {
symbol: Some('€'),
value: 100,
};
let _ = dollars(100) - euro;
}
}