datasynth_generators/sourcing/
rfx_generator.rs

1//! RFx event generator.
2
3use chrono::NaiveDate;
4use datasynth_config::schema::RfxConfig;
5use datasynth_core::models::sourcing::{
6    RfxEvaluationCriterion, RfxEvent, RfxLineItem, RfxStatus, RfxType, ScoringMethod,
7};
8use datasynth_core::utils::seeded_rng;
9use datasynth_core::uuid_factory::{DeterministicUuidFactory, GeneratorType};
10use rand::prelude::*;
11use rand_chacha::ChaCha8Rng;
12use rust_decimal::Decimal;
13
14/// Generates RFx events (RFI/RFP/RFQ).
15pub struct RfxGenerator {
16    rng: ChaCha8Rng,
17    uuid_factory: DeterministicUuidFactory,
18    config: RfxConfig,
19}
20
21impl RfxGenerator {
22    /// Create a new RFx generator.
23    pub fn new(seed: u64) -> Self {
24        Self {
25            rng: seeded_rng(seed, 0),
26            uuid_factory: DeterministicUuidFactory::new(seed, GeneratorType::RfxEvent),
27            config: RfxConfig::default(),
28        }
29    }
30
31    /// Create with custom configuration.
32    pub fn with_config(seed: u64, config: RfxConfig) -> Self {
33        Self {
34            rng: seeded_rng(seed, 0),
35            uuid_factory: DeterministicUuidFactory::new(seed, GeneratorType::RfxEvent),
36            config,
37        }
38    }
39
40    /// Generate an RFx event for a sourcing project.
41    pub fn generate(
42        &mut self,
43        company_code: &str,
44        sourcing_project_id: &str,
45        category_id: &str,
46        qualified_vendor_ids: &[String],
47        owner_id: &str,
48        publish_date: NaiveDate,
49        estimated_spend: f64,
50    ) -> RfxEvent {
51        let rfx_type = if estimated_spend > self.config.rfi_threshold {
52            if self.rng.random_bool(0.3) {
53                RfxType::Rfi
54            } else {
55                RfxType::Rfp
56            }
57        } else {
58            RfxType::Rfq
59        };
60
61        let invited_count = self
62            .rng
63            .random_range(self.config.min_invited_vendors..=self.config.max_invited_vendors)
64            .min(qualified_vendor_ids.len() as u32) as usize;
65
66        let invited_vendors: Vec<String> = qualified_vendor_ids
67            .sample(&mut self.rng, invited_count)
68            .cloned()
69            .collect();
70
71        let response_deadline =
72            publish_date + chrono::Duration::days(self.rng.random_range(14..=45));
73
74        let bid_count = (invited_vendors.len() as f64 * self.config.response_rate).round() as u32;
75
76        let criteria = vec![
77            RfxEvaluationCriterion {
78                name: "Price".to_string(),
79                weight: self.config.default_price_weight,
80                description: "Total cost of ownership".to_string(),
81            },
82            RfxEvaluationCriterion {
83                name: "Quality".to_string(),
84                weight: self.config.default_quality_weight,
85                description: "Quality management and track record".to_string(),
86            },
87            RfxEvaluationCriterion {
88                name: "Delivery".to_string(),
89                weight: self.config.default_delivery_weight,
90                description: "Lead time and reliability".to_string(),
91            },
92        ];
93
94        let line_count = self.rng.random_range(1u16..=5);
95        let line_items: Vec<RfxLineItem> = (1..=line_count)
96            .map(|i| RfxLineItem {
97                item_number: i,
98                description: format!("Item {i}"),
99                material_id: None,
100                quantity: Decimal::from(self.rng.random_range(10..=1000)),
101                uom: "EA".to_string(),
102                target_price: Some(Decimal::from(self.rng.random_range(10..=5000))),
103            })
104            .collect();
105
106        let scoring_method = match rfx_type {
107            RfxType::Rfq => ScoringMethod::LowestPrice,
108            RfxType::Rfp => ScoringMethod::BestValue,
109            RfxType::Rfi => ScoringMethod::QualityBased,
110        };
111
112        RfxEvent {
113            rfx_id: self.uuid_factory.next().to_string(),
114            rfx_type,
115            company_code: company_code.to_string(),
116            title: format!("RFx for {category_id}"),
117            description: format!(
118                "Sourcing event for category {category_id} under project {sourcing_project_id}"
119            ),
120            status: RfxStatus::Awarded,
121            sourcing_project_id: sourcing_project_id.to_string(),
122            category_id: category_id.to_string(),
123            scoring_method,
124            criteria,
125            line_items,
126            invited_vendors,
127            publish_date,
128            response_deadline,
129            bid_count,
130            owner_id: owner_id.to_string(),
131            awarded_vendor_id: None,
132            awarded_bid_id: None,
133        }
134    }
135}
136
137#[cfg(test)]
138mod tests {
139    use super::*;
140
141    fn test_vendor_ids() -> Vec<String> {
142        (1..=6).map(|i| format!("V{:04}", i)).collect()
143    }
144
145    #[test]
146    fn test_basic_generation() {
147        let mut gen = RfxGenerator::new(42);
148        let date = NaiveDate::from_ymd_opt(2024, 3, 1).unwrap();
149        let rfx = gen.generate(
150            "C001",
151            "SP-001",
152            "CAT-001",
153            &test_vendor_ids(),
154            "BUYER-01",
155            date,
156            200_000.0,
157        );
158
159        assert!(!rfx.rfx_id.is_empty());
160        assert_eq!(rfx.company_code, "C001");
161        assert_eq!(rfx.sourcing_project_id, "SP-001");
162        assert_eq!(rfx.category_id, "CAT-001");
163        assert_eq!(rfx.owner_id, "BUYER-01");
164        assert_eq!(rfx.status, RfxStatus::Awarded);
165        assert!(!rfx.invited_vendors.is_empty());
166        assert!(!rfx.criteria.is_empty());
167        assert!(!rfx.line_items.is_empty());
168        assert!(rfx.response_deadline > date);
169    }
170
171    #[test]
172    fn test_deterministic() {
173        let date = NaiveDate::from_ymd_opt(2024, 3, 1).unwrap();
174        let vendors = test_vendor_ids();
175
176        let mut gen1 = RfxGenerator::new(42);
177        let mut gen2 = RfxGenerator::new(42);
178
179        let r1 = gen1.generate(
180            "C001", "SP-001", "CAT-001", &vendors, "BUYER-01", date, 200_000.0,
181        );
182        let r2 = gen2.generate(
183            "C001", "SP-001", "CAT-001", &vendors, "BUYER-01", date, 200_000.0,
184        );
185
186        assert_eq!(r1.rfx_id, r2.rfx_id);
187        assert_eq!(r1.rfx_type, r2.rfx_type);
188        assert_eq!(r1.invited_vendors, r2.invited_vendors);
189        assert_eq!(r1.line_items.len(), r2.line_items.len());
190        assert_eq!(r1.bid_count, r2.bid_count);
191    }
192
193    #[test]
194    fn test_field_constraints() {
195        let mut gen = RfxGenerator::new(99);
196        let date = NaiveDate::from_ymd_opt(2024, 3, 1).unwrap();
197
198        // High spend should produce RFI or RFP
199        let rfx_high = gen.generate(
200            "C001",
201            "SP-001",
202            "CAT-001",
203            &test_vendor_ids(),
204            "BUYER-01",
205            date,
206            500_000.0,
207        );
208        assert!(matches!(rfx_high.rfx_type, RfxType::Rfi | RfxType::Rfp));
209
210        // Low spend should produce RFQ
211        let rfx_low = gen.generate(
212            "C001",
213            "SP-002",
214            "CAT-002",
215            &test_vendor_ids(),
216            "BUYER-01",
217            date,
218            50_000.0,
219        );
220        assert_eq!(rfx_low.rfx_type, RfxType::Rfq);
221
222        // Criteria weights should exist
223        assert_eq!(rfx_high.criteria.len(), 3);
224        let weight_sum: f64 = rfx_high.criteria.iter().map(|c| c.weight).sum();
225        assert!((weight_sum - 1.0).abs() < 0.01);
226
227        // Line items should have valid item numbers
228        for (i, item) in rfx_high.line_items.iter().enumerate() {
229            assert_eq!(item.item_number, (i + 1) as u16);
230            assert!(item.quantity > Decimal::ZERO);
231        }
232    }
233}
datasynth_generators/sourcing/rfx_generator.rs

datasynth_generators/sourcing/
rfx_generator.rs
