diff --git a/tests/test_organizations.py b/tests/test_organizations.py index ddd08060..87598466 100644 --- a/tests/test_organizations.py +++ b/tests/test_organizations.py @@ -38,6 +38,8 @@ def mock_organization_updated(self): "organization_id": "org_01EHT88Z8J8795GZNQ4ZP1J81T", "verification_strategy": "dns", "verification_token": "token", + "created_at": datetime.datetime.now().isoformat(), + "updated_at": datetime.datetime.now().isoformat(), } ], } @@ -189,15 +191,17 @@ def test_update_organization_with_domain_data( } assert updated_organization.id == "org_01EHT88Z8J8795GZNQ4ZP1J81T" assert updated_organization.name == "Example Organization" - assert updated_organization.domains[0].dict() == { - "domain": "example.io", - "object": "organization_domain", - "id": "org_domain_01EHT88Z8WZEFWYPM6EC9BX2R8", - "state": "verified", - "organization_id": "org_01EHT88Z8J8795GZNQ4ZP1J81T", - "verification_strategy": "dns", - "verification_token": "token", - } + domain = updated_organization.domains[0] + assert domain.domain == "example.io" + assert domain.object == "organization_domain" + assert domain.id == "org_domain_01EHT88Z8WZEFWYPM6EC9BX2R8" + assert domain.state == "verified" + assert domain.organization_id == "org_01EHT88Z8J8795GZNQ4ZP1J81T" + assert domain.verification_strategy == "dns" + assert domain.verification_token == "token" + assert domain.verification_prefix is None + assert isinstance(domain.created_at, str) + assert isinstance(domain.updated_at, str) def test_delete_organization(self, capture_and_mock_http_client_request): request_kwargs = capture_and_mock_http_client_request( diff --git a/tests/utils/fixtures/mock_organization.py b/tests/utils/fixtures/mock_organization.py index 3316105b..d401a398 100644 --- a/tests/utils/fixtures/mock_organization.py +++ b/tests/utils/fixtures/mock_organization.py @@ -20,6 +20,8 @@ def __init__(self, id): id="org_domain_01EHT88Z8WZEFWYPM6EC9BX2R8", organization_id="org_12345", domain="example.io", + created_at=now, + updated_at=now, ) ], metadata={"key": "value"}, diff --git a/workos/types/events/event.py b/workos/types/events/event.py index 8e03be5d..26b1826d 100644 --- a/workos/types/events/event.py +++ b/workos/types/events/event.py @@ -193,6 +193,18 @@ class OrganizationDomainVerifiedEvent(EventModel[OrganizationDomain]): event: Literal["organization_domain.verified"] +class OrganizationDomainCreatedEvent(EventModel[OrganizationDomain]): + event: Literal["organization_domain.created"] + + +class OrganizationDomainUpdatedEvent(EventModel[OrganizationDomain]): + event: Literal["organization_domain.updated"] + + +class OrganizationDomainDeletedEvent(EventModel[OrganizationDomain]): + event: Literal["organization_domain.deleted"] + + class OrganizationMembershipCreatedEvent(EventModel[OrganizationMembership]): event: Literal["organization_membership.created"] @@ -272,6 +284,9 @@ class UserUpdatedEvent(EventModel[User]): OrganizationCreatedEvent, OrganizationDeletedEvent, OrganizationUpdatedEvent, + OrganizationDomainCreatedEvent, + OrganizationDomainDeletedEvent, + OrganizationDomainUpdatedEvent, OrganizationDomainVerificationFailedEvent, OrganizationDomainVerifiedEvent, OrganizationMembershipCreatedEvent, diff --git a/workos/types/events/event_type.py b/workos/types/events/event_type.py index ade1614b..cefb8054 100644 --- a/workos/types/events/event_type.py +++ b/workos/types/events/event_type.py @@ -36,6 +36,9 @@ "organization.updated", "organization_domain.verification_failed", "organization_domain.verified", + "organization_domain.created", + "organization_domain.deleted", + "organization_domain.updated", "organization_membership.created", "organization_membership.deleted", "organization_membership.updated", diff --git a/workos/types/organizations/organization_domain.py b/workos/types/organizations/organization_domain.py index 955b23ff..3a6dafcd 100644 --- a/workos/types/organizations/organization_domain.py +++ b/workos/types/organizations/organization_domain.py @@ -13,3 +13,6 @@ class OrganizationDomain(WorkOSModel): ] = None verification_strategy: Optional[LiteralOrUntyped[Literal["manual", "dns"]]] = None verification_token: Optional[str] = None + verification_prefix: Optional[str] = None + created_at: str + updated_at: str