diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/integration/test_gemma.py | 1 | ||||
| -rw-r--r-- | tests/unit/test_commands.py | 5 |
2 files changed, 5 insertions, 1 deletions
diff --git a/tests/integration/test_gemma.py b/tests/integration/test_gemma.py index 7f84b50..ed05cac 100644 --- a/tests/integration/test_gemma.py +++ b/tests/integration/test_gemma.py @@ -60,6 +60,7 @@ class GemmaAPITest(unittest.TestCase): mock_queue_cmd.assert_has_calls( [mock.call(conn=_redis_conn, email="me@me.com", + job_queue="GN3::job-queue", cmd=("gemma-wrapper --json -- -g " "genofile.txt -p test.txt " "-a genofile_snps.txt -gk > " diff --git a/tests/unit/test_commands.py b/tests/unit/test_commands.py index d96706c..846d20c 100644 --- a/tests/unit/test_commands.py +++ b/tests/unit/test_commands.py @@ -41,6 +41,7 @@ class TestCommands(unittest.TestCase): self.assertRaises(RedisConnectionError, queue_cmd, cmd="ls", + job_queue="GN2::job-queue", conn=MockRedis(ping=lambda: False, hset=mock.MagicMock(), rpush=mock.MagicMock())) @@ -59,7 +60,8 @@ class TestCommands(unittest.TestCase): rpush=mock.MagicMock()) actual_unique_id = "cmd::2021-02-1217-3224-3224-1234" self.assertEqual(queue_cmd(cmd="ls", - conn=mock_redis_conn), + conn=mock_redis_conn, + job_queue="GN2::job-queue"), actual_unique_id) mock_redis_conn.hset.assert_has_calls( [mock.call("cmd", "ls", actual_unique_id), @@ -84,6 +86,7 @@ class TestCommands(unittest.TestCase): actual_unique_id = "cmd::2021-02-1217-3224-3224-1234" self.assertEqual(queue_cmd(cmd="ls", conn=mock_redis_conn, + job_queue="GN2::job-queue", email="me@me.com"), actual_unique_id) mock_redis_conn.hset.assert_has_calls( |