| 23 | |
| 24 | class TestQueueUtils(TestCase): |
| 25 | def test_get_queue_name(self): |
| 26 | self.assertRaises( |
| 27 | ValueError, |
| 28 | queue_utils.get_queue_name, |
| 29 | queue_name_base=None, |
| 30 | queue_name_suffix=None, |
| 31 | ) |
| 32 | self.assertRaises( |
| 33 | ValueError, |
| 34 | queue_utils.get_queue_name, |
| 35 | queue_name_base="", |
| 36 | queue_name_suffix=None, |
| 37 | ) |
| 38 | self.assertEqual( |
| 39 | queue_utils.get_queue_name( |
| 40 | queue_name_base="st2.test.watch", queue_name_suffix=None |
| 41 | ), |
| 42 | "st2.test.watch", |
| 43 | ) |
| 44 | self.assertEqual( |
| 45 | queue_utils.get_queue_name( |
| 46 | queue_name_base="st2.test.watch", queue_name_suffix="" |
| 47 | ), |
| 48 | "st2.test.watch", |
| 49 | ) |
| 50 | queue_name = queue_utils.get_queue_name( |
| 51 | queue_name_base="st2.test.watch", |
| 52 | queue_name_suffix="foo", |
| 53 | add_random_uuid_to_suffix=True, |
| 54 | ) |
| 55 | pattern = re.compile(r"st2.test.watch.foo-\w") |
| 56 | self.assertTrue(re.match(pattern, queue_name)) |
| 57 | |
| 58 | queue_name = queue_utils.get_queue_name( |
| 59 | queue_name_base="st2.test.watch", |
| 60 | queue_name_suffix="foo", |
| 61 | add_random_uuid_to_suffix=False, |
| 62 | ) |
| 63 | self.assertEqual(queue_name, "st2.test.watch.foo") |