Adding SQS tests

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)