diff options
Diffstat (limited to 'test/polyjuice/client_test.exs')
| -rw-r--r-- | test/polyjuice/client_test.exs | 40 |
1 files changed, 8 insertions, 32 deletions
diff --git a/test/polyjuice/client_test.exs b/test/polyjuice/client_test.exs index d9f95a8..96dc829 100644 --- a/test/polyjuice/client_test.exs +++ b/test/polyjuice/client_test.exs @@ -235,10 +235,7 @@ defmodule Polyjuice.ClientTest do Polyjuice.Client.log_in_with_password(client, "@alice:example.org", "password") - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client.pid, :get_state) == %{ access_token: "m.id.user_login", user_id: "@alice:example.org", device_id: "foo" @@ -264,10 +261,7 @@ defmodule Polyjuice.ClientTest do assert Polyjuice.Client.Storage.kv_get(storage, "ca.uhoreg.polyjuice", "user_id") == nil assert Polyjuice.Client.Storage.kv_get(storage, "ca.uhoreg.polyjuice", "device_id") == nil - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.fetch!(&1, :access_token) - ) == nil + assert GenServer.call(client.pid, :get_state) |> Map.get(:access_token) == nil Polyjuice.Client.log_in_with_password( client, @@ -277,10 +271,7 @@ defmodule Polyjuice.ClientTest do assert_receive({:polyjuice_client, :logged_in, {"@alice:example.org", "foo", _}}) - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client.pid, :get_state) == %{ access_token: "m.id.thirdparty_login", user_id: "@alice:example.org", device_id: "foo" @@ -294,10 +285,7 @@ defmodule Polyjuice.ClientTest do assert_receive({:polyjuice_client, :logged_in, {"@alice:example.org", "foo", _}}) - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client.pid, :get_state) == %{ access_token: "m.id.phone_login", user_id: "@alice:example.org", device_id: "foo" @@ -313,10 +301,7 @@ defmodule Polyjuice.ClientTest do assert_receive({:polyjuice_client, :logged_in, {"@alice:example.org", "foo", _}}) - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client.pid, :get_state) == %{ access_token: "ca.uhoreg.foo_login", user_id: "@alice:example.org", device_id: "foo" @@ -349,10 +334,7 @@ defmodule Polyjuice.ClientTest do client = Polyjuice.Client.get_client(client_pid) - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client.pid, :get_state) == %{ access_token: "an_access_token", user_id: "@alice:example.org", device_id: "a_device" @@ -382,10 +364,7 @@ defmodule Polyjuice.ClientTest do client2 = Polyjuice.Client.get_client(client2_pid) - assert Agent.get( - Polyjuice.Client.process_name(client2.id, :state), - &Map.take(&1, [:access_token, :user_id, :device_id]) - ) == %{ + assert GenServer.call(client2.pid, :get_state) == %{ access_token: "an_access_token", user_id: "@alice:example.org", device_id: "a_device" @@ -431,10 +410,7 @@ defmodule Polyjuice.ClientTest do Polyjuice.Client.Room.send_event(client, "!a_room:example.org", "m.room.message", %{}) - assert Agent.get( - Polyjuice.Client.process_name(client.id, :state), - &Map.fetch!(&1, :access_token) - ) == nil + assert GenServer.call(client.pid, :get_state) |> Map.get(:state) == nil assert Polyjuice.Client.Storage.kv_get(storage, "ca.uhoreg.polyjuice", "access_token") == nil |