131 lines
4.4 KiB
Elixir
131 lines
4.4 KiB
Elixir
## The contents of this file are subject to the Mozilla Public License
|
|
## Version 1.1 (the "License"); you may not use this file except in
|
|
## compliance with the License. You may obtain a copy of the License
|
|
## at http://www.mozilla.org/MPL/
|
|
##
|
|
## Software distributed under the License is distributed on an "AS IS"
|
|
## basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
|
|
## the License for the specific language governing rights and
|
|
## limitations under the License.
|
|
##
|
|
## The Original Code is RabbitMQ.
|
|
##
|
|
## The Initial Developer of the Original Code is GoPivotal, Inc.
|
|
## Copyright (c) 2007-2016 Pivotal Software, Inc. All rights reserved.
|
|
|
|
|
|
defmodule SetPermissionsCommandTest do
|
|
use ExUnit.Case, async: false
|
|
import TestHelper
|
|
|
|
@vhost "test1"
|
|
@user "guest"
|
|
@root "/"
|
|
|
|
setup_all do
|
|
RabbitMQCtl.start_distribution(%{})
|
|
:net_kernel.connect_node(get_rabbit_hostname)
|
|
|
|
add_vhost @vhost
|
|
|
|
on_exit([], fn ->
|
|
delete_vhost @vhost
|
|
:erlang.disconnect_node(get_rabbit_hostname)
|
|
:net_kernel.stop()
|
|
end)
|
|
|
|
:ok
|
|
end
|
|
|
|
setup context do
|
|
|
|
on_exit(context, fn ->
|
|
set_permissions context[:user], context[:vhost], [".*", ".*", ".*"]
|
|
end)
|
|
|
|
{
|
|
:ok,
|
|
opts: %{
|
|
node: get_rabbit_hostname
|
|
}
|
|
}
|
|
end
|
|
|
|
test "validate: wrong number of arguments leads to an arg count error" do
|
|
assert SetPermissionsCommand.validate([], %{}) == {:validation_failure, :not_enough_args}
|
|
assert SetPermissionsCommand.validate(["insufficient"], %{}) == {:validation_failure, :not_enough_args}
|
|
assert SetPermissionsCommand.validate(["not", "enough"], %{}) == {:validation_failure, :not_enough_args}
|
|
assert SetPermissionsCommand.validate(["not", "quite", "enough"], %{}) == {:validation_failure, :not_enough_args}
|
|
assert SetPermissionsCommand.validate(["this", "is", "way", "too", "many"], %{}) == {:validation_failure, :too_many_args}
|
|
end
|
|
|
|
@tag user: @user, vhost: @vhost
|
|
test "run: a well-formed, host-specific command returns okay", context do
|
|
vhost_opts = Map.merge(context[:opts], %{vhost: context[:vhost]})
|
|
|
|
assert SetPermissionsCommand.run(
|
|
[context[:user], "^#{context[:user]}-.*", ".*", ".*"],
|
|
vhost_opts
|
|
) == :ok
|
|
|
|
assert List.first(list_permissions(context[:vhost]))[:configure] == "^#{context[:user]}-.*"
|
|
end
|
|
|
|
test "run: An invalid rabbitmq node throws a badrpc" do
|
|
target = :jake@thedog
|
|
:net_kernel.connect_node(target)
|
|
opts = %{node: target}
|
|
|
|
assert SetPermissionsCommand.run([@user, ".*", ".*", ".*"], opts) == {:badrpc, :nodedown}
|
|
end
|
|
|
|
@tag user: @user, vhost: @root
|
|
test "run: a well-formed command with no vhost runs against the default", context do
|
|
assert SetPermissionsCommand.run(
|
|
[context[:user], "^#{context[:user]}-.*", ".*", ".*"],
|
|
context[:opts]
|
|
) == :ok
|
|
|
|
assert List.first(list_permissions(context[:vhost]))[:configure] == "^#{context[:user]}-.*"
|
|
end
|
|
|
|
@tag user: "interloper", vhost: @root
|
|
test "run: an invalid user returns a no-such-user error", context do
|
|
assert SetPermissionsCommand.run(
|
|
[context[:user], "^#{context[:user]}-.*", ".*", ".*"],
|
|
context[:opts]
|
|
) == {:error, {:no_such_user, context[:user]}}
|
|
|
|
assert List.first(list_permissions(context[:vhost]))[:configure] == ".*"
|
|
end
|
|
|
|
@tag user: @user, vhost: "wintermute"
|
|
test "run: an invalid vhost returns a no-such-vhost error", context do
|
|
vhost_opts = Map.merge(context[:opts], %{vhost: context[:vhost]})
|
|
|
|
assert SetPermissionsCommand.run(
|
|
[context[:user], "^#{context[:user]}-.*", ".*", ".*"],
|
|
vhost_opts
|
|
) == {:error, {:no_such_vhost, context[:vhost]}}
|
|
end
|
|
|
|
@tag user: @user, vhost: @root
|
|
test "run: Invalid regex patterns return error", context do
|
|
assert SetPermissionsCommand.run(
|
|
[context[:user], "^#{context[:user]}-.*", ".*", "*"],
|
|
context[:opts]
|
|
) == {:error, {:invalid_regexp, '*', {'nothing to repeat', 0}}}
|
|
|
|
# asserts that the bad command didn't change anything
|
|
assert List.first(list_permissions(context[:vhost])) == [user: context[:user], configure: ".*", write: ".*", read: ".*"]
|
|
end
|
|
|
|
@tag user: @user, vhost: @vhost
|
|
test "banner", context do
|
|
vhost_opts = Map.merge(context[:opts], %{vhost: context[:vhost]})
|
|
|
|
assert SetPermissionsCommand.banner([context[:user], "^#{context[:user]}-.*", ".*", ".*"], vhost_opts)
|
|
=~ ~r/Setting permissions for user \"#{context[:user]}\" in vhost \"#{context[:vhost]}\" \.\.\./
|
|
end
|
|
end
|