Skip to content

Commit

Permalink
According to what I said on getmoto#1026 Add a unittest and attempt a…
Browse files Browse the repository at this point in the history
… fix.
  • Loading branch information
hoangelos committed Feb 23, 2018
1 parent 553d074 commit aec58d2
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 3 deletions.
3 changes: 2 additions & 1 deletion moto/core/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,8 @@ def reset(self):
responses.reset()

def enable_patching(self):
responses.start()
if BaseMockAWS.nested_count == 1:
responses.start()
for method in RESPONSES_METHODS:
for backend in self.backends_for_urls.values():
for key, value in backend.urls.items():
Expand Down
17 changes: 15 additions & 2 deletions tests/test_core/test_nested.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,10 @@
from boto.sqs.connection import SQSConnection
from boto.sqs.message import Message
from boto.ec2 import EC2Connection
import requests

from moto import mock_sqs_deprecated, mock_ec2_deprecated
from moto import mock_sqs_deprecated, mock_ec2_deprecated, mock_sqs, mock_sns
from moto.packages import httpretty


class TestNestedDecorators(unittest.TestCase):
Expand All @@ -24,6 +26,17 @@ def setup_sqs_queue(self):
@mock_ec2_deprecated
def test_nested(self):
self.setup_sqs_queue()

conn = EC2Connection()
conn.run_instances('ami-123456')

@mock_sqs
@mock_sns
def test_multiple_mocks_part1(self):
pass

@httpretty.activate
def test_multiple_mocks_part2(self):
httpretty.register_uri(httpretty.GET, 'http://www.google.com')
response = requests.get('http://www.google.com')
assert(response.status_code == 200)
assert(response.text == 'HTTPretty :)')

0 comments on commit aec58d2

Please sign in to comment.