def setUp(self):
self.address = os.getenv('DEVO_SENDER_SERVER', "0.0.0.0")
self.port = int(os.getenv('DEVO_SENDER_PORT', 4488))
self.tcp_address = os.getenv('DEVO_SENDER_TCP_SERVER', "0.0.0.0")
self.tcp_port = int(os.getenv('DEVO_SENDER_TCP_PORT', 4489))
self.key = os.getenv('DEVO_SENDER_KEY', CLIENT_KEY)
self.cert = os.getenv('DEVO_SENDER_CERT', CLIENT_CERT)
self.chain = os.getenv('DEVO_SENDER_CHAIN', CLIENT_CHAIN)
self.local_key = os.getenv(CLIENT_KEY)
self.test_tcp = os.getenv('DEVO_TEST_TCP', "True")
self.my_app = 'test.drop.free'
self.my_bapp = b'test.drop.free'
self.my_date = 'my.date.test.sender'
self.test_file = "".join((os.path.dirname(os.path.abspath(__file__)),
os.sep, "testfile_multiline.txt"))
self.test_msg = 'Test send msg\n'
self.localhost = socket.gethostname()
# change this value if you want to send another number of test string
self.default_numbers_sendings = 10
configuration = Configuration()
configuration.set(
"sender", {
"key": self.key,
"cert": self.cert,
"chain": self.chain,
"address": self.address,
"port": self.port,
})
self.config_path = "/tmp/devo_sender_tests_config.json"
configuration.save(path=self.config_path)
def configure(args):
"""
Load CLI configuration
:param args: args from files, launch vars, etc
:return: Client API Object and Config values in array
"""
config = Configuration()
try:
if args.get('config'):
config.load_config(args.get('config'), 'api')
if args.get('env'):
config.set("key", os.environ.get('DEVO_API_KEY', None))
config.set("secret", os.environ.get('DEVO_API_SECRET', None))
config.set("url", os.environ.get('DEVO_API_URL', None))
config.set("user", os.environ.get('DEVO_API_USER', None))
config.set("comment", os.environ.get('DEVO_API_COMMENT', None))
if args.get('default'):
config.load_default_config(section="api")
finally:
config.mix(dict(args))
conf = config.get()
# Try to compose the api
api = None
try:
api = Client.from_config(conf)
except DevoClientException as error:
print_error(str(error), show_help=True)
return api, conf
def setUp(self):
self.query = 'from demo.ecommerce.data select * limit 1'
self.app_name = "testing-app_name"
self.uri = os.getenv('DEVO_API_ADDRESS',
'https://apiv2-us.devo.com/search/query')
self.key = os.getenv('DEVO_API_KEY', None)
self.secret = os.getenv('DEVO_API_SECRET', None)
self.token = os.getenv('DEVO_AUTH_TOKEN', None)
self.query_id = os.getenv('DEVO_API_QUERYID', None)
self.user = os.getenv('DEVO_API_USER', "python-sdk-user")
self.comment = os.getenv('DEVO_API_COMMENT', None)
configuration = Configuration()
configuration.set(
"api", {
"query": self.query,
"address": self.uri,
"key": self.key,
"secret": self.secret,
"token": self.token,
"query_id": self.query_id,
"user": self.user,
"comment": self.comment,
"app_name": self.app_name
})
self.config_path = "/tmp/devo_api_tests_config.json"
configuration.save(path=self.config_path)
def init_conf(args):
""" Generic configuration of CLI, from config file and cli arguments """
config = Configuration()
try:
if args.get('config'):
config.load_config(args.get('config'), 'sender')
if args.get('env'):
config.set("address",
os.environ.get('DEVO_SENDER_ADDRESS',
os.environ.get('DEVO_SENDER_URL', None)))
config.set("port", os.environ.get('DEVO_SENDER_PORT', None))
config.set("key", os.environ.get('DEVO_SENDER_KEY', None))
config.set("cert", os.environ.get('DEVO_SENDER_CERT', None))
config.set("chain", os.environ.get('DEVO_SENDER_CHAIN', None))
if args.get('default'):
config.load_default_config(section="sender")
finally:
config.mix(dict(args))
return config
def configure(args):
"""
Load CLI configuration
:param args: args from files, launch vars, etc
:return: Client API Object and Config values in array
"""
config = Configuration()
if args.get('config') != "~/.devo.json":
config.load_json(args.get('config'), 'api')
config.mix(dict(args))
if "key" not in args.keys() and "api" not in args.keys() \
and "token" not in args.keys():
config.set("key", os.environ.get('DEVO_API_KEY', None))
config.set("secret", os.environ.get('DEVO_API_SECRET', None))
if "url" not in args.keys():
config.set("url", os.environ.get('DEVO_API_URL', None))
if not config.keys("key") and not config.keys("api") \
and not config.keys("token") \
and os.path.exists("~/.devo.json"):
config.load_default_json('api')
config.keys('from')
config.keys('to')
# Try to compose the api
api = None
try:
api = Client.from_config(config.get())
except DevoClientException as error:
print_error(str(error), show_help=True)
return api, config.get()
def configure(args):
"""
Load CLI configuration
:param args: args from files, launch vars, etc
:return: Clien t API Object and Config values in array
"""
config = Configuration()
try:
if args.get('config'):
config.load_config(args.get('config'), 'api')
if args.get('env'):
config.set("key", os.environ.get('DEVO_API_KEY', None))
config.set("secret", os.environ.get('DEVO_API_SECRET', None))
config.set("address", os.environ.get('DEVO_API_ADDRESS', None))
config.set("user", os.environ.get('DEVO_API_USER', None))
config.set("comment", os.environ.get('DEVO_API_COMMENT', None))
if args.get('default'):
config.load_default_config(section="api")
except Exception as error:
print_error(str(error), show_help=True)
finally:
config.mix(dict(args))
# Try to compose the api
try:
api = Client(config=config)
except DevoClientException as error:
print_error(str(error), show_help=True)
if isinstance(error, DevoClientException):
raise DevoClientException(error.args[0])
else:
raise DevoClientException(str(error.args[0]))
return api, config
def init_conf(args):
""" Generic configuration of CLI, from config file and cli arguments """
config = Configuration()
if args.get('config'):
config.load_json(args.get('config'), 'sender')
config.mix(dict(args))
if "address" not in args.keys() and "url" in args.keys():
config.set('address', args['url'])
if "address" not in args.keys():
address = os.environ.get('DEVO_SENDER_ADDRESS', None)
if not address:
address = os.environ.get('DEVO_SENDER_URL', None)
config.set("address", address)
config.set("port", os.environ.get('DEVO_SENDER_PORT', None))
if config.get()['cert_reqs']:
if "config" not in args.keys() and "key" not in args.keys() \
and "cert" not in args.keys() and "chain" not in args.keys():
config.set("key", os.environ.get('DEVO_SENDER_KEY', None))
config.set("cert", os.environ.get('DEVO_SENDER_CERT', None))
config.set("chain", os.environ.get('DEVO_SENDER_CHAIN', None))
if not config.keys("key") and not config.keys("cert") and not \
config.keys("chain") and os.path.exists("~/.devo.json"):
config.load_default_json('api')
config.keys('from')
config.keys('to')
return config
def test_add_key_chain(self):
config = Configuration(self.config_path + ".yaml")
config.set(["devo", "old", "name"], "logtrust")
self.assertEqual(config["devo"]['old']['name'], "logtrust")
self.assertEqual(config.get("devo", 'old', 'name'), "logtrust")
def test_add_key(self):
config = Configuration(self.config_path + ".yaml")
config.set("logtrust", "old")
self.assertEqual(config["logtrust"], "old")
