def test_rqinfo_can_choose_queue(self):
parser = rqinfo.setup_parser()
args = parser.parse_args(self.base_arguments + ["B"])
setup_default_arguments(args, {})
setup_redis(args)
output = self.capture_stdout(rqinfo.show_queues, args)
connections.use_connection(self.testconn)
expected_output = ["queue B 1"]
self.assertEqual(output.splitlines(), expected_output)
def test_rqinfo_defaults_to_all_queues(self):
parser = rqinfo.setup_parser()
args = parser.parse_args(self.base_arguments)
setup_default_arguments(args, {})
setup_redis(args) # This uses use_connection and clears self.testconn from the stack
output = self.capture_stdout(rqinfo.show_queues, args)
connections.use_connection(self.testconn)
expected_output = set(["queue B 1", "queue C 1", "queue A 1"])
self.assertEqual(set(output.splitlines()), expected_output)
