move tester command under debug

Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-06-23 09:06:25 +00:00
parent 5805394ca5
commit 2387f7f955
4 changed files with 13 additions and 12 deletions

View file

@ -1,8 +1,10 @@
mod commands; mod commands;
pub(crate) mod tester;
use clap::Subcommand; use clap::Subcommand;
use conduit::Result; use conduit::Result;
use ruma::{events::room::message::RoomMessageEventContent, EventId, OwnedRoomOrAliasId, RoomId, ServerName}; use ruma::{events::room::message::RoomMessageEventContent, EventId, OwnedRoomOrAliasId, RoomId, ServerName};
use tester::TesterCommand;
use self::commands::*; use self::commands::*;
@ -157,6 +159,10 @@ pub(super) enum DebugCommand {
/// - Print extended memory usage /// - Print extended memory usage
MemoryStats, MemoryStats,
/// - Developer test stubs
#[command(subcommand)]
Tester(TesterCommand),
} }
pub(super) async fn process(command: DebugCommand, body: Vec<&str>) -> Result<RoomMessageEventContent> { pub(super) async fn process(command: DebugCommand, body: Vec<&str>) -> Result<RoomMessageEventContent> {
@ -207,5 +213,6 @@ pub(super) async fn process(command: DebugCommand, body: Vec<&str>) -> Result<Ro
no_cache, no_cache,
} => resolve_true_destination(body, server_name, no_cache).await?, } => resolve_true_destination(body, server_name, no_cache).await?,
DebugCommand::MemoryStats => memory_stats(), DebugCommand::MemoryStats => memory_stats(),
DebugCommand::Tester(command) => tester::process(command, body).await?,
}) })
} }

View file

@ -4,15 +4,15 @@ use crate::Result;
#[derive(clap::Subcommand)] #[derive(clap::Subcommand)]
#[cfg_attr(test, derive(Debug))] #[cfg_attr(test, derive(Debug))]
pub(super) enum TesterCommands { pub(crate) enum TesterCommand {
Tester, Tester,
Timer, Timer,
} }
pub(super) async fn process(command: TesterCommands, body: Vec<&str>) -> Result<RoomMessageEventContent> { pub(super) async fn process(command: TesterCommand, body: Vec<&str>) -> Result<RoomMessageEventContent> {
match command { match command {
TesterCommands::Tester => tester(body).await, TesterCommand::Tester => tester(body).await,
TesterCommands::Timer => timer(body).await, TesterCommand::Timer => timer(body).await,
} }
} }

View file

@ -13,11 +13,10 @@ use conduit::Result;
pub(crate) use service::admin::{Command, Service}; pub(crate) use service::admin::{Command, Service};
use service::admin::{CommandOutput, CommandResult, HandlerResult}; use service::admin::{CommandOutput, CommandResult, HandlerResult};
use self::{fsck::FsckCommand, tester::TesterCommands};
use crate::{ use crate::{
appservice, appservice::AppserviceCommand, debug, debug::DebugCommand, federation, federation::FederationCommand, appservice, appservice::AppserviceCommand, debug, debug::DebugCommand, federation, federation::FederationCommand,
fsck, media, media::MediaCommand, query, query::QueryCommand, room, room::RoomCommand, server, fsck, fsck::FsckCommand, media, media::MediaCommand, query, query::QueryCommand, room, room::RoomCommand, server,
server::ServerCommand, services, tester, user, user::UserCommand, server::ServerCommand, services, user, user::UserCommand,
}; };
pub(crate) const PAGE_SIZE: usize = 100; pub(crate) const PAGE_SIZE: usize = 100;
@ -60,9 +59,6 @@ pub(crate) enum AdminCommand {
#[command(subcommand)] #[command(subcommand)]
/// - Query all the database getters and iterators /// - Query all the database getters and iterators
Fsck(FsckCommand), Fsck(FsckCommand),
#[command(subcommand)]
Tester(TesterCommands),
} }
#[must_use] #[must_use]
@ -169,7 +165,6 @@ async fn process_admin_command(command: AdminCommand, body: Vec<&str>) -> Result
AdminCommand::Debug(command) => debug::process(command, body).await?, AdminCommand::Debug(command) => debug::process(command, body).await?,
AdminCommand::Query(command) => query::process(command, body).await?, AdminCommand::Query(command) => query::process(command, body).await?,
AdminCommand::Fsck(command) => fsck::process(command, body).await?, AdminCommand::Fsck(command) => fsck::process(command, body).await?,
AdminCommand::Tester(command) => tester::process(command, body).await?,
}; };
Ok(reply_message_content) Ok(reply_message_content)

View file

@ -9,7 +9,6 @@ pub(crate) mod media;
pub(crate) mod query; pub(crate) mod query;
pub(crate) mod room; pub(crate) mod room;
pub(crate) mod server; pub(crate) mod server;
pub(crate) mod tester;
pub(crate) mod user; pub(crate) mod user;
pub(crate) mod utils; pub(crate) mod utils;