workos 10.2.0 → 10.3.0

data/test/workos/test_shared_model_round_trip.rb

@@ -176,14 +176,15 @@ class SharedModelRoundTripTest < Minitest::Test
 
   def test_waitlist_user_round_trip
     fixture = {
-      "object" => "waitlist_user",
       "id" => "stub",
       "email" => "stub",
       "state" => "stub",
       "approved_at" => nil,
+      "additional_fields" => {},
       "waitlist_id" => nil,
       "created_at" => "stub",
-      "updated_at" => "stub"
+      "updated_at" => "stub",
+      "object" => "waitlist_user"
     }
     model = WorkOS::WaitlistUser.new(fixture.to_json)
     json = model.to_h
@@ -214,6 +215,336 @@ class SharedModelRoundTripTest < Minitest::Test
     fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
   end
 
+  def test_agent_blueprint_created_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.blueprint.created",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentBlueprintCreated.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_created_data_round_trip
+    fixture = {
+      "object" => "agent_blueprint",
+      "id" => "stub",
+      "name" => "stub",
+      "description" => nil,
+      "permissions" => [],
+      "invocable_by" => {},
+      "session_settings" => {},
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentBlueprintCreatedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["name"], json[:name]
+    assert_nil json[:description]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_created_data_invocable_by_round_trip
+    fixture = {
+      "role_slugs" => [],
+      "organization_ids" => []
+    }
+    model = WorkOS::AgentBlueprintCreatedDataInvocableBy.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_created_data_session_setting_round_trip
+    fixture = {
+      "max_age_seconds" => 1.0,
+      "access_token_ttl_seconds" => 1.0,
+      "refresh_token_ttl_seconds" => 1.0
+    }
+    model = WorkOS::AgentBlueprintCreatedDataSessionSetting.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["max_age_seconds"], json[:max_age_seconds]
+    assert_equal fixture["access_token_ttl_seconds"], json[:access_token_ttl_seconds]
+    assert_equal fixture["refresh_token_ttl_seconds"], json[:refresh_token_ttl_seconds]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_deleted_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.blueprint.deleted",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentBlueprintDeleted.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_deleted_data_round_trip
+    fixture = {
+      "object" => "agent_blueprint",
+      "id" => "stub",
+      "name" => "stub",
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentBlueprintDeletedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["name"], json[:name]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_updated_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.blueprint.updated",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentBlueprintUpdated.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_updated_data_round_trip
+    fixture = {
+      "object" => "agent_blueprint",
+      "id" => "stub",
+      "name" => "stub",
+      "description" => nil,
+      "permissions" => [],
+      "invocable_by" => {},
+      "session_settings" => {},
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentBlueprintUpdatedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["name"], json[:name]
+    assert_nil json[:description]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_updated_data_invocable_by_round_trip
+    fixture = {
+      "role_slugs" => [],
+      "organization_ids" => []
+    }
+    model = WorkOS::AgentBlueprintUpdatedDataInvocableBy.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_blueprint_updated_data_session_setting_round_trip
+    fixture = {
+      "max_age_seconds" => 1.0,
+      "access_token_ttl_seconds" => 1.0,
+      "refresh_token_ttl_seconds" => 1.0
+    }
+    model = WorkOS::AgentBlueprintUpdatedDataSessionSetting.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["max_age_seconds"], json[:max_age_seconds]
+    assert_equal fixture["access_token_ttl_seconds"], json[:access_token_ttl_seconds]
+    assert_equal fixture["refresh_token_ttl_seconds"], json[:refresh_token_ttl_seconds]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_created_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.instance.created",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentInstanceCreated.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_created_data_round_trip
+    fixture = {
+      "object" => "agent_instance",
+      "id" => "stub",
+      "agent_blueprint_id" => "stub",
+      "organization_id" => "stub",
+      "organization_membership_id" => nil,
+      "type" => "stub",
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentInstanceCreatedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["agent_blueprint_id"], json[:agent_blueprint_id]
+    assert_equal fixture["organization_id"], json[:organization_id]
+    assert_nil json[:organization_membership_id]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_deleted_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.instance.deleted",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentInstanceDeleted.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_deleted_data_round_trip
+    fixture = {
+      "object" => "agent_instance",
+      "id" => "stub",
+      "agent_blueprint_id" => "stub",
+      "organization_id" => "stub",
+      "organization_membership_id" => nil,
+      "type" => "stub",
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentInstanceDeletedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["agent_blueprint_id"], json[:agent_blueprint_id]
+    assert_equal fixture["organization_id"], json[:organization_id]
+    assert_nil json[:organization_membership_id]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_session_created_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.instance.session.created",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentInstanceSessionCreated.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_session_created_data_round_trip
+    fixture = {
+      "object" => "agent_instance_session",
+      "id" => "stub",
+      "agent_instance_id" => "stub",
+      "organization_id" => "stub",
+      "expires_at" => "stub",
+      "revoked_at" => nil,
+      "created_at" => "stub",
+      "updated_at" => "stub",
+      "permission_slugs" => []
+    }
+    model = WorkOS::AgentInstanceSessionCreatedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["agent_instance_id"], json[:agent_instance_id]
+    assert_equal fixture["organization_id"], json[:organization_id]
+    assert_equal fixture["expires_at"], json[:expires_at]
+    assert_nil json[:revoked_at]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_session_revoked_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "agent.instance.session.revoked",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::AgentInstanceSessionRevoked.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_agent_instance_session_revoked_data_round_trip
+    fixture = {
+      "object" => "agent_instance_session",
+      "id" => "stub",
+      "agent_instance_id" => "stub",
+      "organization_id" => "stub",
+      "expires_at" => "stub",
+      "revoked_at" => nil,
+      "created_at" => "stub",
+      "updated_at" => "stub"
+    }
+    model = WorkOS::AgentInstanceSessionRevokedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["agent_instance_id"], json[:agent_instance_id]
+    assert_equal fixture["organization_id"], json[:organization_id]
+    assert_equal fixture["expires_at"], json[:expires_at]
+    assert_nil json[:revoked_at]
+    assert_equal fixture["created_at"], json[:created_at]
+    assert_equal fixture["updated_at"], json[:updated_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
   def test_agent_registration_claim_attempt_created_round_trip
     fixture = {
       "object" => "event",
@@ -804,6 +1135,122 @@ class SharedModelRoundTripTest < Minitest::Test
     fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
   end
 
+  def test_resource_export_completed_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "resource_export.completed",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::ResourceExportCompleted.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_completed_data_round_trip
+    fixture = {
+      "id" => "stub",
+      "resource_type" => "stub"
+    }
+    model = WorkOS::ResourceExportCompletedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_created_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "resource_export.created",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::ResourceExportCreated.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_created_data_round_trip
+    fixture = {
+      "id" => "stub",
+      "resource_type" => "stub"
+    }
+    model = WorkOS::ResourceExportCreatedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_downloaded_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "resource_export.downloaded",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::ResourceExportDownloaded.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_downloaded_data_round_trip
+    fixture = {
+      "id" => "stub",
+      "resource_type" => "stub"
+    }
+    model = WorkOS::ResourceExportDownloadedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_failed_round_trip
+    fixture = {
+      "object" => "event",
+      "id" => "stub",
+      "event" => "resource_export.failed",
+      "data" => {},
+      "created_at" => "stub",
+      "context" => {}
+    }
+    model = WorkOS::ResourceExportFailed.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    assert_equal fixture["created_at"], json[:created_at]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
+  def test_resource_export_failed_data_round_trip
+    fixture = {
+      "id" => "stub",
+      "resource_type" => "stub"
+    }
+    model = WorkOS::ResourceExportFailedData.new(fixture.to_json)
+    json = model.to_h
+    assert_kind_of Hash, json
+    assert_equal fixture["id"], json[:id]
+    fixture.each_key { |k| assert json.key?(k.to_sym) || json.key?(k), "Expected to_h to include key #{k}" }
+  end
+
   def test_waitlist_user_approved_round_trip
     fixture = {
       "object" => "event",

data/test/workos/test_sso.rb

@@ -18,6 +18,85 @@ class SSOTest < Minitest::Test
     assert_kind_of WorkOS::Types::ListStruct, result
   end
 
+  def test_create_connection_returns_expected_result
+    stub_request(:post, %r{\Ahttps://api\.workos\.com/connections(\?|\z)})
+      .with(body: hash_including("organization_id" => "stub", "saml_options" => {}))
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.create_connection(organization_id: "stub", protocol_options: WorkOS::SSO::CreateProtocolOptionsSAML.new(saml_options: {}))
+    refute_nil result
+  end
+
+  def test_create_connection_with_protocol_options_oidc_returns_expected_result
+    stub_request(:post, %r{\Ahttps://api\.workos\.com/connections(\?|\z)})
+      .with(body: hash_including("organization_id" => "stub", "oidc_options" => {}))
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.create_connection(organization_id: "stub", protocol_options: WorkOS::SSO::CreateProtocolOptionsOIDC.new(oidc_options: {}))
+    refute_nil result
+  end
+
+  def test_list_connection_saml_idp_signing_certs_returns_expected_result
+    stub_request(:get, %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.list_connection_saml_idp_signing_certs(connection_id: "stub")
+    refute_nil result
+  end
+
+  def test_create_connection_saml_idp_signing_cert_returns_expected_result
+    stub_request(:post, %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.create_connection_saml_idp_signing_cert(connection_id: "stub", value: "stub")
+    refute_nil result
+  end
+
+  def test_delete_connection_saml_idp_signing_cert_returns_expected_result
+    stub_request(:delete, %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs/stub(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.delete_connection_saml_idp_signing_cert(connection_id: "stub", certificate_id: "stub")
+    assert_nil result
+  end
+
+  def test_list_connection_saml_sp_encryption_certs_returns_expected_result
+    stub_request(:get, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.list_connection_saml_sp_encryption_certs(connection_id: "stub")
+    refute_nil result
+  end
+
+  def test_create_connection_saml_sp_encryption_cert_returns_expected_result
+    stub_request(:post, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.create_connection_saml_sp_encryption_cert(connection_id: "stub")
+    refute_nil result
+  end
+
+  def test_delete_connection_saml_sp_encryption_cert_returns_expected_result
+    stub_request(:delete, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs/stub(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.delete_connection_saml_sp_encryption_cert(connection_id: "stub", certificate_id: "stub")
+    assert_nil result
+  end
+
+  def test_list_connection_saml_sp_signing_cert_returns_expected_result
+    stub_request(:get, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.list_connection_saml_sp_signing_cert(connection_id: "stub")
+    refute_nil result
+  end
+
+  def test_create_connection_saml_sp_signing_cert_returns_expected_result
+    stub_request(:post, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.create_connection_saml_sp_signing_cert(connection_id: "stub")
+    refute_nil result
+  end
+
+  def test_delete_connection_saml_sp_signing_cert_returns_expected_result
+    stub_request(:delete, %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert/stub(\?|\z)})
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.delete_connection_saml_sp_signing_cert(connection_id: "stub", certificate_id: "stub")
+    assert_nil result
+  end
+
   def test_get_connection_returns_expected_result
     stub_request(:get, %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)})
       .to_return(body: "{}", status: 200)
@@ -25,6 +104,22 @@ class SSOTest < Minitest::Test
     refute_nil result
   end
 
+  def test_update_connection_returns_expected_result
+    stub_request(:patch, %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)})
+      .with(body: hash_including("saml_options" => {}))
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.update_connection(id: "stub", protocol_options: WorkOS::SSO::PatchProtocolOptionsSAML.new(saml_options: {}))
+    refute_nil result
+  end
+
+  def test_update_connection_with_protocol_options_oidc_returns_expected_result
+    stub_request(:patch, %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)})
+      .with(body: hash_including("oidc_options" => {}))
+      .to_return(body: "{}", status: 200)
+    result = @client.sso.update_connection(id: "stub", protocol_options: WorkOS::SSO::PatchProtocolOptionsOIDC.new(oidc_options: {}))
+    refute_nil result
+  end
+
   def test_delete_connection_returns_expected_result
     stub_request(:delete, %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)})
       .to_return(body: "{}", status: 200)
@@ -49,18 +144,29 @@ class SSOTest < Minitest::Test
   def test_get_profile_and_token_returns_expected_result
     stub_request(:post, %r{\Ahttps://api\.workos\.com/sso/token(\?|\z)})
       .to_return(body: "{}", status: 200)
-    result = @client.sso.get_profile_and_token(code: "stub")
+    result = @client.sso.get_profile_and_token
     refute_nil result
   end
 
   # Parameterized authentication error tests (one per endpoint).
   [
     {name: :list_connections, verb: :get, url: %r{\Ahttps://api\.workos\.com/connections(\?|\z)}},
+    {name: :create_connection, verb: :post, url: %r{\Ahttps://api\.workos\.com/connections(\?|\z)}, args: {organization_id: "stub", protocol_options: WorkOS::SSO::CreateProtocolOptionsSAML.new(saml_options: {})}},
+    {name: :list_connection_saml_idp_signing_certs, verb: :get, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs(\?|\z)}, args: {connection_id: "stub"}},
+    {name: :create_connection_saml_idp_signing_cert, verb: :post, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs(\?|\z)}, args: {connection_id: "stub", value: "stub"}},
+    {name: :delete_connection_saml_idp_signing_cert, verb: :delete, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_idp_signing_certs/stub(\?|\z)}, args: {connection_id: "stub", certificate_id: "stub"}},
+    {name: :list_connection_saml_sp_encryption_certs, verb: :get, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs(\?|\z)}, args: {connection_id: "stub"}},
+    {name: :create_connection_saml_sp_encryption_cert, verb: :post, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs(\?|\z)}, args: {connection_id: "stub"}},
+    {name: :delete_connection_saml_sp_encryption_cert, verb: :delete, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_encryption_certs/stub(\?|\z)}, args: {connection_id: "stub", certificate_id: "stub"}},
+    {name: :list_connection_saml_sp_signing_cert, verb: :get, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert(\?|\z)}, args: {connection_id: "stub"}},
+    {name: :create_connection_saml_sp_signing_cert, verb: :post, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert(\?|\z)}, args: {connection_id: "stub"}},
+    {name: :delete_connection_saml_sp_signing_cert, verb: :delete, url: %r{\Ahttps://api\.workos\.com/connections/stub/saml_sp_signing_cert/stub(\?|\z)}, args: {connection_id: "stub", certificate_id: "stub"}},
     {name: :get_connection, verb: :get, url: %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)}, args: {id: "stub"}},
+    {name: :update_connection, verb: :patch, url: %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)}, args: {id: "stub", protocol_options: WorkOS::SSO::PatchProtocolOptionsSAML.new(saml_options: {})}},
     {name: :delete_connection, verb: :delete, url: %r{\Ahttps://api\.workos\.com/connections/stub(\?|\z)}, args: {id: "stub"}},
     {name: :authorize_logout, verb: :post, url: %r{\Ahttps://api\.workos\.com/sso/logout/authorize(\?|\z)}, args: {profile_id: "stub"}},
     {name: :get_profile, verb: :get, url: %r{\Ahttps://api\.workos\.com/sso/profile(\?|\z)}},
-    {name: :get_profile_and_token, verb: :post, url: %r{\Ahttps://api\.workos\.com/sso/token(\?|\z)}, args: {code: "stub"}}
+    {name: :get_profile_and_token, verb: :post, url: %r{\Ahttps://api\.workos\.com/sso/token(\?|\z)}}
   ].each do |spec|
     define_method("test_#{spec[:name]}_raises_authentication_error_on_401") do
       stub_request(spec[:verb], spec[:url])