From 1ab92602e802a3a33fca5b5f549a665f3c6d4cc4 Mon Sep 17 00:00:00 2001
From: zhangliyong <zhangliyong@umeng.com>
Date: Fri, 22 Aug 2014 10:08:53 +0800
Subject: [PATCH] Add testcase for rq_cli

---
 tests/test_scripts.py | 45 ++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/tests/test_scripts.py b/tests/test_scripts.py
index 88689c3..54b0104 100644
--- a/tests/test_scripts.py
+++ b/tests/test_scripts.py
@@ -2,8 +2,14 @@
 from __future__ import (absolute_import, division, print_function,
                         unicode_literals)
 
+from click.testing import CliRunner
+from rq import get_failed_queue
 from rq.compat import is_python_version
-from rq.scripts import read_config_file
+from rq.job import Job
+from rq.scripts import read_config_file, rq_cli
+
+from tests import RQTestCase
+from tests.fixtures import div_by_zero
 
 if is_python_version((2, 7), (3, 2)):
     from unittest import TestCase
@@ -16,3 +22,40 @@ class TestScripts(TestCase):
         settings = read_config_file("tests.dummy_settings")
         self.assertIn("REDIS_HOST", settings)
         self.assertEqual(settings['REDIS_HOST'], "testhost.example.com")
+
+
+class TestRQCli(RQTestCase):
+    """Test rq_cli script"""
+    def setUp(self):
+        super(TestRQCli, self).setUp()
+        db_num = self.testconn.connection_pool.connection_kwargs['db']
+        self.redis_url = 'redis://127.0.0.1:6379/%d' % db_num
+
+        job = Job.create(func=div_by_zero, args=(1, 2, 3))
+        job.origin = 'fake'
+        job.save()
+        get_failed_queue().quarantine(job, Exception('Some fake error'))  # noqa
+
+    def test_empty(self):
+        """rq -u <url> empty -y"""
+        runner = CliRunner()
+        result = runner.invoke(rq_cli.main,
+                ['-u', self.redis_url, 'empty', "-y"])
+        self.assertEqual(result.exit_code, 0)
+        self.assertEqual(result.output, '1 jobs removed from failed queue\n')
+
+    def test_requeue(self):
+        """rq -u <url> requeue"""
+        runner = CliRunner()
+        result = runner.invoke(rq_cli.main, ['-u', self.redis_url, 'requeue'])
+        self.assertEqual(result.exit_code, 0)
+        self.assertIn('Requeue failed jobs: 1', result.output)
+        self.assertIn('Requeue failed: 0', result.output)
+
+    def test_info(self):
+        """rq -u <url> info -i 0"""
+        runner = CliRunner()
+        result = runner.invoke(rq_cli.main,
+                ['-u', self.redis_url, 'info', '-i 0'])
+        self.assertEqual(result.exit_code, 0)
+        self.assertIn('1 queues, 1 jobs total', result.output)