Merge pull request phoenixframework#153 from knewter/feature/add_sock…

…et_assign_api Add assign API for Socket
brightroll · Jul 4, 2014 · f6bdc4f · f6bdc4f
2 parents 9466968 + 9cea3da
commit f6bdc4f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/lib/phoenix/socket.ex b/lib/phoenix/socket.ex
@@ -7,7 +7,7 @@ defmodule Phoenix.Socket do
             topic: nil,
             router: nil,
             channels: [],
-            assigns: []
+            assigns: %{}
 
   def set_current_channel(socket, channel, topic) do
     %Socket{socket | channel: channel, topic: topic}
@@ -28,6 +28,10 @@ defmodule Phoenix.Socket do
   def authenticated?(socket, channel, topic) do
     Enum.member? socket.channels, {channel, topic}
   end
+
+  def assign(socket, key, value) do
+    %Socket{socket | assigns: Map.put(socket.assigns, key, value)}
+  end
 end
 
 
diff --git a/test/phoenix/socket/socket_test.exs b/test/phoenix/socket/socket_test.exs
@@ -6,7 +6,7 @@ defmodule Phoenix.Socket.SocketTest do
     %Socket{pid: self,
             router: nil,
             channels: [],
-            assigns: []}
+            assigns: %{}}
   end
 
   test "#set_current_channel sets the current channel" do
@@ -47,5 +47,12 @@ defmodule Phoenix.Socket.SocketTest do
     socket = Socket.delete_channel(socket, "test", "topic")
     assert socket.channels == []
   end
+
+  test "#assign assigns into the assigns map (yo dog)" do
+    socket = new_socket
+    assert socket.assigns == %{}
+    socket = Socket.assign(socket, :foo, "bar")
+    assert socket.assigns == %{foo: "bar"}
+  end
 end