archives/steamos-manager
archives/steamos-manager
2
0
Fork 1
mirror of https://gitlab.steamos.cloud/holo/steamos-manager.git synced 2025-07-06 14:40:29 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

wifi: Add power management tests

Browse source
This commit is contained in:
Vicki Pfau 2024-09-03 21:27:37 -07:00
parent b8f3f79e0b
commit fc9747a114
1 changed files with 91 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

91
src/wifi.rs
View file

@ -281,6 +281,7 @@ pub(crate) async fn set_wifi_power_management_state(state: WifiPowerManagement)
mod test {
use super::*;
use crate::{enum_on_off, enum_roundtrip, testing};
use std::ffi::OsStr;
use tokio::fs::{create_dir_all, read_to_string, remove_dir, try_exists, write};
#[test]
@ -374,6 +375,96 @@ mod test {
);
}
#[tokio::test]
async fn test_power_management() {
let h = testing::start();
fn process_output(executable: &OsStr, args: &[&OsStr]) -> Result<(i32, String)> {
ensure!(executable.to_string_lossy() == "/usr/bin/iw", "Not iw");
ensure!(args[0] == "dev", "Not dev");
if args.len() < 2 {
return Ok((0, String::from("Interface eth0")));
}
ensure!(args[1] == "eth0", "Not eth0");
ensure!(args[3] == "power_save", "Not power_save");
match args[2].to_str() {
Some("get") => Ok((0, String::from("Power save: on"))),
Some("set") => {
ensure!(args[4] == "on");
Ok((0, String::new()))
}
_ => bail!("Unknown query"),
}
}
h.test.process_cb.set(process_output);
assert_eq!(
get_wifi_power_management_state().await.expect("get"),
WifiPowerManagement::Enabled
);
assert!(
set_wifi_power_management_state(WifiPowerManagement::Enabled)
.await
.is_ok()
);
assert!(
set_wifi_power_management_state(WifiPowerManagement::Disabled)
.await
.is_err()
);
}
#[tokio::test]
async fn test_power_management_disabled() {
let h = testing::start();
fn process_output(executable: &OsStr, args: &[&OsStr]) -> Result<(i32, String)> {
ensure!(executable.to_string_lossy() == "/usr/bin/iw", "Not iw");
ensure!(args[0] == "dev", "Not dev");
if args.len() < 2 {
return Ok((0, String::from("Interface eth0")));
}
ensure!(args[1] == "eth0", "Not eth0");
ensure!(args[3] == "power_save", "Not power_save");
match args[2].to_str() {
Some("get") => Ok((0, String::from("Power save: off"))),
_ => bail!("Unknown query"),
}
}
h.test.process_cb.set(process_output);
assert_eq!(
get_wifi_power_management_state().await.expect("get"),
WifiPowerManagement::Disabled
);
}
#[tokio::test]
async fn test_power_management_multi_iface() {
let h = testing::start();
fn process_output(executable: &OsStr, args: &[&OsStr]) -> Result<(i32, String)> {
ensure!(executable.to_string_lossy() == "/usr/bin/iw", "Not iw");
ensure!(args[0] == "dev", "Not dev");
if args.len() < 2 {
return Ok((0, String::from("Interface eth0\nInterface eth1")));
}
ensure!(args[3] == "power_save", "Not power_save");
match args[1].to_str() {
Some("eth0") => Ok((0, String::from("Power save: off"))),
Some("eth1") => Ok((0, String::from("Power save: on"))),
_ => bail!("Unknown query"),
}
}
h.test.process_cb.set(process_output);
assert_eq!(
get_wifi_power_management_state().await.expect("get"),
WifiPowerManagement::Enabled
);
}
#[test]
fn wifi_debug_mode_roundtrip() {
enum_roundtrip!(WifiDebugMode {