use crate::{
cilassembly::{ChangeRefRc, CilAssembly},
metadata::{
tables::{AssemblyOsRaw, TableDataOwned, TableId},
token::Token,
},
Error, Result,
};
#[derive(Debug, Clone)]
#[allow(clippy::struct_field_names)]
pub struct AssemblyOSBuilder {
os_platform_id: Option<u32>,
os_major_version: Option<u32>,
os_minor_version: Option<u32>,
}
impl AssemblyOSBuilder {
#[must_use]
pub fn new() -> Self {
Self {
os_platform_id: None,
os_major_version: None,
os_minor_version: None,
}
}
#[must_use]
pub fn os_platform_id(mut self, os_platform_id: u32) -> Self {
self.os_platform_id = Some(os_platform_id);
self
}
#[must_use]
pub fn os_major_version(mut self, os_major_version: u32) -> Self {
self.os_major_version = Some(os_major_version);
self
}
#[must_use]
pub fn os_minor_version(mut self, os_minor_version: u32) -> Self {
self.os_minor_version = Some(os_minor_version);
self
}
pub fn build(self, assembly: &mut CilAssembly) -> Result<ChangeRefRc> {
let os_platform_id = self.os_platform_id.ok_or_else(|| {
Error::ModificationInvalid(
"OS platform identifier is required for AssemblyOS".to_string(),
)
})?;
let os_major_version = self.os_major_version.ok_or_else(|| {
Error::ModificationInvalid("OS major version is required for AssemblyOS".to_string())
})?;
let os_minor_version = self.os_minor_version.ok_or_else(|| {
Error::ModificationInvalid("OS minor version is required for AssemblyOS".to_string())
})?;
let assembly_os = AssemblyOsRaw {
rid: 0,
token: Token::new(0),
offset: 0,
os_platform_id,
os_major_version,
os_minor_version,
};
assembly.table_row_add(TableId::AssemblyOS, TableDataOwned::AssemblyOS(assembly_os))
}
}
impl Default for AssemblyOSBuilder {
fn default() -> Self {
Self::new()
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::{
cilassembly::ChangeRefKind, test::factories::table::assemblyref::get_test_assembly,
};
#[test]
fn test_assemblyos_builder_new() {
let builder = AssemblyOSBuilder::new();
assert!(builder.os_platform_id.is_none());
assert!(builder.os_major_version.is_none());
assert!(builder.os_minor_version.is_none());
}
#[test]
fn test_assemblyos_builder_default() {
let builder = AssemblyOSBuilder::default();
assert!(builder.os_platform_id.is_none());
assert!(builder.os_major_version.is_none());
assert!(builder.os_minor_version.is_none());
}
#[test]
fn test_assemblyos_builder_windows10() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(1) .os_major_version(10) .os_minor_version(0) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_windows7() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(1) .os_major_version(6) .os_minor_version(1) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_linux() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(2) .os_major_version(5) .os_minor_version(4) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_custom() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(99) .os_major_version(1) .os_minor_version(0) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_missing_platform_id() -> Result<()> {
let mut assembly = get_test_assembly()?;
let result = AssemblyOSBuilder::new()
.os_major_version(10)
.os_minor_version(0)
.build(&mut assembly);
assert!(result.is_err());
match result.unwrap_err() {
Error::ModificationInvalid(details) => {
assert!(details.contains("OS platform identifier is required"));
}
_ => panic!("Expected ModificationInvalid error"),
}
Ok(())
}
#[test]
fn test_assemblyos_builder_missing_major_version() -> Result<()> {
let mut assembly = get_test_assembly()?;
let result = AssemblyOSBuilder::new()
.os_platform_id(1)
.os_minor_version(0)
.build(&mut assembly);
assert!(result.is_err());
match result.unwrap_err() {
Error::ModificationInvalid(details) => {
assert!(details.contains("OS major version is required"));
}
_ => panic!("Expected ModificationInvalid error"),
}
Ok(())
}
#[test]
fn test_assemblyos_builder_missing_minor_version() -> Result<()> {
let mut assembly = get_test_assembly()?;
let result = AssemblyOSBuilder::new()
.os_platform_id(1)
.os_major_version(10)
.build(&mut assembly);
assert!(result.is_err());
match result.unwrap_err() {
Error::ModificationInvalid(details) => {
assert!(details.contains("OS minor version is required"));
}
_ => panic!("Expected ModificationInvalid error"),
}
Ok(())
}
#[test]
fn test_assemblyos_builder_clone() {
let builder = AssemblyOSBuilder::new()
.os_platform_id(1)
.os_major_version(10)
.os_minor_version(0);
let cloned = builder.clone();
assert_eq!(builder.os_platform_id, cloned.os_platform_id);
assert_eq!(builder.os_major_version, cloned.os_major_version);
assert_eq!(builder.os_minor_version, cloned.os_minor_version);
}
#[test]
fn test_assemblyos_builder_debug() {
let builder = AssemblyOSBuilder::new()
.os_platform_id(2)
.os_major_version(5)
.os_minor_version(4);
let debug_str = format!("{builder:?}");
assert!(debug_str.contains("AssemblyOSBuilder"));
assert!(debug_str.contains("os_platform_id"));
assert!(debug_str.contains("os_major_version"));
assert!(debug_str.contains("os_minor_version"));
}
#[test]
fn test_assemblyos_builder_fluent_interface() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(3)
.os_major_version(12)
.os_minor_version(5)
.build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_multiple_builds() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref1 = AssemblyOSBuilder::new()
.os_platform_id(1) .os_major_version(10)
.os_minor_version(0)
.build(&mut assembly)
.expect("Should build first OS entry");
let ref2 = AssemblyOSBuilder::new()
.os_platform_id(2) .os_major_version(5)
.os_minor_version(4)
.build(&mut assembly)
.expect("Should build second OS entry");
assert_eq!(ref1.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
assert_eq!(ref2.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
assert!(!std::sync::Arc::ptr_eq(&ref1, &ref2));
Ok(())
}
#[test]
fn test_assemblyos_builder_zero_values() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(0) .os_major_version(0) .os_minor_version(0) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
#[test]
fn test_assemblyos_builder_max_values() -> Result<()> {
let mut assembly = get_test_assembly()?;
let ref_ = AssemblyOSBuilder::new()
.os_platform_id(u32::MAX) .os_major_version(u32::MAX) .os_minor_version(u32::MAX) .build(&mut assembly)
.expect("Should build successfully");
assert_eq!(ref_.kind(), ChangeRefKind::TableRow(TableId::AssemblyOS));
Ok(())
}
}