fix: tests

supabase · Jan 10, 2023 · 41293c5 · 41293c5
1 parent cca6ac0
commit 41293c5
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git 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
@@ -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", %{})