Add test for keep alive messages

maartenvanvliet · Apr 11, 2021 · 1b54138 · 1b54138
1 parent c94cd7a
commit 1b54138
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/test/integration/subscription_transport_ws_test.exs b/test/integration/subscription_transport_ws_test.exs
@@ -230,6 +230,26 @@ defmodule SubscriptionsTransportWS.Integration.SocketTest do
     assert_receive %{"type" => "ka"}
   end
 
+  test "continues to receive keep alive messages" do
+    {:ok, socket} =
+      WebsocketClient.start_link(
+        self(),
+        "ws://127.0.0.1:#{@port}/ws/websocket",
+        @json_module
+      )
+
+    WebsocketClient.send_event(socket, ~s({
+      "type": "connection_init",
+      "payload": {
+        "token": "correct"
+      }
+    }))
+    assert_receive %{"type" => "connection_ack"}
+
+    assert_receive %{"type" => "ka"}
+    assert_receive %{"type" => "ka"}
+  end
+
   test "returns error for invalid graphql document" do
     {:ok, socket} = WebsocketClient.start_link(self(), @path, @json_module)