Started on batch

This commit is contained in:
Terry Cain 2017-09-26 17:37:26 +01:00
parent b17136e36c
commit bba6d23eae
6 changed files with 73 additions and 0 deletions

6
moto/batch/__init__.py Normal file
View File

@ -0,0 +1,6 @@
from __future__ import unicode_literals
from .models import batch_backends
from ..core.models import base_decorator
batch_backend = batch_backends['us-east-1']
mock_batch = base_decorator(batch_backends)

3
moto/batch/exceptions.py Normal file
View File

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

23
moto/batch/models.py Normal file
View File

@ -0,0 +1,23 @@
from __future__ import unicode_literals
import boto3
from moto.core import BaseBackend, BaseModel
class BatchBackend(BaseBackend):
def __init__(self, region_name=None):
super(BatchBackend, self).__init__()
self.region_name = region_name
def reset(self):
region_name = self.region_name
self.__dict__ = {}
self.__init__(region_name)
def create_compute_environment(self, compute_environment_name, type, state, compute_resources, service_role):
# implement here
return compute_environment_name, compute_environment_arn
# add methods from here
available_regions = boto3.session.Session().get_available_regions("batch")
batch_backends = {region: BatchBackend for region in available_regions}

14
moto/batch/responses.py Normal file
View File

@ -0,0 +1,14 @@
from __future__ import unicode_literals
from moto.core.responses import BaseResponse
from .models import batch_backends
class BatchResponse(BaseResponse):
@property
def batch_backend(self):
return batch_backends[self.region]
# add methods from here
# add teampltes from here

View File

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

View File

@ -0,0 +1,16 @@
from __future__ import unicode_literals
import sure # noqa
import moto.server as server
from moto import mock_batch
'''
Test the different server responses
'''
@mock_batch
def test_batch_list():
backend = server.create_backend_app("batch")
test_client = backend.test_client()
# do test