NLnetLabs-krill/tests/suspend.rs

//! Test suspension and un-suspension logic.

use std::fs;

use krill::test::*;
use rpki::ca::idexchange::CaHandle;
use rpki::repository::resources::ResourceSet;

#[tokio::test]
async fn test_suspension() {
    //  Uses the following lay-out:
    //
    //                  TA
    //                   |
    //                testbed
    //                   |
    //                  CA

    // Start krill with:
    //  testbed enabled
    //  ca_refresh disabled (we will trigger individual CA refreshes manually)
    //  suspend enabled
    let krill_dir = start_krill_with_default_test_config(true, false, true, false).await;

    let testbed = ca_handle("testbed");
    let ca = ca_handle("CA");
    let ca_res = ipv4_resources("10.0.0.0/16");

    async fn expect_not_suspended(ca: &CaHandle, child: &CaHandle) {
        let rcn_0 = rcn(0);
        let child_handle = child.convert();

        let mut expected_files = expected_mft_and_crl(ca, &rcn_0).await;
        expected_files.push(expected_issued_cer(&child.convert(), &rcn_0).await);
        assert!(will_publish_embedded("CA should have mft, crl and cert for child", ca, &expected_files).await);

        let ca_info = ca_details(ca).await;
        assert!(ca_info.children().contains(&child_handle));
        assert!(!ca_info.suspended_children().contains(&child_handle));
    }

    async fn expect_suspended(ca: &CaHandle, child: &CaHandle) {
        let rcn_0 = rcn(0);
        let child_handle = child.convert();

        let expected_files = expected_mft_and_crl(ca, &rcn_0).await;
        assert!(will_publish_embedded("CA should have mft, crl only", ca, &expected_files).await);

        let ca_info = ca_details(ca).await;
        assert!(ca_info.children().contains(&child_handle));
        assert!(ca_info.suspended_children().contains(&child_handle));
    }

    // Wait for testbed to come up
    {
        assert!(ca_contains_resources(&testbed, &ResourceSet::all()).await);
    }

    // Set up CA under testbed and verify that the certificate is published
    {
        set_up_ca_with_repo(&ca).await;
        set_up_ca_under_parent_with_resources(&ca, &testbed, &ca_res).await;
    }

    // Verify that testbed published the certificate for CA, and that its state is 'active'
    {
        expect_not_suspended(&testbed, &ca).await;
    }

    // Wait a bit, and then refresh testbed only, it should find that
    // the child 'CA' has not been updating, and will suspend it.
    {
        sleep_seconds(15).await;

        cas_refresh_single(&testbed).await;

        // schedule check suspension bg job to now
        cas_suspend_all().await;
        expect_suspended(&testbed, &ca).await;
    }

    // Let "CA" refresh with testbed, this should 'un-suspend' it.
    {
        cas_refresh_single(&ca).await;

        // schedule check suspension bg job to now
        cas_suspend_all().await;
        expect_not_suspended(&testbed, &ca).await;
    }

    // CAs can also be suspended explicitly, regardless of their last known connection
    {
        ca_suspend_child(&testbed, &ca).await;
        expect_suspended(&testbed, &ca).await;
    }

    // And they can be manually unsuspended as well
    {
        ca_unsuspend_child(&testbed, &ca).await;
        expect_not_suspended(&testbed, &ca).await;
    }

    let _ = fs::remove_dir_all(krill_dir);
}