use async_trait::async_trait;
use thiserror::Error;
use crate::organizations::{Organization, OrganizationId, Organizations};
use crate::{ResponseExt, WorkOsError, WorkOsResult};
#[derive(Debug, Error)]
pub enum GetOrganizationError {}
impl From<GetOrganizationError> for WorkOsError<GetOrganizationError> {
fn from(err: GetOrganizationError) -> Self {
Self::Operation(err)
}
}
#[async_trait]
pub trait GetOrganization {
async fn get_organization(
&self,
id: &OrganizationId,
) -> WorkOsResult<Organization, GetOrganizationError>;
}
#[async_trait]
impl GetOrganization for Organizations<'_> {
async fn get_organization(
&self,
id: &OrganizationId,
) -> WorkOsResult<Organization, GetOrganizationError> {
let url = self
.workos
.base_url()
.join(&format!("/organizations/{id}"))?;
let organization = self
.workos
.client()
.get(url)
.bearer_auth(self.workos.key())
.send()
.await?
.handle_unauthorized_or_generic_error()
.await?
.json::<Organization>()
.await?;
Ok(organization)
}
}
#[cfg(test)]
mod test {
use serde_json::json;
use tokio;
use crate::{ApiKey, WorkOs};
use super::*;
#[tokio::test]
async fn it_calls_the_get_organization_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", "/organizations/org_01EHZNVPK3SFK441A1RGBFSHRT")
.match_header("Authorization", "Bearer sk_example_123456789")
.with_status(200)
.with_body(
json!({
"id": "org_01EHZNVPK3SFK441A1RGBFSHRT",
"object": "organization",
"name": "Foo Corporation",
"allow_profiles_outside_organization": false,
"created_at": "2021-06-25T19:07:33.155Z",
"updated_at": "2021-06-25T19:07:33.155Z",
"domains": [
{
"object": "organization_domain",
"id": "org_domain_01EHZNVPK2QXHMVWCEDQEKY69A",
"domain": "foo-corp.com",
"organization_id": "org_01EHZNVPK3SFK441A1RGBFSHRT",
"state": "verified",
"verification_strategy": "dns",
"verification_token": "m5Oztg3jdK4NJLgs8uIlIprMw",
"created_at": "2021-06-25T19:07:33.155Z",
"updated_at": "2021-06-25T19:07:33.155Z"
},
{
"object": "organization_domain",
"id": "org_domain_01EHZNS0H9W90A90FV79GAB6AB",
"domain": "another-foo-corp-domain.com",
"organization_id": "org_01EHZNVPK3SFK441A1RGBFSHRT",
"state": "verified",
"verification_strategy": "dns",
"verification_token": "m5Oztg3jdK4NJLgs8uIlIprMw",
"created_at": "2021-06-25T19:07:33.155Z",
"updated_at": "2021-06-25T19:07:33.155Z"
}
]
})
.to_string(),
)
.create_async()
.await;
let organization = workos
.organizations()
.get_organization(&OrganizationId::from("org_01EHZNVPK3SFK441A1RGBFSHRT"))
.await
.unwrap();
assert_eq!(
organization.id,
OrganizationId::from("org_01EHZNVPK3SFK441A1RGBFSHRT")
)
}
}