diff --git a/crates/_types/enums/src/by_table/model_weights/weights_types.rs b/crates/_types/enums/src/by_table/model_weights/weights_types.rs index e50b65d560..ea985c7ab9 100644 --- a/crates/_types/enums/src/by_table/model_weights/weights_types.rs +++ b/crates/_types/enums/src/by_table/model_weights/weights_types.rs @@ -5,6 +5,7 @@ use strum::EnumCount; #[cfg(test)] use strum::EnumIter; + #[cfg_attr(test, derive(EnumIter, EnumCount))] #[derive(Clone, Copy, Eq, PartialEq, Hash, Ord, PartialOrd, sqlx::Type, Deserialize, Serialize,Debug)] pub enum WeightsType { diff --git a/crates/lib/mysql_queries/src/queries/model_weights/get_weight.rs b/crates/lib/mysql_queries/src/queries/model_weights/get_weight.rs index 041f14ae92..6c05df7e27 100644 --- a/crates/lib/mysql_queries/src/queries/model_weights/get_weight.rs +++ b/crates/lib/mysql_queries/src/queries/model_weights/get_weight.rs @@ -4,10 +4,10 @@ use sqlx::{ MySqlPool, MySql }; use sqlx::pool::PoolConnection; use anyhow::anyhow; use log::{ error }; -use enums::by_table::model_weights::{ - weights_types::WeightsType, - weights_category::WeightsCategory, -}; + +use enums::by_table::model_weights::weights_types::WeightsType; +use enums::by_table::model_weights::weights_category::WeightsCategory; + use enums::common::visibility::Visibility; use tokens::tokens::{ users::UserToken, model_weights::ModelWeightToken }; use chrono::{ DateTime, Utc }; diff --git a/crates/lib/mysql_queries/src/queries/model_weights/tests.rs b/crates/lib/mysql_queries/src/queries/model_weights/tests.rs index eadf785465..30f90aaaa0 100644 --- a/crates/lib/mysql_queries/src/queries/model_weights/tests.rs +++ b/crates/lib/mysql_queries/src/queries/model_weights/tests.rs @@ -56,7 +56,6 @@ mod tests { #[tokio::test] async fn test_create_weights() -> AnyhowResult<()> { - let db_connection_string = DEFAULT_MYSQL_CONNECTION_STRING; let mut rng = rand::thread_rng(); let random_number: u32 = rng.gen(); @@ -128,4 +127,12 @@ mod tests { Ok(()) } + #[tokio::test] + async fn test_delete_weights() -> AnyhowResult<()> { + + + Ok(()) + } + + }