Generalize decorator code.

This commit is contained in:
Steve Pulec 2017-02-12 00:22:29 -05:00
commit d3df810065
21 changed files with 53 additions and 160 deletions

View file

@ -1,12 +1,6 @@
from __future__ import unicode_literals
from .models import cloudformation_backends
from ..core.models import MockAWS
from ..core.models import MockAWS, base_decorator
cloudformation_backend = cloudformation_backends['us-east-1']
def mock_cloudformation(func=None):
if func:
return MockAWS(cloudformation_backends)(func)
else:
return MockAWS(cloudformation_backends)
mock_cloudformation = base_decorator(cloudformation_backends)