gitops/patroni
1
0
Fork 0
mirror of https://github.com/outbackdingo/patroni.git synced 2026-08-26 15:40:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increase test coverage for aws script.

Browse Source
This commit is contained in:
Oleksii Kliukin
2015-07-27 13:04:45 +02:00
parent b92bfb34a5
commit d85451efae
2 changed files with 17 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/aws.py
+6 -18
View File
@@ -10,16 +10,9 @@ logger = logging.getLogger(__name__)
class AWSConnection:
def __init__(self, config):
def __init__(self, cluster_name):
self.available = False
self.config = config
if 'cluster_name' in config:
self.cluster_name = config.get('cluster_name')
elif 'etcd' in config and isinstance(config['etcd'], dict):
self.cluster_name = config['etcd'].get('scope', 'unknown')
else:
self.cluster_name = 'unknown'
self.cluster_name = cluster_name if cluster_name is not None else 'unknown'
try:
# get the instance id
r = requests.get('http://169.254.169.254/latest/dynamic/instance-identity/document', timeout=0.1)
@@ -73,12 +66,7 @@ class AWSConnection:
if __name__ == '__main__':
if len(sys.argv) != 4:
print ("Usage: {0} action role name".format(sys.argv[0]))
sys.exit(1)
action, role, name = sys.argv[1:]
if action in ('on_start', 'on_stop', 'on_role_change'):
aws = AWSConnection({'cluster_name': name})
aws.on_role_change(role)
sys.exit(0)
sys.exit(2)
if len(sys.argv) != 4 and sys.argv[1] in ('on_start', 'on_stop', 'on_role_change'):
AWSConnection(cluster_name=sys.argv[3]).on_role_change(sys.argv[2])
else:
sys.exit("Usage: {0} action role name".format(sys.argv[0]))