diff options
-rw-r--r-- | t/unit/asynchronous/aws/sqs/test_connection.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/t/unit/asynchronous/aws/sqs/test_connection.py b/t/unit/asynchronous/aws/sqs/test_connection.py index abb63ef2..7623a6d1 100644 --- a/t/unit/asynchronous/aws/sqs/test_connection.py +++ b/t/unit/asynchronous/aws/sqs/test_connection.py @@ -90,7 +90,10 @@ class test_AsyncSQSConnection(AWSCase): callback=self.callback, ) self.x.get_list.assert_called_with( - 'ReceiveMessage', {'MaxNumberOfMessages': 4}, + 'ReceiveMessage', { + 'MaxNumberOfMessages': 4, + 'AttributeName.1': 'ApproximateReceiveCount' + }, [('Message', AsyncMessage)], 'http://aws.com', callback=self.callback, parent=queue, @@ -109,6 +112,7 @@ class test_AsyncSQSConnection(AWSCase): 'ReceiveMessage', { 'MaxNumberOfMessages': 4, 'VisibilityTimeout': 3666, + 'AttributeName.1': 'ApproximateReceiveCount', }, [('Message', AsyncMessage)], 'http://aws.com', callback=self.callback, @@ -128,6 +132,7 @@ class test_AsyncSQSConnection(AWSCase): 'ReceiveMessage', { 'MaxNumberOfMessages': 4, 'WaitTimeSeconds': 303, + 'AttributeName.1': 'ApproximateReceiveCount', }, [('Message', AsyncMessage)], 'http://aws.com', callback=self.callback, |