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