No OneTemporary
Actions

Authored By

Unknown

Size

34 KB

Referenced Files

None

Subscribers

None

View Options

	diff --git a/lib/mix/pleroma.ex b/lib/mix/pleroma.ex
	index cd3f44074..7575f0ef8 100644
	--- a/lib/mix/pleroma.ex
	+++ b/lib/mix/pleroma.ex
	@@ -1,132 +1,126 @@
	# Pleroma: A lightweight social networking server
	# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
	# SPDX-License-Identifier: AGPL-3.0-only

	defmodule Mix.Pleroma do
	@apps [
	:restarter,
	:ecto,
	:ecto_sql,
	:postgrex,
	:db_connection,
	:cachex,
	:flake_id,
	:swoosh,
	:timex
	]
	@cachex_children ["object", "user", "scrubber", "web_resp"]
	@doc "Common functions to be reused in mix tasks"
	def start_pleroma do
	Pleroma.Config.Holder.save_default()
	Pleroma.Config.Oban.warn()
	Pleroma.Application.limiters_setup()
	Application.put_env(:phoenix, :serve_endpoints, false, persistent: true)

	unless System.get_env("DEBUG") do
	Logger.remove_backend(:console)
	end

	adapter = Application.get_env(:tesla, :adapter)

	apps =
	if adapter == Tesla.Adapter.Gun do
	[:gun \| @apps]
	else
	[:hackney \| @apps]
	end

	Enum.each(apps, &Application.ensure_all_started/1)

	children =
	[
	Pleroma.Repo,
	{Pleroma.Config.TransferTask, false},
	Pleroma.Web.Endpoint,
	{Oban, Pleroma.Config.get(Oban)}
	] ++
	http_children(adapter)

	cachex_children = Enum.map(@cachex_children, &Pleroma.Application.build_cachex(&1, []))

	Supervisor.start_link(children ++ cachex_children,
	strategy: :one_for_one,
	name: Pleroma.Supervisor
	)

	if Pleroma.Config.get(:env) not in [:test, :benchmark] do
	pleroma_rebooted?()
	end
	end

	defp pleroma_rebooted? do
	if Restarter.Pleroma.rebooted?() do
	:ok
	else
	Process.sleep(10)
	pleroma_rebooted?()
	end
	end

	def load_pleroma do
	Application.load(:pleroma)
	end

	def get_option(options, opt, prompt, defval \\ nil, defname \\ nil) do
	Keyword.get(options, opt) \|\| shell_prompt(prompt, defval, defname)
	end

	def shell_prompt(prompt, defval \\ nil, defname \\ nil) do
	prompt_message = "#{prompt} [#{defname \|\| defval}] "

	input =
	if mix_shell?(),
	do: Mix.shell().prompt(prompt_message),
	else: :io.get_line(prompt_message)

	case input do
	"\n" ->
	case defval do
	nil ->
	shell_prompt(prompt, defval, defname)

	defval ->
	defval
	end

	input ->
	String.trim(input)
	end
	end

	- def shell_yes?(message) do
	- if mix_shell?(),
	- do: Mix.shell().yes?("Continue?"),
	- else: shell_prompt(message, "Continue?") in ~w(Yn Y y)
	- end
	-
	def shell_info(message) do
	if mix_shell?(),
	do: Mix.shell().info(message),
	else: IO.puts(message)
	end

	def shell_error(message) do
	if mix_shell?(),
	do: Mix.shell().error(message),
	else: IO.puts(:stderr, message)
	end

	@doc "Performs a safe check whether `Mix.shell/0` is available (does not raise if Mix is not loaded)"
	def mix_shell?, do: :erlang.function_exported(Mix, :shell, 0)

	def escape_sh_path(path) do
	~S(') <> String.replace(path, ~S('), ~S(\')) <> ~S(')
	end

	defp http_children(Tesla.Adapter.Gun) do
	Pleroma.Gun.ConnectionPool.children() ++
	[{Task, &Pleroma.HTTP.AdapterHelper.Gun.limiter_setup/0}]
	end

	defp http_children(_), do: []
	end
	diff --git a/lib/mix/tasks/pleroma/user.ex b/lib/mix/tasks/pleroma/user.ex
	index a8d251411..ca9c8579f 100644
	--- a/lib/mix/tasks/pleroma/user.ex
	+++ b/lib/mix/tasks/pleroma/user.ex
	@@ -1,463 +1,463 @@
	# Pleroma: A lightweight social networking server
	# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
	# SPDX-License-Identifier: AGPL-3.0-only

	defmodule Mix.Tasks.Pleroma.User do
	use Mix.Task
	import Mix.Pleroma
	alias Ecto.Changeset
	alias Pleroma.User
	alias Pleroma.UserInviteToken
	alias Pleroma.Web.ActivityPub.Builder
	alias Pleroma.Web.ActivityPub.Pipeline

	@shortdoc "Manages Pleroma users"
	@moduledoc File.read!("docs/administration/CLI_tasks/user.md")

	def run(["new", nickname, email \| rest]) do
	{options, [], []} =
	OptionParser.parse(
	rest,
	strict: [
	name: :string,
	bio: :string,
	password: :string,
	moderator: :boolean,
	admin: :boolean,
	assume_yes: :boolean
	],
	aliases: [
	y: :assume_yes
	]
	)

	name = Keyword.get(options, :name, nickname)
	bio = Keyword.get(options, :bio, "")

	{password, generated_password?} =
	case Keyword.get(options, :password) do
	nil ->
	{:crypto.strong_rand_bytes(16) \|> Base.encode64(), true}

	password ->
	{password, false}
	end

	moderator? = Keyword.get(options, :moderator, false)
	admin? = Keyword.get(options, :admin, false)
	assume_yes? = Keyword.get(options, :assume_yes, false)

	shell_info("""
	A user will be created with the following information:
	- nickname: #{nickname}
	- email: #{email}
	- password: #{
	if(generated_password?, do: "[generated; a reset link will be created]", else: password)
	}
	- name: #{name}
	- bio: #{bio}
	- moderator: #{if(moderator?, do: "true", else: "false")}
	- admin: #{if(admin?, do: "true", else: "false")}
	""")

	- proceed? = assume_yes? or shell_yes?("Continue?")
	+ proceed? = assume_yes? or shell_prompt("Continue?", "n") in ~w(Yn Y y)

	if proceed? do
	start_pleroma()

	params = %{
	nickname: nickname,
	email: email,
	password: password,
	password_confirmation: password,
	name: name,
	bio: bio
	}

	changeset = User.register_changeset(%User{}, params, need_confirmation: false)
	{:ok, _user} = User.register(changeset)

	shell_info("User #{nickname} created")

	if moderator? do
	run(["set", nickname, "--moderator"])
	end

	if admin? do
	run(["set", nickname, "--admin"])
	end

	if generated_password? do
	run(["reset_password", nickname])
	end
	else
	shell_info("User will not be created.")
	end
	end

	def run(["rm", nickname]) do
	start_pleroma()

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname),
	{:ok, delete_data, _} <- Builder.delete(user, user.ap_id),
	{:ok, _delete, _} <- Pipeline.common_pipeline(delete_data, local: true) do
	shell_info("User #{nickname} deleted.")
	else
	_ -> shell_error("No local user #{nickname}")
	end
	end

	def run(["toggle_activated", nickname]) do
	start_pleroma()

	with %User{} = user <- User.get_cached_by_nickname(nickname) do
	{:ok, user} = User.deactivate(user, !user.deactivated)

	shell_info(
	"Activation status of #{nickname}: #{if(user.deactivated, do: "de", else: "")}activated"
	)
	else
	_ ->
	shell_error("No user #{nickname}")
	end
	end

	def run(["reset_password", nickname]) do
	start_pleroma()

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname),
	{:ok, token} <- Pleroma.PasswordResetToken.create_token(user) do
	shell_info("Generated password reset token for #{user.nickname}")

	IO.puts(
	"URL: #{
	Pleroma.Web.Router.Helpers.reset_password_url(
	Pleroma.Web.Endpoint,
	:reset,
	token.token
	)
	}"
	)
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["reset_mfa", nickname]) do
	start_pleroma()

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname),
	{:ok, _token} <- Pleroma.MFA.disable(user) do
	shell_info("Multi-Factor Authentication disabled for #{user.nickname}")
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["deactivate", nickname]) do
	start_pleroma()

	with %User{} = user <- User.get_cached_by_nickname(nickname) do
	shell_info("Deactivating #{user.nickname}")
	User.deactivate(user)
	:timer.sleep(500)

	user = User.get_cached_by_id(user.id)

	if Enum.empty?(Enum.filter(User.get_friends(user), & &1.local)) do
	shell_info("Successfully unsubscribed all local followers from #{user.nickname}")
	end
	else
	_ ->
	shell_error("No user #{nickname}")
	end
	end

	def run(["deactivate_all_from_instance", instance]) do
	start_pleroma()

	Pleroma.User.Query.build(%{nickname: "@#{instance}"})
	\|> Pleroma.Repo.chunk_stream(500, :batches)
	\|> Stream.each(fn users ->
	users
	\|> Enum.each(fn user ->
	run(["deactivate", user.nickname])
	end)
	end)
	\|> Stream.run()
	end

	def run(["set", nickname \| rest]) do
	start_pleroma()

	{options, [], []} =
	OptionParser.parse(
	rest,
	strict: [
	admin: :boolean,
	confirmed: :boolean,
	locked: :boolean,
	moderator: :boolean
	]
	)

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname) do
	user =
	case Keyword.get(options, :admin) do
	nil -> user
	value -> set_admin(user, value)
	end

	user =
	case Keyword.get(options, :confirmed) do
	nil -> user
	value -> set_confirmed(user, value)
	end

	user =
	case Keyword.get(options, :locked) do
	nil -> user
	value -> set_locked(user, value)
	end

	_user =
	case Keyword.get(options, :moderator) do
	nil -> user
	value -> set_moderator(user, value)
	end
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["tag", nickname \| tags]) do
	start_pleroma()

	with %User{} = user <- User.get_cached_by_nickname(nickname) do
	user = user \|> User.tag(tags)

	shell_info("Tags of #{user.nickname}: #{inspect(user.tags)}")
	else
	_ ->
	shell_error("Could not change user tags for #{nickname}")
	end
	end

	def run(["untag", nickname \| tags]) do
	start_pleroma()

	with %User{} = user <- User.get_cached_by_nickname(nickname) do
	user = user \|> User.untag(tags)

	shell_info("Tags of #{user.nickname}: #{inspect(user.tags)}")
	else
	_ ->
	shell_error("Could not change user tags for #{nickname}")
	end
	end

	def run(["invite" \| rest]) do
	{options, [], []} =
	OptionParser.parse(rest,
	strict: [
	expires_at: :string,
	max_use: :integer
	]
	)

	options =
	options
	\|> Keyword.update(:expires_at, {:ok, nil}, fn
	nil -> {:ok, nil}
	val -> Date.from_iso8601(val)
	end)
	\|> Enum.into(%{})

	start_pleroma()

	with {:ok, val} <- options[:expires_at],
	options = Map.put(options, :expires_at, val),
	{:ok, invite} <- UserInviteToken.create_invite(options) do
	shell_info("Generated user invite token " <> String.replace(invite.invite_type, "_", " "))

	url =
	Pleroma.Web.Router.Helpers.redirect_url(
	Pleroma.Web.Endpoint,
	:registration_page,
	invite.token
	)

	IO.puts(url)
	else
	error ->
	shell_error("Could not create invite token: #{inspect(error)}")
	end
	end

	def run(["invites"]) do
	start_pleroma()

	shell_info("Invites list:")

	UserInviteToken.list_invites()
	\|> Enum.each(fn invite ->
	expire_info =
	with expires_at when not is_nil(expires_at) <- invite.expires_at do
	" \| Expires at: #{Date.to_string(expires_at)}"
	end

	using_info =
	with max_use when not is_nil(max_use) <- invite.max_use do
	" \| Max use: #{max_use} Left use: #{max_use - invite.uses}"
	end

	shell_info(
	"ID: #{invite.id} \| Token: #{invite.token} \| Token type: #{invite.invite_type} \| Used: #{
	invite.used
	}#{expire_info}#{using_info}"
	)
	end)
	end

	def run(["revoke_invite", token]) do
	start_pleroma()

	with {:ok, invite} <- UserInviteToken.find_by_token(token),
	{:ok, _} <- UserInviteToken.update_invite(invite, %{used: true}) do
	shell_info("Invite for token #{token} was revoked.")
	else
	_ -> shell_error("No invite found with token #{token}")
	end
	end

	def run(["delete_activities", nickname]) do
	start_pleroma()

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname) do
	User.delete_user_activities(user)
	shell_info("User #{nickname} statuses deleted.")
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["toggle_confirmed", nickname]) do
	start_pleroma()

	with %User{} = user <- User.get_cached_by_nickname(nickname) do
	{:ok, user} = User.toggle_confirmation(user)

	message = if user.confirmation_pending, do: "needs", else: "doesn't need"

	shell_info("#{nickname} #{message} confirmation.")
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["confirm_all"]) do
	start_pleroma()

	Pleroma.User.Query.build(%{
	local: true,
	deactivated: false,
	is_moderator: false,
	is_admin: false,
	invisible: false
	})
	\|> Pleroma.Repo.chunk_stream(500, :batches)
	\|> Stream.each(fn users ->
	users
	\|> Enum.each(fn user -> User.need_confirmation(user, false) end)
	end)
	\|> Stream.run()
	end

	def run(["unconfirm_all"]) do
	start_pleroma()

	Pleroma.User.Query.build(%{
	local: true,
	deactivated: false,
	is_moderator: false,
	is_admin: false,
	invisible: false
	})
	\|> Pleroma.Repo.chunk_stream(500, :batches)
	\|> Stream.each(fn users ->
	users
	\|> Enum.each(fn user -> User.need_confirmation(user, true) end)
	end)
	\|> Stream.run()
	end

	def run(["sign_out", nickname]) do
	start_pleroma()

	with %User{local: true} = user <- User.get_cached_by_nickname(nickname) do
	User.global_sign_out(user)

	shell_info("#{nickname} signed out from all apps.")
	else
	_ ->
	shell_error("No local user #{nickname}")
	end
	end

	def run(["list"]) do
	start_pleroma()

	Pleroma.User.Query.build(%{local: true})
	\|> Pleroma.Repo.chunk_stream(500, :batches)
	\|> Stream.each(fn users ->
	users
	\|> Enum.each(fn user ->
	shell_info(
	"#{user.nickname} moderator: #{user.is_moderator}, admin: #{user.is_admin}, locked: #{
	user.is_locked
	}, deactivated: #{user.deactivated}"
	)
	end)
	end)
	\|> Stream.run()
	end

	defp set_moderator(user, value) do
	{:ok, user} =
	user
	\|> Changeset.change(%{is_moderator: value})
	\|> User.update_and_set_cache()

	shell_info("Moderator status of #{user.nickname}: #{user.is_moderator}")
	user
	end

	defp set_admin(user, value) do
	{:ok, user} = User.admin_api_update(user, %{is_admin: value})

	shell_info("Admin status of #{user.nickname}: #{user.is_admin}")
	user
	end

	defp set_locked(user, value) do
	{:ok, user} =
	user
	\|> Changeset.change(%{is_locked: value})
	\|> User.update_and_set_cache()

	shell_info("Locked status of #{user.nickname}: #{user.is_locked}")
	user
	end

	defp set_confirmed(user, value) do
	{:ok, user} = User.need_confirmation(user, !value)

	shell_info("Confirmation pending status of #{user.nickname}: #{user.confirmation_pending}")
	user
	end
	end
	diff --git a/test/mix/tasks/pleroma/user_test.exs b/test/mix/tasks/pleroma/user_test.exs
	index ce819f815..ae0c50443 100644
	--- a/test/mix/tasks/pleroma/user_test.exs
	+++ b/test/mix/tasks/pleroma/user_test.exs
	@@ -1,619 +1,619 @@
	# Pleroma: A lightweight social networking server
	# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
	# SPDX-License-Identifier: AGPL-3.0-only

	defmodule Mix.Tasks.Pleroma.UserTest do
	alias Pleroma.Activity
	alias Pleroma.MFA
	alias Pleroma.Object
	alias Pleroma.Repo
	alias Pleroma.Tests.ObanHelpers
	alias Pleroma.User
	alias Pleroma.Web.CommonAPI
	alias Pleroma.Web.OAuth.Authorization
	alias Pleroma.Web.OAuth.Token

	use Pleroma.DataCase
	use Oban.Testing, repo: Pleroma.Repo

	import ExUnit.CaptureIO
	import Mock
	import Pleroma.Factory

	setup_all do
	Mix.shell(Mix.Shell.Process)

	on_exit(fn ->
	Mix.shell(Mix.Shell.IO)
	end)

	:ok
	end

	describe "running new" do
	test "user is created" do
	# just get random data
	unsaved = build(:user)

	# prepare to answer yes
	- send(self(), {:mix_shell_input, :yes?, true})
	+ send(self(), {:mix_shell_input, :prompt, "Y"})

	Mix.Tasks.Pleroma.User.run([
	"new",
	unsaved.nickname,
	unsaved.email,
	"--name",
	unsaved.name,
	"--bio",
	unsaved.bio,
	"--password",
	"test",
	"--moderator",
	"--admin"
	])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "user will be created"

	- assert_received {:mix_shell, :yes?, [message]}
	+ assert_received {:mix_shell, :prompt, [message]}
	assert message =~ "Continue"

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "created"

	user = User.get_cached_by_nickname(unsaved.nickname)
	assert user.name == unsaved.name
	assert user.email == unsaved.email
	assert user.bio == unsaved.bio
	assert user.is_moderator
	assert user.is_admin
	end

	test "user is not created" do
	unsaved = build(:user)

	# prepare to answer no
	- send(self(), {:mix_shell_input, :yes?, false})
	+ send(self(), {:mix_shell_input, :prompt, "N"})

	Mix.Tasks.Pleroma.User.run(["new", unsaved.nickname, unsaved.email])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "user will be created"

	- assert_received {:mix_shell, :yes?, [message]}
	+ assert_received {:mix_shell, :prompt, [message]}
	assert message =~ "Continue"

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "will not be created"

	refute User.get_cached_by_nickname(unsaved.nickname)
	end
	end

	describe "running rm" do
	test "user is deleted" do
	clear_config([:instance, :federating], true)
	user = insert(:user)

	with_mock Pleroma.Web.Federator,
	publish: fn _ -> nil end do
	Mix.Tasks.Pleroma.User.run(["rm", user.nickname])
	ObanHelpers.perform_all()

	assert_received {:mix_shell, :info, [message]}
	assert message =~ " deleted"
	assert %{deactivated: true} = User.get_by_nickname(user.nickname)

	assert called(Pleroma.Web.Federator.publish(:_))
	end
	end

	test "a remote user's create activity is deleted when the object has been pruned" do
	user = insert(:user)
	user2 = insert(:user)

	{:ok, post} = CommonAPI.post(user, %{status: "uguu"})
	{:ok, post2} = CommonAPI.post(user2, %{status: "test"})
	obj = Object.normalize(post2)

	{:ok, like_object, meta} = Pleroma.Web.ActivityPub.Builder.like(user, obj)

	{:ok, like_activity, _meta} =
	Pleroma.Web.ActivityPub.Pipeline.common_pipeline(
	like_object,
	Keyword.put(meta, :local, true)
	)

	like_activity.data["object"]
	\|> Pleroma.Object.get_by_ap_id()
	\|> Repo.delete()

	clear_config([:instance, :federating], true)

	object = Object.normalize(post)
	Object.prune(object)

	with_mock Pleroma.Web.Federator,
	publish: fn _ -> nil end do
	Mix.Tasks.Pleroma.User.run(["rm", user.nickname])
	ObanHelpers.perform_all()

	assert_received {:mix_shell, :info, [message]}
	assert message =~ " deleted"
	assert %{deactivated: true} = User.get_by_nickname(user.nickname)

	assert called(Pleroma.Web.Federator.publish(:_))
	refute Pleroma.Repo.get(Pleroma.Activity, like_activity.id)
	end

	refute Activity.get_by_id(post.id)
	end

	test "no user to delete" do
	Mix.Tasks.Pleroma.User.run(["rm", "nonexistent"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "running toggle_activated" do
	test "user is deactivated" do
	user = insert(:user)

	Mix.Tasks.Pleroma.User.run(["toggle_activated", user.nickname])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ " deactivated"

	user = User.get_cached_by_nickname(user.nickname)
	assert user.deactivated
	end

	test "user is activated" do
	user = insert(:user, deactivated: true)

	Mix.Tasks.Pleroma.User.run(["toggle_activated", user.nickname])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ " activated"

	user = User.get_cached_by_nickname(user.nickname)
	refute user.deactivated
	end

	test "no user to toggle" do
	Mix.Tasks.Pleroma.User.run(["toggle_activated", "nonexistent"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No user"
	end
	end

	describe "running deactivate" do
	test "user is unsubscribed" do
	followed = insert(:user)
	remote_followed = insert(:user, local: false)
	user = insert(:user)

	User.follow(user, followed, :follow_accept)
	User.follow(user, remote_followed, :follow_accept)

	Mix.Tasks.Pleroma.User.run(["deactivate", user.nickname])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Deactivating"

	# Note that the task has delay :timer.sleep(500)
	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Successfully unsubscribed"

	user = User.get_cached_by_nickname(user.nickname)
	assert Enum.empty?(Enum.filter(User.get_friends(user), & &1.local))
	assert user.deactivated
	end

	test "no user to deactivate" do
	Mix.Tasks.Pleroma.User.run(["deactivate", "nonexistent"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No user"
	end
	end

	describe "running set" do
	test "All statuses set" do
	user = insert(:user)

	Mix.Tasks.Pleroma.User.run([
	"set",
	user.nickname,
	"--admin",
	"--confirmed",
	"--locked",
	"--moderator"
	])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Admin status .* true/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Confirmation pending .* false/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Locked status .* true/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Moderator status .* true/

	user = User.get_cached_by_nickname(user.nickname)
	assert user.is_moderator
	assert user.is_locked
	assert user.is_admin
	refute user.confirmation_pending
	end

	test "All statuses unset" do
	user =
	insert(:user,
	is_locked: true,
	is_moderator: true,
	is_admin: true,
	confirmation_pending: true
	)

	Mix.Tasks.Pleroma.User.run([
	"set",
	user.nickname,
	"--no-admin",
	"--no-confirmed",
	"--no-locked",
	"--no-moderator"
	])

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Admin status .* false/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Confirmation pending .* true/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Locked status .* false/

	assert_received {:mix_shell, :info, [message]}
	assert message =~ ~r/Moderator status .* false/

	user = User.get_cached_by_nickname(user.nickname)
	refute user.is_moderator
	refute user.is_locked
	refute user.is_admin
	assert user.confirmation_pending
	end

	test "no user to set status" do
	Mix.Tasks.Pleroma.User.run(["set", "nonexistent", "--moderator"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "running reset_password" do
	test "password reset token is generated" do
	user = insert(:user)

	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run(["reset_password", user.nickname])
	end) =~ "URL:"

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Generated"
	end

	test "no user to reset password" do
	Mix.Tasks.Pleroma.User.run(["reset_password", "nonexistent"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "running reset_mfa" do
	test "disables MFA" do
	user =
	insert(:user,
	multi_factor_authentication_settings: %MFA.Settings{
	enabled: true,
	totp: %MFA.Settings.TOTP{secret: "xx", confirmed: true}
	}
	)

	Mix.Tasks.Pleroma.User.run(["reset_mfa", user.nickname])

	assert_received {:mix_shell, :info, [message]}
	assert message == "Multi-Factor Authentication disabled for #{user.nickname}"

	assert %{enabled: false, totp: false} ==
	user.nickname
	\|> User.get_cached_by_nickname()
	\|> MFA.mfa_settings()
	end

	test "no user to reset MFA" do
	Mix.Tasks.Pleroma.User.run(["reset_password", "nonexistent"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "running invite" do
	test "invite token is generated" do
	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run(["invite"])
	end) =~ "http"

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Generated user invite token one time"
	end

	test "token is generated with expires_at" do
	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run([
	"invite",
	"--expires-at",
	Date.to_string(Date.utc_today())
	])
	end)

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Generated user invite token date limited"
	end

	test "token is generated with max use" do
	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run([
	"invite",
	"--max-use",
	"5"
	])
	end)

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Generated user invite token reusable"
	end

	test "token is generated with max use and expires date" do
	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run([
	"invite",
	"--max-use",
	"5",
	"--expires-at",
	Date.to_string(Date.utc_today())
	])
	end)

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Generated user invite token reusable date limited"
	end
	end

	describe "running invites" do
	test "invites are listed" do
	{:ok, invite} = Pleroma.UserInviteToken.create_invite()

	{:ok, invite2} =
	Pleroma.UserInviteToken.create_invite(%{expires_at: Date.utc_today(), max_use: 15})

	# assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run([
	"invites"
	])

	# end)

	assert_received {:mix_shell, :info, [message]}
	assert_received {:mix_shell, :info, [message2]}
	assert_received {:mix_shell, :info, [message3]}
	assert message =~ "Invites list:"
	assert message2 =~ invite.invite_type
	assert message3 =~ invite2.invite_type
	end
	end

	describe "running revoke_invite" do
	test "invite is revoked" do
	{:ok, invite} = Pleroma.UserInviteToken.create_invite(%{expires_at: Date.utc_today()})

	assert capture_io(fn ->
	Mix.Tasks.Pleroma.User.run([
	"revoke_invite",
	invite.token
	])
	end)

	assert_received {:mix_shell, :info, [message]}
	assert message =~ "Invite for token #{invite.token} was revoked."
	end

	test "it prints an error message when invite is not exist" do
	Mix.Tasks.Pleroma.User.run(["revoke_invite", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No invite found"
	end
	end

	describe "running delete_activities" do
	test "activities are deleted" do
	%{nickname: nickname} = insert(:user)

	assert :ok == Mix.Tasks.Pleroma.User.run(["delete_activities", nickname])
	assert_received {:mix_shell, :info, [message]}
	assert message == "User #{nickname} statuses deleted."
	end

	test "it prints an error message when user is not exist" do
	Mix.Tasks.Pleroma.User.run(["delete_activities", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "running toggle_confirmed" do
	test "user is confirmed" do
	%{id: id, nickname: nickname} = insert(:user, confirmation_pending: false)

	assert :ok = Mix.Tasks.Pleroma.User.run(["toggle_confirmed", nickname])
	assert_received {:mix_shell, :info, [message]}
	assert message == "#{nickname} needs confirmation."

	user = Repo.get(User, id)
	assert user.confirmation_pending
	assert user.confirmation_token
	end

	test "user is not confirmed" do
	%{id: id, nickname: nickname} =
	insert(:user, confirmation_pending: true, confirmation_token: "some token")

	assert :ok = Mix.Tasks.Pleroma.User.run(["toggle_confirmed", nickname])
	assert_received {:mix_shell, :info, [message]}
	assert message == "#{nickname} doesn't need confirmation."

	user = Repo.get(User, id)
	refute user.confirmation_pending
	refute user.confirmation_token
	end

	test "it prints an error message when user is not exist" do
	Mix.Tasks.Pleroma.User.run(["toggle_confirmed", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "search" do
	test "it returns users matching" do
	user = insert(:user)
	moon = insert(:user, nickname: "moon", name: "fediverse expert moon")
	moot = insert(:user, nickname: "moot")
	kawen = insert(:user, nickname: "kawen", name: "fediverse expert moon")

	{:ok, user} = User.follow(user, moon)

	assert [moon.id, kawen.id] == User.Search.search("moon") \|> Enum.map(& &1.id)

	res = User.search("moo") \|> Enum.map(& &1.id)
	assert Enum.sort([moon.id, moot.id, kawen.id]) == Enum.sort(res)

	assert [kawen.id, moon.id] == User.Search.search("expert fediverse") \|> Enum.map(& &1.id)

	assert [moon.id, kawen.id] ==
	User.Search.search("expert fediverse", for_user: user) \|> Enum.map(& &1.id)
	end
	end

	describe "signing out" do
	test "it deletes all user's tokens and authorizations" do
	user = insert(:user)
	insert(:oauth_token, user: user)
	insert(:oauth_authorization, user: user)

	assert Repo.get_by(Token, user_id: user.id)
	assert Repo.get_by(Authorization, user_id: user.id)

	:ok = Mix.Tasks.Pleroma.User.run(["sign_out", user.nickname])

	refute Repo.get_by(Token, user_id: user.id)
	refute Repo.get_by(Authorization, user_id: user.id)
	end

	test "it prints an error message when user is not exist" do
	Mix.Tasks.Pleroma.User.run(["sign_out", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "No local user"
	end
	end

	describe "tagging" do
	test "it add tags to a user" do
	user = insert(:user)

	:ok = Mix.Tasks.Pleroma.User.run(["tag", user.nickname, "pleroma"])

	user = User.get_cached_by_nickname(user.nickname)
	assert "pleroma" in user.tags
	end

	test "it prints an error message when user is not exist" do
	Mix.Tasks.Pleroma.User.run(["tag", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "Could not change user tags"
	end
	end

	describe "untagging" do
	test "it deletes tags from a user" do
	user = insert(:user, tags: ["pleroma"])
	assert "pleroma" in user.tags

	:ok = Mix.Tasks.Pleroma.User.run(["untag", user.nickname, "pleroma"])

	user = User.get_cached_by_nickname(user.nickname)
	assert Enum.empty?(user.tags)
	end

	test "it prints an error message when user is not exist" do
	Mix.Tasks.Pleroma.User.run(["untag", "foo"])

	assert_received {:mix_shell, :error, [message]}
	assert message =~ "Could not change user tags"
	end
	end

	describe "bulk confirm and unconfirm" do
	test "confirm all" do
	user1 = insert(:user, confirmation_pending: true)
	user2 = insert(:user, confirmation_pending: true)

	assert user1.confirmation_pending
	assert user2.confirmation_pending

	Mix.Tasks.Pleroma.User.run(["confirm_all"])

	user1 = User.get_cached_by_nickname(user1.nickname)
	user2 = User.get_cached_by_nickname(user2.nickname)

	refute user1.confirmation_pending
	refute user2.confirmation_pending
	end

	test "unconfirm all" do
	user1 = insert(:user, confirmation_pending: false)
	user2 = insert(:user, confirmation_pending: false)
	admin = insert(:user, is_admin: true, confirmation_pending: false)
	mod = insert(:user, is_moderator: true, confirmation_pending: false)

	refute user1.confirmation_pending
	refute user2.confirmation_pending

	Mix.Tasks.Pleroma.User.run(["unconfirm_all"])

	user1 = User.get_cached_by_nickname(user1.nickname)
	user2 = User.get_cached_by_nickname(user2.nickname)
	admin = User.get_cached_by_nickname(admin.nickname)
	mod = User.get_cached_by_nickname(mod.nickname)

	assert user1.confirmation_pending
	assert user2.confirmation_pending
	refute admin.confirmation_pending
	refute mod.confirmation_pending
	end
	end
	end

File Metadata

Mime Type: text/x-diff
Expires: Sat, Oct 31, 7:01 PM (20 h, 3 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1801693
Default Alt Text: (34 KB)

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions