From 41293c52fd3a1afbd10911c599511406ec718628 Mon Sep 17 00:00:00 2001 From: Chase Granberry Date: Tue, 10 Jan 2023 11:38:36 -0700 Subject: [PATCH] fix: tests --- .../extensions/cdc_rls/cdc_rls_test.exs | 5 ++++- .../channels/realtime_channel_test.exs | 17 ++++++++++++++--- 2 files changed, 18 insertions(+), 4 deletions(-) diff --git a/test/realtime/extensions/cdc_rls/cdc_rls_test.exs b/test/realtime/extensions/cdc_rls/cdc_rls_test.exs index cda6c1dd3..5519090ae 100644 --- a/test/realtime/extensions/cdc_rls/cdc_rls_test.exs +++ b/test/realtime/extensions/cdc_rls/cdc_rls_test.exs @@ -29,7 +29,10 @@ defmodule Realtime.Extensions.CdcRlsTest do claims: %{}, limits: %{ max_concurrent_users: 1, - max_events_per_second: 100 + max_events_per_second: 100, + max_joins_per_second: 500, + max_channels_per_client: 100, + max_bytes_per_second: 100_000 }, is_new_api: false, log_level: :info diff --git a/test/realtime_web/channels/realtime_channel_test.exs b/test/realtime_web/channels/realtime_channel_test.exs index 4d3a6dccf..20a3e982e 100644 --- a/test/realtime_web/channels/realtime_channel_test.exs +++ b/test/realtime_web/channels/realtime_channel_test.exs @@ -9,6 +9,14 @@ defmodule RealtimeWeb.RealtimeChannelTest do @tenant "dev_tenant" + @default_limits %{ + max_concurrent_users: 200, + max_events_per_second: 100, + max_joins_per_second: 500, + max_channels_per_client: 100, + max_bytes_per_second: 100_000 + } + setup do {:ok, _pid} = start_supervised(CurrentTime.Mock) :ok @@ -30,7 +38,7 @@ defmodule RealtimeWeb.RealtimeChannelTest do x_headers: [{"x-api-key", "token123"}] }) - socket = Socket.assign(socket, %{limits: %{max_concurrent_users: 1}}) + socket = Socket.assign(socket, %{limits: %{@default_limits | max_concurrent_users: 1}}) assert {:ok, _, %Socket{}} = subscribe_and_join(socket, "realtime:test", %{}) end end @@ -50,8 +58,11 @@ defmodule RealtimeWeb.RealtimeChannelTest do x_headers: [{"x-api-key", "token123"}] }) - socket_at_capacity = Socket.assign(socket, %{limits: %{max_concurrent_users: 0}}) - socket_over_capacity = Socket.assign(socket, %{limits: %{max_concurrent_users: -1}}) + socket_at_capacity = + Socket.assign(socket, %{limits: %{@default_limits | max_concurrent_users: 0}}) + + socket_over_capacity = + Socket.assign(socket, %{limits: %{@default_limits | max_concurrent_users: -1}}) assert {:error, %{reason: "{:error, :too_many_connections}"}} = subscribe_and_join(socket_at_capacity, "realtime:test", %{})