def main(): domain = getTestDomainName("mkgroups_perf") print("domain: {}".format(domain)) log.info("initializing") signal.signal(signal.SIGTERM, sig_handler) # add handlers for early exit signal.signal(signal.SIGINT, sig_handler) loop = asyncio.get_event_loop() globals["loop"] = loop #domain = helper.getTestDomainName() # create a client Session here so that all client requests # will share the same connection pool max_tcp_connections = int(config.get("max_tcp_connections")) client = ClientSession(loop=loop, connector=TCPConnector(limit=max_tcp_connections)) globals["client"] = client globals["group_count"] = 0 globals["grp_request_count"] = 0 globals["lnk_request_count"] = 0 globals["grp_failed_posts"] = 0 globals["lnk_failed_posts"] = 0 globals["group_target"] = config.get("group_target") max_concurrent_tasks = config.get("max_concurrent_tasks") loop.run_until_complete(getEndpoints()) if len(globals["sn_endpoints"]) == 0: log.error("no SN endpoints found!") loop.close() client.close() sys.exit() for endpoint in globals["sn_endpoints"]: log.info("got endpoint: {}".format(endpoint)) loop.run_until_complete(verifyDomain(domain)) globals["domain"] = domain # save the domain globals["start_time"] = time.time() # start making groups! while globals["grp_request_count"] < globals["group_target"]: tasks = [] count = max_concurrent_tasks if globals["group_target"] - globals["grp_request_count"] < count: count = globals["group_target"] - globals["grp_request_count"] log.info("adding {} tasks".format(count)) for i in range(count): tasks.append(asyncio.ensure_future(createGroup())) # got a batch, move them out! loop.run_until_complete(asyncio.gather(*tasks)) tasks = [] loop.close() client.close() globals["stop_time"] = time.time() print_results()
def __init__(self, *args, **kwargs): super(GroupTest, self).__init__(*args, **kwargs) self.base_domain = helper.getTestDomainName(self.__class__.__name__) helper.setupDomain(self.base_domain)
def __init__(self, *args, **kwargs): super(QueryTest, self).__init__(*args, **kwargs) self.base_domain = helper.getTestDomainName(self.__class__.__name__) helper.setupDomain(self.base_domain) self.endpoint = helper.getEndpoint()
def __init__(self, *args, **kwargs): super(DomainTest, self).__init__(*args, **kwargs) self.base_domain = helper.getTestDomainName(self.__class__.__name__) print("base_domain: {}".format(self.base_domain)) helper.setupDomain(self.base_domain, folder=True)