summaryrefslogtreecommitdiff
path: root/test/web/streamer/state_test.exs
diff options
context:
space:
mode:
authorSteven Fuchs <steven.fuchs@dockyard.com>2019-09-16 07:44:03 +0000
committerkaniini <ariadne@dereferenced.org>2019-09-16 07:44:03 +0000
commitaab264db82054df470075c65ca25c42bbcc5d7a8 (patch)
treecc803e11307818080f6b3cd0c05be133fe33ca62 /test/web/streamer/state_test.exs
parent17142a3720c55e8bb0595f1a1c2f128dfcf47d10 (diff)
downloadpleroma-aab264db82054df470075c65ca25c42bbcc5d7a8.tar.gz
pleroma-aab264db82054df470075c65ca25c42bbcc5d7a8.zip
Streamer refactoring
Diffstat (limited to 'test/web/streamer/state_test.exs')
-rw-r--r--test/web/streamer/state_test.exs54
1 files changed, 54 insertions, 0 deletions
diff --git a/test/web/streamer/state_test.exs b/test/web/streamer/state_test.exs
new file mode 100644
index 000000000..d1aeac541
--- /dev/null
+++ b/test/web/streamer/state_test.exs
@@ -0,0 +1,54 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
+defmodule Pleroma.Web.StateTest do
+ use Pleroma.DataCase
+
+ import Pleroma.Factory
+ alias Pleroma.Web.Streamer
+ alias Pleroma.Web.Streamer.StreamerSocket
+
+ @moduletag needs_streamer: true
+
+ describe "sockets" do
+ setup do
+ user = insert(:user)
+ user2 = insert(:user)
+ {:ok, %{user: user, user2: user2}}
+ end
+
+ test "it can add a socket", %{user: user} do
+ Streamer.add_socket("public", %{transport_pid: 1, assigns: %{user: user}})
+
+ assert(%{"public" => [%StreamerSocket{transport_pid: 1}]} = Streamer.get_sockets())
+ end
+
+ test "it can add multiple sockets per user", %{user: user} do
+ Streamer.add_socket("public", %{transport_pid: 1, assigns: %{user: user}})
+ Streamer.add_socket("public", %{transport_pid: 2, assigns: %{user: user}})
+
+ assert(
+ %{
+ "public" => [
+ %StreamerSocket{transport_pid: 2},
+ %StreamerSocket{transport_pid: 1}
+ ]
+ } = Streamer.get_sockets()
+ )
+ end
+
+ test "it will not add a duplicate socket", %{user: user} do
+ Streamer.add_socket("activity", %{transport_pid: 1, assigns: %{user: user}})
+ Streamer.add_socket("activity", %{transport_pid: 1, assigns: %{user: user}})
+
+ assert(
+ %{
+ "activity" => [
+ %StreamerSocket{transport_pid: 1}
+ ]
+ } = Streamer.get_sockets()
+ )
+ end
+ end
+end