use core::str;
use crate::{
command::{Command, CommandLevel},
error::Result,
};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct SetMinerPowerLimit {
pub limit: u32,
}
impl SetMinerPowerLimit {
pub const fn new(limit: u32) -> Self {
Self { limit }
}
}
impl Command for SetMinerPowerLimit {
type Response = String;
const CMD_NAME: &'static str = "set.miner.power_limit";
const LEVEL: CommandLevel = CommandLevel::Secured;
fn params(&self) -> Result<Option<impl serde::Serialize>> {
Ok(Some(self.limit))
}
}
#[cfg(test)]
mod tests {
use crate::account::Account;
use crate::auth_data::AuthData;
use super::*;
const RESPONSE_JSON: &str =
r#"{"code":0,"when":1692685496,"msg":"ok","desc":"set.miner.power_limit"}"#;
#[test]
fn request_has_param() {
let cmd = SetMinerPowerLimit::new(3000);
assert!(cmd.is_secured());
assert!(!cmd.is_encrypted());
let auth = AuthData::new::<SetMinerPowerLimit>(Account::Super, "pass", "salt").unwrap();
let encoded = cmd.encode(Some(auth)).unwrap();
let json: serde_json::Value = serde_json::from_slice(&encoded.0).unwrap();
assert_eq!(json["cmd"], "set.miner.power_limit");
assert_eq!(json["param"], 3000);
assert_eq!(json["account"], "super");
assert!(json["ts"].as_u64().unwrap() > 0);
assert_eq!(json["token"].as_str().unwrap().len(), 8);
}
#[test]
fn request_without_auth_omits_token() {
let cmd = SetMinerPowerLimit::new(3000);
let encoded = cmd.encode(None).unwrap();
let json: serde_json::Value = serde_json::from_slice(&encoded.0).unwrap();
assert!(json.get("token").is_none());
assert_eq!(json["param"], 3000);
}
#[test]
fn deserialize_response_fixture() {
let response = SetMinerPowerLimit::parse_response(RESPONSE_JSON.as_bytes()).unwrap();
assert!(response.code.is_success());
assert_eq!(response.desc, "set.miner.power_limit");
assert_eq!(response.when, 1692685496);
assert_eq!(response.msg, "ok");
}
}