joystream/runtime-modules/content-directory/src/tests/update_entity_creation_voucher.rs

use super::*;

#[test]
fn create_entity_creation_voucher_success() {
    with_test_externalities(|| {
        // Create simple class with default permissions
        assert_ok!(create_simple_class(LEAD_ORIGIN, ClassType::Valid));

        // Runtime state before tested call

        let entity_controller = EntityController::Member(FIRST_MEMBER_ID);
        assert!(!entity_creation_voucher_exists(
            FIRST_CLASS_ID,
            &entity_controller
        ));

        // Events number before tested call
        let number_of_events_before_call = System::events().len();

        // Create entities creation voucher for chosen controller
        assert_ok!(update_entity_creation_voucher(
            LEAD_ORIGIN,
            FIRST_CLASS_ID,
            entity_controller.clone(),
            IndividualEntitiesCreationLimit::get()
        ));

        // Runtime tested state after call

        // Ensure entity creation voucher for chosen controller created succesfully
        let entity_voucher = EntityCreationVoucher::new(IndividualEntitiesCreationLimit::get());

        assert_eq!(
            entity_creation_vouchers(FIRST_CLASS_ID, &entity_controller),
            entity_voucher,
        );

        let entity_creation_voucher_created_event = get_test_event(
            RawEvent::EntityCreationVoucherCreated(entity_controller, entity_voucher),
        );

        // Last event checked
        assert_event(
            entity_creation_voucher_created_event,
            number_of_events_before_call + 1,
        );
    })
}

#[test]
fn update_entity_creation_voucher_success() {
    with_test_externalities(|| {
        // Create simple class with default permissions
        assert_ok!(create_simple_class(LEAD_ORIGIN, ClassType::Valid));

        let actor = Actor::Member(FIRST_MEMBER_ID);

        // Update class permissions to force any member be available to create entities
        assert_ok!(update_class_permissions(
            LEAD_ORIGIN,
            FIRST_CLASS_ID,
            Some(true),
            None,
            None,
            None
        ));

        let entity_controller = EntityController::<MemberId>::from_actor::<Runtime>(&actor);

        // Create entity
        assert_ok!(create_entity(FIRST_MEMBER_ORIGIN, FIRST_CLASS_ID, actor));

        // Runtime state before tested call

        let mut entity_creation_voucher =
            entity_creation_vouchers(FIRST_CLASS_ID, &entity_controller);

        // Events number before tested call
        let number_of_events_before_call = System::events().len();

        // Update entities creation voucher for chosen controller
        let new_maximum_entities_count = entity_creation_voucher.maximum_entities_count - 1;
        assert_ok!(update_entity_creation_voucher(
            LEAD_ORIGIN,
            FIRST_CLASS_ID,
            entity_controller.clone(),
            new_maximum_entities_count
        ));

        // Runtime tested state after call

        // Ensure  entity creation voucher for chosen controller updated succesfully
        entity_creation_voucher.set_maximum_entities_count(new_maximum_entities_count);
        assert_eq!(
            entity_creation_vouchers(FIRST_CLASS_ID, &entity_controller),
            entity_creation_voucher
        );

        let entity_creation_voucher_created_event = get_test_event(
            RawEvent::EntityCreationVoucherUpdated(entity_controller, entity_creation_voucher),
        );

        // Last event checked
        assert_event(
            entity_creation_voucher_created_event,
            number_of_events_before_call + 1,
        );
    })
}

#[test]
fn update_entity_creation_voucher_lead_auth_failed() {
    with_test_externalities(|| {
        // Create simple class with default permissions
        assert_ok!(create_simple_class(LEAD_ORIGIN, ClassType::Valid));

        // Runtime state before tested call

        let entity_controller = EntityController::Member(FIRST_MEMBER_ID);

        // Events number before tested call
        let number_of_events_before_call = System::events().len();

        // Make an attempt to create entities creation voucher for chosen controller under non lead origin
        let update_entity_creation_voucher_result = update_entity_creation_voucher(
            FIRST_MEMBER_ORIGIN,
            FIRST_CLASS_ID,
            entity_controller,
            IndividualEntitiesCreationLimit::get(),
        );

        // Failure checked
        assert_failure(
            update_entity_creation_voucher_result,
            Error::<Runtime>::LeadAuthFailed,
            number_of_events_before_call,
        );
    })
}

#[test]
fn update_entity_creation_voucher_class_does_not_exist() {
    with_test_externalities(|| {
        // Runtime state before tested call

        let entity_controller = EntityController::Member(FIRST_MEMBER_ID);

        // Events number before tested call
        let number_of_events_before_call = System::events().len();

        // Make an attempt to create entities creation voucher for chosen controller and non existent class
        let update_entity_creation_voucher_result = update_entity_creation_voucher(
            LEAD_ORIGIN,
            UNKNOWN_CLASS_ID,
            entity_controller,
            IndividualEntitiesCreationLimit::get(),
        );

        // Failure checked
        assert_failure(
            update_entity_creation_voucher_result,
            Error::<Runtime>::ClassNotFound,
            number_of_events_before_call,
        );
    })
}

#[test]
fn update_entity_creation_voucher_individual_creation_limit_exceed() {
    with_test_externalities(|| {
        // Create simple class with default permissions
        assert_ok!(create_simple_class(LEAD_ORIGIN, ClassType::Valid));

        // Runtime state before tested call

        let entity_controller = EntityController::Member(FIRST_MEMBER_ID);

        // Events number before tested call
        let number_of_events_before_call = System::events().len();

        // Make an attempt to create entities creation voucher for chosen controller with maximum_entities_count
        // value that exceeds IndividualEntitiesCreationLimit
        let update_entity_creation_voucher_result = update_entity_creation_voucher(
            LEAD_ORIGIN,
            FIRST_CLASS_ID,
            entity_controller,
            IndividualEntitiesCreationLimit::get() + 1,
        );

        // Failure checked
        assert_failure(
            update_entity_creation_voucher_result,
            Error::<Runtime>::NumberOfClassEntitiesPerActorConstraintViolated,
            number_of_events_before_call,
        );
    })
}