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)
