Create service and test directories when they don't exist
This commit is contained in:
parent
5996876043
commit
9cdc0d5070
7 changed files with 140 additions and 16 deletions
7
template/lib/__init__.py.j2
Normal file
7
template/lib/__init__.py.j2
Normal 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)
|
||||
|
||||
4
template/lib/exceptions.py.j2
Normal file
4
template/lib/exceptions.py.j2
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
from __future__ import unicode_literals
|
||||
from moto.core.exceptions import RESTError
|
||||
|
||||
|
||||
20
template/lib/models.py.j2
Normal file
20
template/lib/models.py.j2
Normal 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}
|
||||
15
template/lib/responses.py.j2
Normal file
15
template/lib/responses.py.j2
Normal 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
|
||||
|
||||
16
template/test/test_server.py.j2
Normal file
16
template/test/test_server.py.j2
Normal 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
|
||||
11
template/test/test_service.py.j2
Normal file
11
template/test/test_service.py.j2
Normal 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
|
||||
Loading…
Add table
Add a link
Reference in a new issue