disma 0.16.4

Discord server management has never been easier!
Documentation
use core::fmt::Debug;

use crate::{category::ExistingCategory, core::changes::category::CategoryChange};

pub trait ExtraCategoriesStrategy {
    fn _type(&self) -> ExtraCategoriesStrategyType;
    fn handle_extra_category(
        &self,
        extra_existing: &ExistingCategory,
        changes: &mut Vec<CategoryChange>,
    );
}

#[derive(Debug, PartialEq)]
pub enum ExtraCategoriesStrategyType {
    Keep,
    Remove,
}

impl Debug for dyn ExtraCategoriesStrategy {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{:?}", self._type())
    }
}

pub struct RemoveExtraCategories {}

impl ExtraCategoriesStrategy for RemoveExtraCategories {
    fn _type(&self) -> ExtraCategoriesStrategyType {
        ExtraCategoriesStrategyType::Remove
    }

    fn handle_extra_category(
        &self,
        extra_existing: &ExistingCategory,
        changes: &mut Vec<CategoryChange>,
    ) {
        changes.push(CategoryChange::Delete(extra_existing.clone()));
    }
}

pub struct KeepExtraCategories {}

impl ExtraCategoriesStrategy for KeepExtraCategories {
    fn _type(&self) -> ExtraCategoriesStrategyType {
        ExtraCategoriesStrategyType::Keep
    }

    fn handle_extra_category(
        &self,
        _extra_existing: &ExistingCategory,
        _changes: &mut Vec<CategoryChange>,
    ) {
    }
}

#[cfg(test)]
mod tests {
    use crate::tests::fixtures::existing::ExistingCategoryFixture;

    use super::*;

    #[test]
    fn when_keeping_extra_categories_should_not_add_changes() {
        let mut changes: Vec<CategoryChange> = Vec::new();
        let extra_category = ExistingCategoryFixture::new().build();

        let strategy = KeepExtraCategories {};
        strategy.handle_extra_category(&extra_category, &mut changes);

        assert!(changes.is_empty());
    }

    #[test]
    fn when_removing_extra_categories_should_add_delete_change() {
        let mut changes: Vec<CategoryChange> = Vec::new();
        let extra_category = ExistingCategoryFixture::new().build();

        let strategy = RemoveExtraCategories {};
        strategy.handle_extra_category(&extra_category, &mut changes);

        assert!(!changes.is_empty());
        assert_eq!(changes, vec![CategoryChange::Delete(extra_category)]);
    }
}