def main(name=None, config_path=None):
if name not in (None, '__main__'):
return
context = SigningContext()
context.config = get_default_config()
if config_path is None:
if len(sys.argv) != 2:
usage()
config_path = sys.argv[1]
context.config.update(load_json(path=config_path))
if context.config.get('verbose'):
log_level = logging.DEBUG
else:
log_level = logging.INFO
logging.basicConfig(
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
level=log_level
)
logging.getLogger("taskcluster").setLevel(logging.WARNING)
loop = asyncio.get_event_loop()
kwargs = {}
if context.config.get('ssl_cert'):
sslcontext = ssl.create_default_context(cafile=context.config['ssl_cert'])
kwargs['ssl_context'] = sslcontext
conn = aiohttp.TCPConnector(**kwargs)
with aiohttp.ClientSession(connector=conn) as session:
context.session = session
try:
loop.run_until_complete(async_main(context))
except ScriptWorkerTaskException as exc:
traceback.print_exc()
sys.exit(exc.exit_code)
loop.close()
def test_load_json_from_file(tmpdir):
json_object = {'a_key': 'a_value'}
output_file = os.path.join(tmpdir, 'file.json')
with open(output_file, 'w') as f:
json.dump(json_object, f)
assert utils.load_json(output_file) == json_object
def test_load_json_from_file(tmpdir):
json_object = {"a_key": "a_value"}
output_file = os.path.join(tmpdir, "file.json")
with open(output_file, "w") as f:
json.dump(json_object, f)
assert utils.load_json(output_file) == json_object
def test_load_json_from_file():
json_object = {'a_key': 'a_value'}
with tempfile.TemporaryDirectory() as output_dir:
output_file = os.path.join(output_dir, 'file.json')
with open(output_file, 'w') as f:
json.dump(json_object, f)
assert load_json(output_file) == json_object