[test] discovery_test.py can read key from stdin

This commit is contained in:
Ryabin Sergey
2013-11-01 12:41:35 +04:00
parent 327a576f8f
commit 794f92c47f

View File

@@ -1,6 +1,7 @@
from flask import json from flask import json
from itertools import groupby from itertools import groupby
import logging import logging
import sys
from rubick.common import MarkedIssue, Inspection from rubick.common import MarkedIssue, Inspection
from rubick.discovery import OpenstackDiscovery from rubick.discovery import OpenstackDiscovery
@@ -74,8 +75,11 @@ def main():
logging.getLogger('rubick').setLevel(logging.DEBUG) logging.getLogger('rubick').setLevel(logging.DEBUG)
discovery = OpenstackDiscovery() discovery = OpenstackDiscovery()
with open('test_rsa') as f: try:
private_key = f.read() with open('test_rsa') as f:
private_key = f.read()
except:
private_key = sys.stdin.read()
openstack = discovery.discover( openstack = discovery.discover(
['172.18.65.179'], ['172.18.65.179'],