Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commite1b5a51

Browse files
committed
minor improvements
1 parent2ceabc0 commite1b5a51

File tree

2 files changed

+19
-14
lines changed

2 files changed

+19
-14
lines changed

‎testgres/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def release_port(port):
3838
Free port provided by reserve_port().
3939
"""
4040

41-
bound_ports.remove(port)
41+
bound_ports.discard(port)
4242

4343

4444
defexecute_utility(args,logfile=None):

‎tests/test_simple.py

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
importlogging.config
1313

14+
fromcontextlibimportcontextmanager
1415
fromdistutils.versionimportLooseVersion
1516

1617
fromtestgresimport \
@@ -58,6 +59,15 @@ def good_properties(f):
5859
returnTrue
5960

6061

62+
@contextmanager
63+
defremoving(f):
64+
try:
65+
yieldf
66+
finally:
67+
ifos.path.isfile(f):
68+
os.remove(f)
69+
70+
6171
classSimpleTest(unittest.TestCase):
6272
deftest_custom_init(self):
6373
withget_new_node()asnode:
@@ -382,19 +392,14 @@ def test_dump(self):
382392
node1.execute(query_create)
383393

384394
# take a new dump
385-
dump=node1.dump()
386-
self.assertTrue(os.path.isfile(dump))
387-
388-
withget_new_node().init().start()asnode2:
389-
390-
# restore dump
391-
node2.restore(filename=dump)
392-
393-
res=node2.execute(query_select)
394-
self.assertListEqual(res, [(1, ), (2, )])
395-
396-
# finally, remove dump
397-
os.remove(dump)
395+
withremoving(node1.dump())asdump:
396+
withget_new_node().init().start()asnode2:
397+
# restore dump
398+
self.assertTrue(os.path.isfile(dump))
399+
node2.restore(filename=dump)
400+
401+
res=node2.execute(query_select)
402+
self.assertListEqual(res, [(1, ), (2, )])
398403

399404
deftest_users(self):
400405
withget_new_node().init().start()asnode:

0 commit comments

Comments
 (0)

[8]ページ先頭

©2009-2025 Movatter.jp