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)