diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/tests/basic.py | 24 | ||||
| -rw-r--r-- | python/tests/broker.py | 18 |
2 files changed, 42 insertions, 0 deletions
diff --git a/python/tests/basic.py b/python/tests/basic.py index c3a2414eac..75d49ec805 100644 --- a/python/tests/basic.py +++ b/python/tests/basic.py @@ -113,3 +113,27 @@ class BasicTests(TestBase): except Closed, e: self.assertConnectionException(530, e.args[0]) + def test_basic_cancel(self): + """ + Test compliance of the basic.cancel method + """ + channel = self.channel + #setup, declare a queue: + channel.queue_declare(queue="test-queue-4", exclusive=True) + channel.basic_consume(consumer_tag="my-consumer", queue="test-queue-4") + channel.basic_publish(routing_key="test-queue-4", content=Content("One")) + + #cancel should stop messages being delivered + channel.basic_cancel(consumer_tag="my-consumer") + channel.basic_publish(routing_key="test-queue-4", content=Content("Two")) + myqueue = self.client.queue("my-consumer") + msg = myqueue.get(timeout=1) + self.assertEqual("One", msg.content.body) + try: + msg = myqueue.get(timeout=1) + self.fail("Got message after cancellation: " + msg) + except Empty: None + + #cancellation of non-existant consumers should be handled without error + channel.basic_cancel(consumer_tag="my-consumer") + channel.basic_cancel(consumer_tag="this-never-existed") diff --git a/python/tests/broker.py b/python/tests/broker.py index 1345076604..307d447a6c 100644 --- a/python/tests/broker.py +++ b/python/tests/broker.py @@ -13,6 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. # +from qpid.client import Closed from qpid.queue import Empty from qpid.content import Content from qpid.testlib import testrunner, TestBase @@ -82,3 +83,20 @@ class BrokerTests(TestBase): msg = queue.get(timeout=5) self.assert_(msg.content.body == body) + def test_invalid_channel(self): + other = self.connect() + channel = other.channel(200) + try: + channel.queue_declare(exclusive=True) + self.fail("Expected error on queue_declare for invalid channel") + except Closed, e: + self.assertConnectionException(504, e.args[0]) + + channel = self.client.channel(200) + channel.channel_open() + channel.channel_close() + try: + channel.queue_declare(exclusive=True) + self.fail("Expected error on queue_declare for closed channel") + except Closed, e: + self.assertConnectionException(504, e.args[0]) |