zeronsd/tests/service/context.rs

use zeronsd::utils::{central_client, local_client};
use zerotier_central_api::types::{Member, MemberConfig};

use super::{
    member::MemberUtil,
    utils::{get_authtoken, get_identity},
};

// TestContext provides all the stuff we need to talk to run tests smoothly
#[derive(Clone)]
pub struct TestContext {
    pub member_config: Option<MemberConfig>,
    pub identity: String,
    pub zerotier: zerotier_one_api::Client,
    pub central: zerotier_central_api::Client,
}

impl TestContext {
    pub fn get_member(&mut self, network_id: String) -> Member {
        let mut member = Member::new(network_id, self.identity.clone());
        if let cfg @ Some(_) = self.member_config.clone() {
            member.config = cfg;
        }

        member
    }

    pub async fn default() -> Self {
        let authtoken = get_authtoken(None).expect("Could not read authtoken");
        let zerotier = local_client(authtoken.clone()).unwrap();
        let identity = get_identity(&zerotier)
            .await
            .expect("Could not retrieve identity from zerotier");

        let token = std::env::var("TOKEN").expect("Please provide TOKEN in the environment");
        let central = central_client(token.clone()).unwrap();

        Self {
            member_config: None,
            identity,
            zerotier,
            central,
        }
    }
}