From e892f6b3e2a9c71d8e4d9e02f278a38fdfb1a8b2 Mon Sep 17 00:00:00 2001 From: "d.kovalenko" Date: Sun, 4 May 2025 00:07:38 +0300 Subject: [PATCH] [test] TestTestgresLocal.test_upgrade_node is corrected Let's "release" all our test nodes correctly. --- tests/test_testgres_local.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_testgres_local.py b/tests/test_testgres_local.py index 1dd98fe..53c9e0f 100644 --- a/tests/test_testgres_local.py +++ b/tests/test_testgres_local.py @@ -158,15 +158,15 @@ def test_child_process_dies(self): def test_upgrade_node(self): old_bin_dir = os.path.dirname(get_bin_path("pg_config")) new_bin_dir = os.path.dirname(get_bin_path("pg_config")) - node_old = get_new_node(prefix='node_old', bin_dir=old_bin_dir) - node_old.init() - node_old.start() - node_old.stop() - node_new = get_new_node(prefix='node_new', bin_dir=new_bin_dir) - node_new.init(cached=False) - res = node_new.upgrade_from(old_node=node_old) - node_new.start() - assert (b'Upgrade Complete' in res) + with get_new_node(prefix='node_old', bin_dir=old_bin_dir) as node_old: + node_old.init() + node_old.start() + node_old.stop() + with get_new_node(prefix='node_new', bin_dir=new_bin_dir) as node_new: + node_new.init(cached=False) + res = node_new.upgrade_from(old_node=node_old) + node_new.start() + assert (b'Upgrade Complete' in res) class tagPortManagerProxy: sm_prev_testgres_reserve_port = None