From 1ee8ee89f17d98920fcbe572b5ad52738ef59eb6 Mon Sep 17 00:00:00 2001 From: Jamie Cressey Date: Sun, 6 Dec 2015 20:25:01 +0000 Subject: [PATCH] Adding SQS tests --- beaver/tests/test_sqs_transport.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/beaver/tests/test_sqs_transport.py b/beaver/tests/test_sqs_transport.py index 87ea7592..8fa3d081 100644 --- a/beaver/tests/test_sqs_transport.py +++ b/beaver/tests/test_sqs_transport.py @@ -50,6 +50,7 @@ def test_sqs_default_auth_profile(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_auth_profile(cls): @@ -74,6 +75,7 @@ def test_sqs_auth_key(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_auth_account_id(cls): @@ -87,6 +89,7 @@ def test_sqs_auth_account_id(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_single_queue(cls): @@ -96,6 +99,7 @@ def test_sqs_single_queue(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_single_queue_bulklines(cls): @@ -106,6 +110,7 @@ def test_sqs_single_queue_bulklines(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_multi_queue(cls): @@ -115,6 +120,7 @@ def test_sqs_multi_queue(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() @mock_sqs def test_sqs_multi_queue_bulklines(cls): @@ -125,3 +131,27 @@ def test_sqs_multi_queue_bulklines(cls): transport = create_transport(cls.beaver_config, logger=cls.logger) cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + transport.interrupt() + + def test_sqs_send_single_queue(cls): + cls._create_queues() + cls.beaver_config.set('sqs_aws_queue', 'queue1') + + transport = create_transport(cls.beaver_config, logger=cls.logger) + + cls.assertIsInstance(transport, beaver.transports.sqs_transport.SqsTransport) + + data = {} + lines = [] + n=100 + for i in range(n): + lines.append('log' + str(i) + '\n') + new_lines = [] + for line in lines: + message = unicode_dammit(line) + if len(message) == 0: + continue + new_lines.append(message) + data['lines'] = new_lines + data['fields'] = [] + transport.callback("test.log", **data)