add ses core
This commit is contained in:
parent
41890225e6
commit
9a92e87f02
7 changed files with 339 additions and 0 deletions
72
moto/ses/models.py
Normal file
72
moto/ses/models.py
Normal file
|
|
@ -0,0 +1,72 @@
|
|||
import md5
|
||||
|
||||
from moto.core import BaseBackend
|
||||
from .utils import get_random_message_id
|
||||
|
||||
|
||||
class Message(object):
|
||||
def __init__(self, message_id, source, subject, body, destination):
|
||||
self.id = message_id
|
||||
self.source = source
|
||||
self.subject = subject
|
||||
self.body = body
|
||||
self.destination = destination
|
||||
|
||||
|
||||
class RawMessage(object):
|
||||
def __init__(self, message_id, source, destination, raw_data):
|
||||
self.id = message_id
|
||||
self.source = source
|
||||
self.destination = destination
|
||||
self.raw_data = raw_data
|
||||
|
||||
|
||||
class SESQuota(object):
|
||||
def __init__(self, messages):
|
||||
self.messages = messages
|
||||
|
||||
@property
|
||||
def sent_past_24(self):
|
||||
return len(self.messages)
|
||||
|
||||
|
||||
class SESBackend(BaseBackend):
|
||||
def __init__(self):
|
||||
self.addresses = []
|
||||
self.sent_messages = []
|
||||
|
||||
def verify_email_identity(self, address):
|
||||
self.addresses.append(address)
|
||||
|
||||
def verify_domain(self, domain):
|
||||
self.addresses.append(domain)
|
||||
|
||||
def list_identities(self):
|
||||
return self.addresses
|
||||
|
||||
def delete_identity(self, identity):
|
||||
self.addresses.remove(identity)
|
||||
|
||||
def send_email(self, source, subject, body, destination):
|
||||
if source not in self.addresses:
|
||||
return False
|
||||
|
||||
message_id = get_random_message_id()
|
||||
message = Message(message_id, source, subject, body, destination)
|
||||
self.sent_messages.append(message)
|
||||
return message
|
||||
|
||||
def send_raw_email(self, source, destination, raw_data):
|
||||
if source not in self.addresses:
|
||||
return False
|
||||
|
||||
message_id = get_random_message_id()
|
||||
message = RawMessage(message_id, source, destination, raw_data)
|
||||
self.sent_messages.append(message)
|
||||
return message
|
||||
|
||||
def get_send_quota(self):
|
||||
return SESQuota(self.sent_messages)
|
||||
|
||||
|
||||
ses_backend = SESBackend()
|
||||
Loading…
Add table
Add a link
Reference in a new issue