Create service and test directories when they don't exist

This commit is contained in:
Toshiya Kawasaki 2017-09-20 03:14:14 +09:00
commit 9cdc0d5070
7 changed files with 140 additions and 16 deletions

View file

@ -0,0 +1,7 @@
from __future__ import unicode_literals
from .models import {{ service }}_backends
from ..core.models import base_decorator
{{ service }}_backend = {{ service }}_backends['us-east-1']
mock_{{ service }} = base_decorator({{ service }}_backends)

View file

@ -0,0 +1,4 @@
from __future__ import unicode_literals
from moto.core.exceptions import RESTError

20
template/lib/models.py.j2 Normal file
View file

@ -0,0 +1,20 @@
from __future__ import unicode_literals
import boto3
from moto.core import BaseBackend, BaseModel
class {{ service_class }}Backend(BaseBackend):
def __init__(self, region_name=None):
super({{ service_class }}Backend, self).__init__()
self.region_name = region_name
def reset(self):
region_name = self.region_name
self.__dict__ = {}
self.__init__(region_name)
# add methods from here
available_regions = boto3.session.Session().get_available_regions("{{ service }}")
{{ service }}_backends = {region: {{ service_class }}Backend for region in available_regions}

View file

@ -0,0 +1,15 @@
from __future__ import unicode_literals
from moto.core.responses import BaseResponse
from .models import {{ service }}_backends
class {{ service_class }}Response(BaseResponse):
@property
def {{ service }}_backend(self):
return {{ service }}_backends[self.region]
# add methods from here
# add teampltes from here

View file

@ -0,0 +1,16 @@
from __future__ import unicode_literals
import sure # noqa
import moto.server as server
from moto import mock_{{ service }}
'''
Test the different server responses
'''
@mock_{{ service }}
def test_{{ service }}_list():
backend = server.create_backend_app("{{ service }}")
test_client = backend.test_client()
# do test

View file

@ -0,0 +1,11 @@
from __future__ import unicode_literals
import boto3
import sure # noqa
from moto import mock_{{ service }}
@mock_{{ service }}
def test_list():
# do test
pass