use async_trait::async_trait;
use thiserror::Error;
use crate::directory_sync::{Directory, DirectoryId, DirectorySync};
use crate::{ResponseExt, WorkOsError, WorkOsResult};
#[derive(Debug, Error)]
pub enum GetDirectoryError {}
impl From<GetDirectoryError> for WorkOsError<GetDirectoryError> {
fn from(err: GetDirectoryError) -> Self {
Self::Operation(err)
}
}
#[async_trait]
pub trait GetDirectory {
async fn get_directory(&self, id: &DirectoryId) -> WorkOsResult<Directory, GetDirectoryError>;
}
#[async_trait]
impl GetDirectory for DirectorySync<'_> {
async fn get_directory(&self, id: &DirectoryId) -> WorkOsResult<Directory, GetDirectoryError> {
let url = self.workos.base_url().join(&format!("/directories/{id}"))?;
let directory = self
.workos
.client()
.get(url)
.bearer_auth(self.workos.key())
.send()
.await?
.handle_unauthorized_or_generic_error()
.await?
.json::<Directory>()
.await?;
Ok(directory)
}
}
#[cfg(test)]
mod test {
use matches::assert_matches;
use serde_json::json;
use tokio;
use crate::{ApiKey, WorkOs};
use super::*;
#[tokio::test]
async fn it_calls_the_get_directory_endpoint() {
let mut server = mockito::Server::new_async().await;
let workos = WorkOs::builder(&ApiKey::from("sk_example_123456789"))
.base_url(&server.url())
.unwrap()
.build();
server
.mock("GET", "/directories/directory_01ECAZ4NV9QMV47GW873HDCX74")
.match_header("Authorization", "Bearer sk_example_123456789")
.with_status(200)
.with_body(
json!({
"id": "directory_01ECAZ4NV9QMV47GW873HDCX74",
"domain": "foo-corp.com",
"name": "Foo Corp",
"organization_id": "org_01EHZNVPK3SFK441A1RGBFSHRT",
"state": "unlinked",
"type": "gsuite directory",
"created_at": "2021-06-25T19:07:33.155Z",
"updated_at": "2021-06-25T19:07:33.155Z"
})
.to_string(),
)
.create_async()
.await;
let directory = workos
.directory_sync()
.get_directory(&DirectoryId::from("directory_01ECAZ4NV9QMV47GW873HDCX74"))
.await
.unwrap();
assert_eq!(
directory.id,
DirectoryId::from("directory_01ECAZ4NV9QMV47GW873HDCX74")
)
}
#[tokio::test]
async fn it_returns_an_error_when_the_get_directory_endpoint_returns_unauthorized() {
let mut server = mockito::Server::new_async().await;
let workos = WorkOs::builder(&ApiKey::from("sk_example_123456789"))
.base_url(&server.url())
.unwrap()
.build();
server
.mock("GET", "/directories/directory_01ECAZ4NV9QMV47GW873HDCX74")
.match_header("Authorization", "Bearer sk_example_123456789")
.with_status(401)
.with_body(
json!({
"message": "Unauthorized"
})
.to_string(),
)
.create_async()
.await;
let result = workos
.directory_sync()
.get_directory(&DirectoryId::from("directory_01ECAZ4NV9QMV47GW873HDCX74"))
.await;
assert_matches!(result, Err(WorkOsError::Unauthorized))
}
}