Add BaseModel to all models.

This commit is contained in:
Steve Pulec 2017-03-11 23:41:12 -05:00
commit 6d422d1f37
34 changed files with 122 additions and 116 deletions

View file

@ -2,11 +2,11 @@ from __future__ import unicode_literals
import datetime
import boto.datapipeline
from moto.core import BaseBackend
from moto.core import BaseBackend, BaseModel
from .utils import get_random_pipeline_id, remove_capitalization_of_dict_keys
class PipelineObject(object):
class PipelineObject(BaseModel):
def __init__(self, object_id, name, fields):
self.object_id = object_id
@ -21,7 +21,7 @@ class PipelineObject(object):
}
class Pipeline(object):
class Pipeline(BaseModel):
def __init__(self, name, unique_id):
self.name = name