add accept parameter to SimpleQueue class (#1140)

* add accept parameter to SimpleQueue class

* Fixed missing accept for get_nowait() and added unittests

* Remove unused **kwargs from SimpleQueue.__init__

* Use self.consumer.accept instead of new attribute in SimpleQueue

* Add tests for simple interface when accept=[]

Co-authored-by: Matus Valo <matusvalo@gmail.com>

1 files changed, 33 insertions, 3 deletions

diff --git a/t/unit/test_simple.py b/t/unit/test_simple.py
index 1f9cc8e2..c4badb03 100644
--- a/t/unit/test_simple.py
+++ b/t/unit/test_simple.py
@@ -3,6 +3,7 @@ import pytest
 from unittest.mock import Mock
 
 from kombu import Connection, Exchange, Queue
+from kombu.exceptions import ContentDisallowed
 
 
 class SimpleBase:
@@ -21,13 +22,10 @@ class SimpleBase:
     def setup(self):
         self.connection = Connection(transport='memory')
         self.connection.default_channel.exchange_declare('amq.direct')
-        self.q = self.Queue(None, no_ack=True)
 
     def teardown(self):
-        self.q.close()
         self.connection.close()
         self.connection = None
-        self.q = None
 
     def test_produce__consume(self):
         q = self.Queue('test_produce__consume', no_ack=True)
@@ -50,6 +48,38 @@ class SimpleBase:
         with pytest.raises(q.Empty):
             q.get(block=False)
 
+    def test_get_nowait_accept(self):
+        q = self.Queue('test_accept', serializer='pickle', accept=['json'])
+        q.put({'hello': 'SimpleSync'})
+        with pytest.raises(ContentDisallowed):
+            q.get_nowait().payload
+
+        q = self.Queue('test_accept1', serializer='json', accept=[])
+        q.put({'hello': 'SimpleSync'})
+        with pytest.raises(ContentDisallowed):
+            q.get_nowait().payload
+
+        q = self.Queue(
+            'test_accept2', serializer='pickle', accept=['json', 'pickle'])
+        q.put({'hello': 'SimpleSync'})
+        assert q.get_nowait().payload == {'hello': 'SimpleSync'}
+
+    def test_get_accept(self):
+        q = self.Queue('test_accept', serializer='pickle', accept=['json'])
+        q.put({'hello': 'SimpleSync'})
+        with pytest.raises(ContentDisallowed):
+            q.get().payload
+
+        q = self.Queue('test_accept1', serializer='pickle', accept=[])
+        q.put({'hello': 'SimpleSync'})
+        with pytest.raises(ContentDisallowed):
+            q.get().payload
+
+        q = self.Queue(
+            'test_accept2', serializer='pickle', accept=['json', 'pickle'])
+        q.put({'hello': 'SimpleSync'})
+        assert q.get().payload == {'hello': 'SimpleSync'}
+
     def test_clear(self):
         q = self.Queue('test_clear', no_ack=True)