Merge branch 'fix/trailing-format-plug-oauth' into 'develop'

[akkoma] / test / plugs / user_enabled_plug_test.exs

diff --git a/test/plugs/user_enabled_plug_test.exs b/test/plugs/user_enabled_plug_test.exs
index 225368efcb42b89d46bfcda9cb3fb5f9029e9021..996a7d77b11223078eefe67e564feaea4d4e399d 100644 (file)
--- a/test/plugs/user_enabled_plug_test.exs
+++ b/test/plugs/user_enabled_plug_test.exs
@@ -1,8 +1,12 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
 defmodule Pleroma.Plugs.UserEnabledPlugTest do
   use Pleroma.Web.ConnCase, async: true
 
-  alias Pleroma.User
   alias Pleroma.Plugs.UserEnabledPlug
+  import Pleroma.Factory
 
   test "doesn't do anything if the user isn't set", %{conn: conn} do
     ret_conn =
@@ -13,18 +17,22 @@ defmodule Pleroma.Plugs.UserEnabledPlugTest do
   end
 
   test "with a user that is deactivated, it removes that user", %{conn: conn} do
+    user = insert(:user, deactivated: true)
+
     conn =
       conn
-      |> assign(:user, %User{info: %{"deactivated" => true}})
+      |> assign(:user, user)
       |> UserEnabledPlug.call(%{})
 
     assert conn.assigns.user == nil
   end
 
   test "with a user that is not deactivated, it does nothing", %{conn: conn} do
+    user = insert(:user)
+
     conn =
       conn
-      |> assign(:user, %User{})
+      |> assign(:user, user)
 
     ret_conn =
       conn