Need to run the whole test suite to make sure no regressions were added
This commit is contained in:
Andres Riancho 2013-12-26 13:12:50 -03:00
parent 3b8cd5ad7d
commit 748fa52cca
2 changed files with 40 additions and 3 deletions

View File

@ -7,9 +7,13 @@ from .utils import convert_regex_to_flask_path
class MockAWS(object):
nested_count = 0
def __init__(self, backend):
self.backend = backend
HTTPretty.reset()
if self.__class__.nested_count == 0:
HTTPretty.reset()
def __call__(self, func):
return self.decorate_callable(func)
@ -22,7 +26,9 @@ class MockAWS(object):
def start(self):
self.backend.reset()
HTTPretty.enable()
if not HTTPretty.is_enabled():
HTTPretty.enable()
for method in HTTPretty.METHODS:
for key, value in self.backend.urls.iteritems():
@ -40,7 +46,10 @@ class MockAWS(object):
)
def stop(self):
HTTPretty.disable()
self.__class__.nested_count -= 1
if self.__class__.nested_count == 0:
HTTPretty.disable()
def decorate_callable(self, func):
def wrapper(*args, **kwargs):

View File

@ -0,0 +1,28 @@
import unittest
from boto.sqs.connection import SQSConnection
from boto.sqs.message import Message
from boto.ec2 import EC2Connection
from moto import mock_sqs, mock_ec2
class TestNestedDecorators(unittest.TestCase):
@mock_sqs
def setup_sqs_queue(self):
conn = SQSConnection()
q = conn.create_queue('some-queue')
m = Message()
m.set_body('This is my first message.')
q.write(m)
self.assertEqual(q.count(), 1)
@mock_ec2
def test_nested(self):
self.setup_sqs_queue()
conn = EC2Connection()
conn.run_instances('ami-123456')