xref: /aosp_15_r20/tools/asuite/adevice/tests/common/fakes.rs (revision c2e18aaa1096c836b086f94603d04f4eb9cf37f5)

use adevice::adevice::{Device, Host, Profiler};
use adevice::commands::{AdbAction, AdbCommand};
use adevice::fingerprint::FileMetadata;
use adevice::metrics::MetricSender;
use adevice::tracking::Config;
use anyhow::Result;
use std::cell::RefCell;
use std::collections::{HashMap, HashSet};
use std::path::{Path, PathBuf};

#[derive(Default)]
pub struct FakeHost {
    /// Files on the filesystem, relative to PRODUCT_OUT
    files: HashMap<PathBuf, FileMetadata>,
    /// Dependencies from ninja, relative to PRODUCT_OUT
    tracked_files: Vec<String>,
}

// We allow dead code here and below as not all the tests uses all the methods,
// and they get marked as dead in soong when compiling just one test at a time.
#[allow(dead_code)]
impl FakeHost {
    pub fn new(files: &HashMap<PathBuf, FileMetadata>, tracked_files: &[String]) -> FakeHost {
        FakeHost { files: files.clone(), tracked_files: tracked_files.to_owned() }
    }

    /// Returns true if `path` starts with one of the `partitions`
    fn on_a_partition(path: &Path, partitions: &[PathBuf]) -> bool {
        for p in partitions {
            if path.starts_with(p) {
                return true;
            }
        }
        false
    }
}

#[allow(dead_code)]
#[derive(Default)]
pub struct FakeDevice {
    /// Apks that are installed with "adb install" on the /data partition.
    /// Used to see if we should warn the user about potential problems.
    installed_apks: HashSet<String>,

    /// Files on the filesystem.
    /// User passes some to start, but "push" and "clean" commands will affect it.
    files: HashMap<PathBuf, FileMetadata>,

    // Files pushed to the device via an `adb_command`
    pushes: RefCell<Vec<PathBuf>>,
    // Files and directories removed from the device via `adb_command`
    removes: RefCell<Vec<PathBuf>>,

    // Cmds that are issued with run_raw_adb_command;
    raw_cmds: RefCell<Vec<String>>,

    // How many times has wait() beeng called on the fake.
    wait_called: RefCell<u32>,
}

#[allow(dead_code)]
impl FakeDevice {
    pub fn new(files: &HashMap<PathBuf, FileMetadata>) -> FakeDevice {
        FakeDevice { files: files.clone(), ..Default::default() }
    }

    /// Returns the ordered list of all removed files or dirs.
    pub fn removes(&self) -> Vec<PathBuf> {
        self.removes.borrow().clone()
    }

    /// Returns orderd list of all pushed files.
    #[allow(dead_code)]
    pub fn pushes(&self) -> Vec<PathBuf> {
        self.pushes.borrow().clone()
    }

    /// Returns orderd list of all raw adb commands.
    #[allow(dead_code)]
    pub fn raw_cmds(&self) -> Vec<String> {
        self.raw_cmds.borrow().clone()
    }
    pub fn wait_calls(&self) -> u32 {
        *self.wait_called.borrow()
    }
}

impl Host for FakeHost {
    fn fingerprint(
        &self,
        _partition_root: &Path,
        partitions: &[PathBuf],
    ) -> Result<HashMap<PathBuf, FileMetadata>> {
        let mut files = self.files.clone();
        files.retain(|path, _m| Self::on_a_partition(path, partitions));
        Ok(files)
    }

    fn tracked_files(&self, _config: &Config) -> Result<Vec<String>> {
        Ok(self.tracked_files.clone())
    }
}

impl Device for FakeDevice {
    // Convert "push" into updating the filesystem, ignore everything else.
    fn run_adb_command(&self, cmd: &AdbCommand) -> Result<String> {
        match cmd.action {
            AdbAction::Push { .. } => self.pushes.borrow_mut().push(cmd.file.clone()),
            AdbAction::DeleteDir { .. } | AdbAction::DeleteFile => {
                self.removes.borrow_mut().push(cmd.file.clone())
            }
            _ => (),
        }
        Ok(String::new())
    }
    fn run_raw_adb_command(&self, cmds: &[String]) -> Result<String> {
        self.raw_cmds.borrow_mut().push(cmds.join(" "));
        Ok(String::new())
    }

    // No need to do anything.
    fn reboot(&self) -> Result<String> {
        Ok(String::new())
    }

    // No need to do anything.
    fn soft_restart(&self) -> Result<String> {
        Ok(String::new())
    }

    fn fingerprint(&self, _partitions: &[String]) -> Result<HashMap<PathBuf, FileMetadata>> {
        // Technically, I should filter the result to ensure it only includes `partitions`
        Ok(self.files.clone())
    }

    fn get_installed_apks(&self) -> Result<HashSet<String>> {
        Ok(self.installed_apks.clone())
    }

    fn wait(&self, _profiler: &mut Profiler) -> Result<String> {
        let mut counter = self.wait_called.borrow_mut();
        *counter += 1;
        Ok(String::new())
    }
    fn prep_after_flash(&self, _profiler: &mut Profiler) -> Result<()> {
        Ok(())
    }
}

pub struct FakeMetricSender {}

#[allow(dead_code)]
impl FakeMetricSender {
    pub fn new() -> Self {
        FakeMetricSender {}
    }
}
impl MetricSender for FakeMetricSender {
    // TODO: Capture and test metrics.
    fn add_start_event(&mut self, _command_line: &str, _add_start_event: &str) {}

    fn add_action_event(&mut self, _action: &str, _duration: std::time::Duration) {}

    fn add_action_event_with_files_changed(
        &mut self,
        _action: &str,
        _duration: std::time::Duration,
        _files_changed: std::vec::Vec<String>,
    ) {
    }

    fn add_profiler_events(&mut self, _profiler: &adevice::adevice::Profiler) {}

    fn add_exit_event(&mut self, _output: &str, _exit_code: i32) {}

    fn display_survey(&mut self) {}
}