src/sys/pkg/tests/pkg-resolver/src/dynamic_repositories_disabled.rs

// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

/// This module tests pkg_resolver's RepositoryManager when
/// dynamic repository configs.
use {
    fuchsia_async as fasync,
    fuchsia_zircon::Status,
    lib::{
        get_repos, make_repo, make_repo_config, mock_filesystem, Config, DirOrProxy, MountsBuilder,
        TestEnvBuilder,
    },
};

#[fasync::run_singlethreaded(test)]
async fn no_load_dynamic_repos_if_disabled() {
    let env = TestEnvBuilder::new()
        .mounts(
            MountsBuilder::new()
                .dynamic_repositories(make_repo_config(&make_repo()))
                .config(Config { enable_dynamic_configuration: false })
                .build(),
        )
        .build()
        .await;

    assert_eq!(get_repos(&env.proxies.repo_manager).await, vec![]);

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn add_succeeds() {
    let env = TestEnvBuilder::new()
        .mounts(MountsBuilder::new().config(Config { enable_dynamic_configuration: true }).build())
        .build()
        .await;
    let repo = make_repo();

    let () = env.proxies.repo_manager.add(repo.clone().into()).await.unwrap().unwrap();
    assert_eq!(get_repos(&env.proxies.repo_manager).await, vec![repo]);

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn add_fails_if_disabled() {
    let env = TestEnvBuilder::new()
        .mounts(MountsBuilder::new().config(Config { enable_dynamic_configuration: false }).build())
        .build()
        .await;
    let repo = make_repo();

    assert_eq!(
        env.proxies.repo_manager.add(repo.clone().into()).await.unwrap().unwrap_err(),
        Status::ACCESS_DENIED.into_raw()
    );
    assert_eq!(get_repos(&env.proxies.repo_manager).await, vec![]);

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn remove_fails_with_not_found() {
    let env = TestEnvBuilder::new()
        .mounts(MountsBuilder::new().config(Config { enable_dynamic_configuration: true }).build())
        .build()
        .await;

    assert_eq!(
        env.proxies.repo_manager.remove("fuchsia-pkg://example.com").await.unwrap().unwrap_err(),
        Status::NOT_FOUND.into_raw()
    );

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn remove_fails_with_access_denied_if_disabled() {
    let env = TestEnvBuilder::new()
        .mounts(MountsBuilder::new().config(Config { enable_dynamic_configuration: false }).build())
        .build()
        .await;

    assert_eq!(
        env.proxies.repo_manager.remove("fuchsia-pkg://example.com").await.unwrap().unwrap_err(),
        Status::ACCESS_DENIED.into_raw()
    );

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn attempt_to_open_persisted_dynamic_repos() {
    let (proxy, open_counts) = mock_filesystem::spawn_directory_handler();
    let env = TestEnvBuilder::new()
        .mounts(
            MountsBuilder::new()
                .config(Config { enable_dynamic_configuration: true })
                .pkg_resolver_data(DirOrProxy::Proxy(proxy))
                .build(),
        )
        .build()
        .await;

    // Waits for pkg_resolver to be initialized
    get_repos(&env.proxies.repo_manager).await;

    assert_eq!(open_counts.lock().get("repositories.json"), Some(&1));

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn no_attempt_to_open_persisted_dynamic_repos_if_disabled() {
    let (proxy, open_counts) = mock_filesystem::spawn_directory_handler();
    let env = TestEnvBuilder::new()
        .mounts(
            MountsBuilder::new()
                .config(Config { enable_dynamic_configuration: false })
                .pkg_resolver_data(DirOrProxy::Proxy(proxy))
                .build(),
        )
        .build()
        .await;

    // Waits for pkg_resolver to be initialized
    get_repos(&env.proxies.repo_manager).await;

    assert_eq!(open_counts.lock().get("repositories.json"), None);

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn dynamic_repositories_disabled_if_missing_config() {
    let env = TestEnvBuilder::new().mounts(MountsBuilder::new().build()).build().await;
    let repo = make_repo();

    assert_eq!(
        env.proxies.repo_manager.add(repo.clone().into()).await.unwrap().unwrap_err(),
        Status::ACCESS_DENIED.into_raw()
    );
    assert_eq!(get_repos(&env.proxies.repo_manager).await, vec![]);

    env.stop().await;
}

#[fasync::run_singlethreaded(test)]
async fn load_dynamic_repos() {
    let repo = make_repo();
    let env = TestEnvBuilder::new()
        .mounts(
            MountsBuilder::new()
                .config(Config { enable_dynamic_configuration: true })
                .dynamic_repositories(make_repo_config(&repo))
                .build(),
        )
        .build()
        .await;

    assert_eq!(get_repos(&env.proxies.repo_manager).await, vec![repo]);

    env.stop().await;
}