external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Wait until deamon starts in tests

pull/7254/head
EmelSimsek 2023-10-24 16:06:47 +03:00 committed by Emel Şimşek
parent ea4e9c32b9
commit 4836ede597
1 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/test/regress/citus_tests/test/test_maintenancedeamon.py
View File

@ -2,6 +2,27 @@
# server is restarted. A Citus Maintenance Daemon for the main_db # server is restarted. A Citus Maintenance Daemon for the main_db
# is launched. This should happen even if there is no query run # is launched. This should happen even if there is no query run
# in main_db yet. # in main_db yet.
import time
def wait_until_deamons_start(deamoncount, cluster):
i = 0
n = 0
while i < 10:
i += 1
n = cluster.coordinator.sql_value(
"SELECT count(*) FROM pg_stat_activity WHERE application_name = 'Citus Maintenance Daemon';"
)
if n == deamoncount:
break
time.sleep(0.1)
assert n == deamoncount
def test_set_maindb(cluster_factory): def test_set_maindb(cluster_factory):
cluster = cluster_factory(0) cluster = cluster_factory(0)
@ -11,12 +32,7 @@ def test_set_maindb(cluster_factory):
assert cluster.coordinator.sql_value("SHOW citus.main_db;") == "mymaindb" assert cluster.coordinator.sql_value("SHOW citus.main_db;") == "mymaindb"
assert ( wait_until_deamons_start(2, cluster)
cluster.coordinator.sql_value(
"SELECT count(*) FROM pg_stat_activity WHERE application_name = 'Citus Maintenance Daemon';"
)
== 2
)
assert ( assert (
cluster.coordinator.sql_value( cluster.coordinator.sql_value(
@ -28,11 +44,7 @@ def test_set_maindb(cluster_factory):
cluster.coordinator.configure("citus.main_db=''") cluster.coordinator.configure("citus.main_db=''")
cluster.coordinator.restart() cluster.coordinator.restart()
assert cluster.coordinator.sql_value("SHOW citus.main_db;") == "" assert cluster.coordinator.sql_value("SHOW citus.main_db;") == ""
assert (
cluster.coordinator.sql_value( wait_until_deamons_start(1, cluster)
"SELECT count(*) FROM pg_stat_activity WHERE application_name = 'Citus Maintenance Daemon';"
)
== 1
)
cluster.coordinator.cleanup_databases() cluster.coordinator.cleanup_databases()