Updating tests

This commit is contained in:
zane 2018-06-09 23:42:33 -07:00
commit b1a8ae8299
3 changed files with 12 additions and 11 deletions

View file

@ -1,6 +1,6 @@
from __future__ import unicode_literals
import datetime
import time
import json
import boto3
@ -13,26 +13,27 @@ class SecretsManager(BaseModel):
def __init__(self, region_name, **kwargs):
self.secret_id = kwargs.get('secret_id', '')
self.version_id = kwargs.get('version_id', '')
self.string_id = kwargs.get('string_id', '')
self.version_stage = kwargs.get('version_stage', '')
class SecretsManagerBackend(BaseBackend):
def __init__(self, region_name=None):
def __init__(self, region_name=None, **kwargs):
super(SecretsManagerBackend, self).__init__()
self.region = region_name
self.secret_id = kwargs.get('secret_id', '')
self.createdate = int(time.time())
def get_secret_value(self, secret_id, version_id, string_id):
def get_secret_value(self, secret_id, version_id, version_stage):
response = json.dumps({
"ARN": self.secret_arn(),
"Name": self.secret_id,
"VersionId": "A435958A-D821-4193-B719-B7769357AER4",
"SecretBinary": b"testbytes",
"SecretString": "mysecretstring",
"VersionStages": [
"AWSCURRENT",
],
"CreatedDate": datetime.datetime.utcnow()
"CreatedDate": "2018-05-23 13:16:57.198000"
})
return response