feat(proxy): add mockall automock to all proxy sub-services

This commit is contained in:
GW_MC
2026-07-05 05:29:16 +00:00
parent e44a67f5a8
commit b7891ec200
11 changed files with 11 additions and 0 deletions

View File

@@ -23,6 +23,7 @@ pub struct UpdateAccessRuleParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait AccessRuleService: Send + Sync + 'static { pub trait AccessRuleService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<AccessRuleConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<AccessRuleConfig>;

View File

@@ -17,6 +17,7 @@ pub struct UpdateCacheZoneParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait CacheZoneService: Send + Sync + 'static { pub trait CacheZoneService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<CacheZoneConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<CacheZoneConfig>;

View File

@@ -17,6 +17,7 @@ pub struct ConfigInheritanceRecord {
pub applied_at: chrono::NaiveDateTime, pub applied_at: chrono::NaiveDateTime,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait ConfigInheritanceService: Send + Sync + 'static { pub trait ConfigInheritanceService: Send + Sync + 'static {
async fn add( async fn add(

View File

@@ -19,6 +19,7 @@ pub struct UpdateLimitRuleParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait LimitRuleService: Send + Sync + 'static { pub trait LimitRuleService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<LimitRuleConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<LimitRuleConfig>;

View File

@@ -17,6 +17,7 @@ pub struct UpdateLimitZoneParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait LimitZoneService: Send + Sync + 'static { pub trait LimitZoneService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<LimitZoneConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<LimitZoneConfig>;

View File

@@ -21,6 +21,7 @@ pub struct UpdateLocationBlockParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait LocationBlockService: Send + Sync + 'static { pub trait LocationBlockService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<LocationBlockConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<LocationBlockConfig>;

View File

@@ -19,6 +19,7 @@ pub struct UpdateLogSettingParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait LogSettingService: Send + Sync + 'static { pub trait LogSettingService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<LogSettingConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<LogSettingConfig>;

View File

@@ -23,6 +23,7 @@ pub struct UpdateProxySettingParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait ProxySettingService: Send + Sync + 'static { pub trait ProxySettingService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<ProxySettingConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<ProxySettingConfig>;

View File

@@ -21,6 +21,7 @@ pub struct UpdateRewriteRuleParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait RewriteRuleService: Send + Sync + 'static { pub trait RewriteRuleService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<RewriteRuleConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<RewriteRuleConfig>;

View File

@@ -22,6 +22,7 @@ pub struct UpdateServerBlockParams {
pub override_of_id: Option<Option<Uuid>>, pub override_of_id: Option<Option<Uuid>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait ServerBlockService: Send + Sync + 'static { pub trait ServerBlockService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<ServerBlockConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<ServerBlockConfig>;

View File

@@ -18,6 +18,7 @@ pub struct UpdateSslCertificateParams {
pub expiry_date: Option<chrono::DateTime<Utc>>, pub expiry_date: Option<chrono::DateTime<Utc>>,
} }
#[cfg_attr(test, mockall::automock)]
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait SslCertificateService: Send + Sync + 'static { pub trait SslCertificateService: Send + Sync + 'static {
async fn get(&self, id: Uuid) -> ProxyServiceResult<SslCertificateConfig>; async fn get(&self, id: Uuid) -> ProxyServiceResult<SslCertificateConfig>;