use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use crate::com::TimePeriod;
use crate::enums::Division;
use crate::traits::{Bo4eMeta, Bo4eObject};
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Balancing {
#[serde(flatten)]
pub meta: Bo4eMeta,
#[serde(skip_serializing_if = "Option::is_none")]
pub balance_group_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub balance_group_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 market_area: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub balance_responsible_party: Option<Box<super::MarketParticipant>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub validity_period: Option<TimePeriod>,
#[serde(skip_serializing_if = "Option::is_none")]
pub start_date: Option<DateTime<Utc>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub end_date: Option<DateTime<Utc>>,
}
impl Bo4eObject for Balancing {
fn type_name_german() -> &'static str {
"Bilanzierung"
}
fn type_name_english() -> &'static str {
"Balancing"
}
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_balancing_creation() {
let balancing = Balancing {
balance_group_id: Some("11XBALANCEGROUP".to_string()),
division: Some(Division::Electricity),
..Default::default()
};
assert_eq!(
balancing.balance_group_id,
Some("11XBALANCEGROUP".to_string())
);
assert_eq!(balancing.division, Some(Division::Electricity));
}
#[test]
fn test_gas_balancing() {
let balancing = Balancing {
balance_group_id: Some("GASBALANCE001".to_string()),
balance_group_name: Some("Gas Balance Group 1".to_string()),
division: Some(Division::Gas),
market_area: Some("NCG".to_string()),
..Default::default()
};
assert_eq!(balancing.division, Some(Division::Gas));
assert_eq!(balancing.market_area, Some("NCG".to_string()));
}
#[test]
fn test_serialize() {
let balancing = Balancing {
meta: Bo4eMeta::with_type("Bilanzierung"),
balance_group_id: Some("11XTEST".to_string()),
division: Some(Division::Electricity),
..Default::default()
};
let json = serde_json::to_string(&balancing).unwrap();
assert!(json.contains(r#""balanceGroupId":"11XTEST""#));
}
#[test]
fn test_roundtrip() {
let balancing = Balancing {
meta: Bo4eMeta::with_type("Bilanzierung"),
balance_group_id: Some("11XTEST".to_string()),
balance_group_name: Some("Test Balance Group".to_string()),
description: Some("Test description".to_string()),
division: Some(Division::Electricity),
market_area: Some("DE".to_string()),
..Default::default()
};
let json = serde_json::to_string(&balancing).unwrap();
let parsed: Balancing = serde_json::from_str(&json).unwrap();
assert_eq!(balancing, parsed);
}
#[test]
fn test_bo4e_object_impl() {
assert_eq!(Balancing::type_name_german(), "Bilanzierung");
assert_eq!(Balancing::type_name_english(), "Balancing");
}
}