#[cfg(any(test, feature = "test-support"))]
use crate::mock_client::{ExpectationBuilder, MockClient};
#[cfg(any(test, feature = "test-support"))]
pub trait CosmosdbMockHelpers {
fn expect_list_accounts(&mut self, subscription_id: &str) -> ExpectationBuilder<'_>;
fn expect_get_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_create_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_delete_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_list_sql_databases(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_get_sql_database(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_create_sql_database(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_list_sql_containers(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> ExpectationBuilder<'_>;
fn expect_get_sql_container(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
container_name: &str,
) -> ExpectationBuilder<'_>;
}
#[cfg(any(test, feature = "test-support"))]
impl CosmosdbMockHelpers for MockClient {
fn expect_list_accounts(
&mut self,
subscription_id: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/providers/Microsoft.DocumentDB/databaseAccounts"
);
self.expect_get(&path)
}
fn expect_get_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}"
);
self.expect_get(&path)
}
fn expect_create_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}"
);
self.expect_put(&path)
}
fn expect_delete_account(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}"
);
self.expect_delete(&path)
}
fn expect_list_sql_databases(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}/sqlDatabases"
);
self.expect_get(&path)
}
fn expect_get_sql_database(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}/sqlDatabases/{database_name}"
);
self.expect_get(&path)
}
fn expect_create_sql_database(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}/sqlDatabases/{database_name}"
);
self.expect_put(&path)
}
fn expect_list_sql_containers(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}/sqlDatabases/{database_name}/containers"
);
self.expect_get(&path)
}
fn expect_get_sql_container(
&mut self,
subscription_id: &str,
resource_group_name: &str,
account_name: &str,
database_name: &str,
container_name: &str,
) -> crate::mock_client::ExpectationBuilder<'_> {
let path = format!(
"/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.DocumentDB/databaseAccounts/{account_name}/sqlDatabases/{database_name}/containers/{container_name}"
);
self.expect_get(&path)
}
}