IntenseWebs/pgadmin4
3
0
Fork 0
mirror of https://github.com/pgadmin-org/pgadmin4.git synced 2025-02-25 18:55:31 -06:00
Code Issues Packages Projects Releases Wiki Activity

More hardening of the query tool tests.

Browse Source
This commit is contained in:
Khushboo Vashi 2018-03-08 09:12:48 +00:00 committed by Dave Page
parent 1136e37cf0
commit f86c313c18
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
web/pgadmin/tools/sqleditor/utils/tests/test_start_running_query.py
View File

@ -402,6 +402,12 @@ class StartRunningQueryTest(BaseTestGenerator):
blueprint_mock = MagicMock( blueprint_mock = MagicMock(
info_notifier_timeout=MagicMock(get=lambda: 5)) info_notifier_timeout=MagicMock(get=lambda: 5))
# Save value for the later use
self.is_begin_required_for_sql_query = \
StartRunningQuery.is_begin_required_for_sql_query
self.is_rollback_statement_required = \
StartRunningQuery.is_rollback_statement_required
if self.is_begin_required: if self.is_begin_required:
StartRunningQuery.is_begin_required_for_sql_query = MagicMock( StartRunningQuery.is_begin_required_for_sql_query = MagicMock(
return_value=True return_value=True
@ -519,3 +525,10 @@ class StartRunningQueryTest(BaseTestGenerator):
self.connection.execute_void.assert_called_with('ROLLBACK;') self.connection.execute_void.assert_called_with('ROLLBACK;')
elif not self.is_begin_required: elif not self.is_begin_required:
self.connection.execute_void.assert_not_called() self.connection.execute_void.assert_not_called()
def tearDown(self):
# Reset methods to the original state
StartRunningQuery.is_rollback_statement_required = \
self.is_rollback_statement_required
StartRunningQuery.is_rollback_statement_required = \
self.is_rollback_statement_required