def init_worker(param_queue, result_queue):
global result
# Make sure the generator is re-seeded, as we have inherited
# the seed from the parent process.
random.seed()
result = ChannelingTestResult(result_queue)
if not param_queue.empty():
server_addr = param_queue.get()
cluster = edgedb_cluster.RunningCluster(**server_addr)
tb._set_default_cluster(cluster)
def run(self, result):
server_addr = next(iter(self.server_conns))
cluster = edgedb_cluster.RunningCluster(**server_addr)
tb._set_default_cluster(cluster)
suite = StreamingTestSuite()
for test in self.tests:
suite.run(test, result)
if self.stop_requested:
break
return result
def _start_cluster(*, cleanup_atexit=True):
global _default_cluster
if _default_cluster is None:
cluster_addr = os.environ.get('EDGEDB_TEST_CLUSTER_ADDR')
if cluster_addr:
conn_spec = json.loads(cluster_addr)
_default_cluster = edgedb_cluster.RunningCluster(**conn_spec)
else:
data_dir = os.environ.get('EDGEDB_TEST_DATA_DIR')
_default_cluster = _init_cluster(data_dir=data_dir,
cleanup_atexit=cleanup_atexit)
return _default_cluster
def run(self, result_):
global result
result = result_
if self.server_conn:
cluster = edgedb_cluster.RunningCluster(**self.server_conn)
tb._set_default_cluster(cluster)
for test in self.tests:
_run_test(test)
if self.stop_requested:
break
finalize_result_testcases()
return result
def init_worker(param_queue, result_queue):
global result
# Make sure the generator is re-seeded, as we have inherited
# the seed from the parent process.
random.seed()
result = ChannelingTestResult(result_queue)
if not param_queue.empty():
server_addr = param_queue.get()
if server_addr is not None:
cluster = edgedb_cluster.RunningCluster(**server_addr)
tb._set_default_cluster(cluster)
multiprocessing.util.Finalize(suite_cache, finalize_result_testcases,
exitpriority=100)
