summaryrefslogtreecommitdiff
path: root/deps/rabbitmq_cli/test/ctl/list_queues_command_test.exs
blob: a6635c79339f0fde84172b86efc61b20d429d6a0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
defmodule ListQueuesCommandTest do
  use ExUnit.Case, async: false
  import TestHelper

  @command RabbitMQ.CLI.Ctl.Commands.ListQueuesCommand

  @vhost "test1"
  @user "guest"
  @default_timeout 15000

  setup_all do
    RabbitMQ.CLI.Core.Distribution.start()

    reset_vm_memory_high_watermark()
    delete_all_queues()
    close_all_connections(get_rabbit_hostname())

    on_exit([], fn ->
      delete_all_queues()
      close_all_connections(get_rabbit_hostname())
    end)

    :ok
  end

  setup context do
    add_vhost @vhost
    set_permissions @user, @vhost, [".*", ".*", ".*"]
    on_exit(fn ->
      delete_vhost @vhost
    end)
    {
      :ok,
      opts: %{
        quiet: true,
        node: get_rabbit_hostname(),
        timeout: context[:test_timeout] || @default_timeout,
        vhost: @vhost,
        offline: false,
        online: false,
        local: false
      }
    }
  end

  test "merge_defaults: no info keys returns names and message count" do
    assert match?({["name", "messages"], _}, @command.merge_defaults([], %{}))
  end

  test "validate: returns bad_info_key on a single bad arg", context do
    assert @command.validate(["quack"], context[:opts]) ==
      {:validation_failure, {:bad_info_key, [:quack]}}
  end

  test "validate: multiple bad args return a list of bad info key values", context do
    assert @command.validate(["quack", "oink"], context[:opts]) ==
      {:validation_failure, {:bad_info_key, [:oink, :quack]}}
  end

  test "validate: return bad_info_key on mix of good and bad args", context do
    assert @command.validate(["quack", "messages"], context[:opts]) ==
      {:validation_failure, {:bad_info_key, [:quack]}}
    assert @command.validate(["name", "oink"], context[:opts]) ==
      {:validation_failure, {:bad_info_key, [:oink]}}
    assert @command.validate(["name", "oink", "messages"], context[:opts]) ==
      {:validation_failure, {:bad_info_key, [:oink]}}
  end

  @tag test_timeout: 0
  test "run: timeout causes command to return badrpc", context do
    assert run_command_to_list(@command, [["name"], context[:opts]]) ==
      [{:badrpc, {:timeout, 0.0, "Some queue(s) are unresponsive, use list_unresponsive_queues command."}}]
  end

  @tag test_timeout: 1
  test "run: command timeout (several thousands queues in 1ms) return badrpc with timeout value in seconds", context do
    # we assume it will take longer than 1 ms to list thousands of queues
    n = 5000
    for i <- 1..n do
        declare_queue("test_queue_" <> Integer.to_string(i), @vhost)
    end
    assert run_command_to_list(@command, [["name"], context[:opts]]) ==
      [{:badrpc, {:timeout, 0.001, "Some queue(s) are unresponsive, use list_unresponsive_queues command."}}]
    for i <- 1..n do
        delete_queue("test_queue_" <> Integer.to_string(i), @vhost)
    end
  end

  @tag test_timeout: 5000
  test "run: return multiple queues", context do
    declare_queue("test_queue_1", @vhost)
    publish_messages(@vhost, "test_queue_1", 3)
    declare_queue("test_queue_2", @vhost)
    publish_messages(@vhost, "test_queue_2", 1)
    assert Keyword.equal?(run_command_to_list(@command, [["name", "messages"], context[:opts]]),
      [[name: "test_queue_1", messages: 3],
       [name: "test_queue_2", messages: 1]])
  end

  @tag test_timeout: 5000
  test "run: info keys filter single key", context do
    declare_queue("test_queue_1", @vhost)
    declare_queue("test_queue_2", @vhost)
    assert Keyword.equal?(run_command_to_list(@command, [["name"], context[:opts]]),
      [[name: "test_queue_1"],
       [name: "test_queue_2"]])
  end

  @tag test_timeout: 5000
  test "run: info keys add additional keys", context do
    declare_queue("durable_queue", @vhost, true)
    publish_messages(@vhost, "durable_queue", 3)
    declare_queue("auto_delete_queue", @vhost, false, true)
    publish_messages(@vhost, "auto_delete_queue", 1)
    assert Keyword.equal?(
      run_command_to_list(@command, [["name", "messages", "durable", "auto_delete"], context[:opts]]),
      [[name: "durable_queue", messages: 3, durable: true, auto_delete: false],
       [name: "auto_delete_queue", messages: 1, durable: false, auto_delete: true]])
  end

  @tag test_timeout: 5000
  test "run: info keys order is preserved", context do
    declare_queue("durable_queue", @vhost, true)
    publish_messages(@vhost, "durable_queue", 3)
    declare_queue("auto_delete_queue", @vhost, false, true)
    publish_messages(@vhost, "auto_delete_queue", 1)
    assert Keyword.equal?(
      run_command_to_list(@command, [["messages", "durable", "name", "auto_delete"], context[:opts]]),
      [[messages: 3, durable: true, name: "durable_queue", auto_delete: false],
       [messages: 1, durable: false, name: "auto_delete_queue", auto_delete: true]])
  end

  @tag test_timeout: 5000
  test "run: specifying a vhost returns the targeted vhost queues", context do
    other_vhost = "other_vhost"
    add_vhost other_vhost
    on_exit(fn ->
      delete_vhost other_vhost
    end)
    declare_queue("test_queue_1", @vhost)
    declare_queue("test_queue_2", other_vhost)
    assert run_command_to_list(@command, [["name"], context[:opts]]) == [[name: "test_queue_1"]]
    assert run_command_to_list(@command, [["name"], %{context[:opts] | :vhost => other_vhost}]) == [[name: "test_queue_2"]]
  end
end