use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use crate::com::{EnergyMix, PriceGuarantee, TariffRestriction, TimePeriod};
use crate::enums::{CustomerType, Division};
use crate::traits::{Bo4eMeta, Bo4eObject};
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct TariffInfo {
#[serde(flatten)]
pub meta: Bo4eMeta,
#[serde(skip_serializing_if = "Option::is_none")]
pub tariff_name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub description: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub division: Option<Division>,
#[serde(skip_serializing_if = "Option::is_none")]
pub customer_type: Option<CustomerType>,
#[serde(skip_serializing_if = "Option::is_none")]
pub website: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub validity_period: Option<TimePeriod>,
#[serde(skip_serializing_if = "Option::is_none")]
pub available_from: Option<DateTime<Utc>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub available_until: Option<DateTime<Utc>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub energy_mix: Option<EnergyMix>,
#[serde(skip_serializing_if = "Option::is_none")]
pub price_guarantee: Option<PriceGuarantee>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub restrictions: Vec<TariffRestriction>,
#[serde(skip_serializing_if = "Option::is_none")]
pub supplier: Option<Box<super::BusinessPartner>>,
}
impl Bo4eObject for TariffInfo {
fn type_name_german() -> &'static str {
"Tarifinfo"
}
fn type_name_english() -> &'static str {
"TariffInfo"
}
fn meta(&self) -> &Bo4eMeta {
&self.meta
}
fn meta_mut(&mut self) -> &mut Bo4eMeta {
&mut self.meta
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_tariff_info_creation() {
let tariff_info = TariffInfo {
tariff_name: Some("Oekostrom Basis".to_string()),
division: Some(Division::Electricity),
customer_type: Some(CustomerType::Household),
website: Some("https://example.com/tariff".to_string()),
..Default::default()
};
assert_eq!(tariff_info.tariff_name, Some("Oekostrom Basis".to_string()));
assert_eq!(tariff_info.customer_type, Some(CustomerType::Household));
}
#[test]
fn test_serialize() {
let tariff_info = TariffInfo {
meta: Bo4eMeta::with_type("Tarifinfo"),
tariff_name: Some("Green Energy".to_string()),
division: Some(Division::Electricity),
..Default::default()
};
let json = serde_json::to_string(&tariff_info).unwrap();
assert!(json.contains(r#""tariffName":"Green Energy""#));
assert!(json.contains(r#""_typ":"Tarifinfo""#));
}
#[test]
fn test_roundtrip() {
let tariff_info = TariffInfo {
meta: Bo4eMeta::with_type("Tarifinfo"),
tariff_name: Some("Test Tariff".to_string()),
description: Some("A test tariff info".to_string()),
division: Some(Division::Gas),
customer_type: Some(CustomerType::Commercial),
website: Some("https://test.com".to_string()),
..Default::default()
};
let json = serde_json::to_string(&tariff_info).unwrap();
let parsed: TariffInfo = serde_json::from_str(&json).unwrap();
assert_eq!(tariff_info, parsed);
}
#[test]
fn test_bo4e_object_impl() {
assert_eq!(TariffInfo::type_name_german(), "Tarifinfo");
assert_eq!(TariffInfo::type_name_english(), "TariffInfo");
}
}