move helper script to scripts dir and add one to Makefile
This commit is contained in:
parent
f8cdb50f46
commit
4cc4b36f15
9 changed files with 5 additions and 1 deletions
7
scripts/template/lib/__init__.py.j2
Normal file
7
scripts/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
scripts/template/lib/exceptions.py.j2
Normal file
4
scripts/template/lib/exceptions.py.j2
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
from __future__ import unicode_literals
|
||||
from moto.core.exceptions import RESTError
|
||||
|
||||
|
||||
20
scripts/template/lib/models.py.j2
Normal file
20
scripts/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
scripts/template/lib/responses.py.j2
Normal file
15
scripts/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
scripts/template/test/test_server.py.j2
Normal file
16
scripts/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
scripts/template/test/test_service.py.j2
Normal file
11
scripts/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