datasynth_generators/sourcing/
catalog_generator.rs

1//! Catalog item generator.
2
3use datasynth_config::schema::CatalogConfig;
4use datasynth_core::models::sourcing::{CatalogItem, ProcurementContract};
5use datasynth_core::utils::seeded_rng;
6use datasynth_core::uuid_factory::{DeterministicUuidFactory, GeneratorType};
7use rand::prelude::*;
8use rand_chacha::ChaCha8Rng;
9
10/// Generates catalog items from active contracts.
11pub struct CatalogGenerator {
12    rng: ChaCha8Rng,
13    uuid_factory: DeterministicUuidFactory,
14    config: CatalogConfig,
15}
16
17impl CatalogGenerator {
18    /// Create a new catalog generator.
19    pub fn new(seed: u64) -> Self {
20        Self {
21            rng: seeded_rng(seed, 0),
22            uuid_factory: DeterministicUuidFactory::new(seed, GeneratorType::CatalogItem),
23            config: CatalogConfig::default(),
24        }
25    }
26
27    /// Create with custom configuration.
28    pub fn with_config(seed: u64, config: CatalogConfig) -> Self {
29        Self {
30            rng: seeded_rng(seed, 0),
31            uuid_factory: DeterministicUuidFactory::new(seed, GeneratorType::CatalogItem),
32            config,
33        }
34    }
35
36    /// Generate catalog items from a list of active contracts.
37    pub fn generate(&mut self, contracts: &[ProcurementContract]) -> Vec<CatalogItem> {
38        let mut items = Vec::new();
39
40        for contract in contracts {
41            for line in &contract.line_items {
42                let is_preferred = self.rng.random_bool(self.config.preferred_vendor_flag_rate);
43
44                items.push(CatalogItem {
45                    catalog_item_id: self.uuid_factory.next().to_string(),
46                    contract_id: contract.contract_id.clone(),
47                    contract_line_number: line.line_number,
48                    vendor_id: contract.vendor_id.clone(),
49                    material_id: line.material_id.clone(),
50                    description: line.description.clone(),
51                    catalog_price: line.unit_price,
52                    uom: line.uom.clone(),
53                    is_preferred,
54                    category: contract.category_id.clone(),
55                    min_order_quantity: line.min_quantity,
56                    lead_time_days: Some(self.rng.random_range(3..=30)),
57                    is_active: true,
58                });
59
60                // Possibly add alternative source
61                if self.rng.random_bool(self.config.multi_source_rate) {
62                    items.push(CatalogItem {
63                        catalog_item_id: self.uuid_factory.next().to_string(),
64                        contract_id: contract.contract_id.clone(),
65                        contract_line_number: line.line_number,
66                        vendor_id: format!("{}-ALT", contract.vendor_id),
67                        material_id: line.material_id.clone(),
68                        description: format!("{} (alternate)", line.description),
69                        catalog_price: line.unit_price
70                            * rust_decimal::Decimal::from_f64_retain(
71                                self.rng.random_range(0.95..=1.10),
72                            )
73                            .unwrap_or(rust_decimal::Decimal::ONE),
74                        uom: line.uom.clone(),
75                        is_preferred: false,
76                        category: contract.category_id.clone(),
77                        min_order_quantity: line.min_quantity,
78                        lead_time_days: Some(self.rng.random_range(5..=45)),
79                        is_active: true,
80                    });
81                }
82            }
83        }
84
85        items
86    }
87}
88
89#[cfg(test)]
90mod tests {
91    use super::*;
92    use datasynth_core::models::sourcing::{
93        ContractLineItem, ContractSla, ContractStatus, ContractTerms, ContractType,
94    };
95    use rust_decimal::Decimal;
96
97    fn test_contract() -> ProcurementContract {
98        ProcurementContract {
99            contract_id: "CTR-001".to_string(),
100            company_code: "C001".to_string(),
101            contract_type: ContractType::FixedPrice,
102            status: ContractStatus::Active,
103            vendor_id: "V001".to_string(),
104            title: "Test Contract".to_string(),
105            sourcing_project_id: None,
106            bid_id: None,
107            start_date: chrono::NaiveDate::from_ymd_opt(2024, 1, 1).unwrap(),
108            end_date: chrono::NaiveDate::from_ymd_opt(2025, 12, 31).unwrap(),
109            total_value: Decimal::from(100_000),
110            consumed_value: Decimal::ZERO,
111            terms: ContractTerms {
112                payment_terms: "NET30".to_string(),
113                delivery_terms: Some("FCA".to_string()),
114                warranty_months: None,
115                early_termination_penalty_pct: None,
116                auto_renewal: false,
117                termination_notice_days: 60,
118                price_adjustment_clause: false,
119                max_annual_price_increase_pct: None,
120            },
121            slas: vec![ContractSla {
122                metric_name: "on_time_delivery".to_string(),
123                target_value: 0.95,
124                minimum_value: 0.90,
125                breach_penalty_pct: 0.02,
126                measurement_frequency: "monthly".to_string(),
127            }],
128            line_items: vec![
129                ContractLineItem {
130                    line_number: 1,
131                    material_id: None,
132                    description: "Widget A".to_string(),
133                    unit_price: Decimal::from(50),
134                    uom: "EA".to_string(),
135                    min_quantity: Some(Decimal::from(100)),
136                    max_quantity: Some(Decimal::from(1200)),
137                    quantity_released: Decimal::ZERO,
138                    value_released: Decimal::ZERO,
139                },
140                ContractLineItem {
141                    line_number: 2,
142                    material_id: Some("MAT-002".to_string()),
143                    description: "Widget B".to_string(),
144                    unit_price: Decimal::from(25),
145                    uom: "EA".to_string(),
146                    min_quantity: Some(Decimal::from(200)),
147                    max_quantity: Some(Decimal::from(2400)),
148                    quantity_released: Decimal::ZERO,
149                    value_released: Decimal::ZERO,
150                },
151            ],
152            category_id: "CAT-001".to_string(),
153            owner_id: "BUYER-01".to_string(),
154            amendment_count: 0,
155            previous_contract_id: None,
156            purchase_order_ids: Vec::new(),
157        }
158    }
159
160    #[test]
161    fn test_basic_generation() {
162        let mut gen = CatalogGenerator::new(42);
163        let contracts = vec![test_contract()];
164        let items = gen.generate(&contracts);
165
166        // Should have at least 2 items (one per contract line)
167        assert!(items.len() >= 2);
168        for item in &items {
169            assert!(!item.catalog_item_id.is_empty());
170            assert_eq!(item.contract_id, "CTR-001");
171            assert!(!item.vendor_id.is_empty());
172            assert!(item.catalog_price > Decimal::ZERO);
173            assert!(item.is_active);
174            assert_eq!(item.category, "CAT-001");
175        }
176    }
177
178    #[test]
179    fn test_deterministic() {
180        let contracts = vec![test_contract()];
181
182        let mut gen1 = CatalogGenerator::new(42);
183        let mut gen2 = CatalogGenerator::new(42);
184
185        let r1 = gen1.generate(&contracts);
186        let r2 = gen2.generate(&contracts);
187
188        assert_eq!(r1.len(), r2.len());
189        for (a, b) in r1.iter().zip(r2.iter()) {
190            assert_eq!(a.catalog_item_id, b.catalog_item_id);
191            assert_eq!(a.vendor_id, b.vendor_id);
192            assert_eq!(a.catalog_price, b.catalog_price);
193            assert_eq!(a.is_preferred, b.is_preferred);
194        }
195    }
196
197    #[test]
198    fn test_field_constraints() {
199        let mut gen = CatalogGenerator::new(99);
200        let contracts = vec![test_contract()];
201        let items = gen.generate(&contracts);
202
203        for item in &items {
204            // Lead time should be within expected range
205            assert!(item.lead_time_days.is_some());
206            let lt = item.lead_time_days.unwrap();
207            assert!((3..=45).contains(&lt));
208
209            // UOM should be set
210            assert!(!item.uom.is_empty());
211
212            // Contract line number should be valid
213            assert!(item.contract_line_number >= 1);
214        }
215
216        // Check alternate sources have "-ALT" suffix on vendor_id
217        let alt_items: Vec<_> = items
218            .iter()
219            .filter(|i| i.vendor_id.contains("-ALT"))
220            .collect();
221        for alt in &alt_items {
222            assert!(!alt.is_preferred); // Alternates should not be preferred
223            assert!(alt.description.contains("(alternate)"));
224        }
225    }
226}
datasynth_generators/sourcing/catalog_generator.rs

datasynth_generators/sourcing/
catalog_generator.rs
