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