def test_query_member(self): with patch('patroni.ctl.get_cursor', Mock(return_value=MockConnect().cursor())): rows = query_member(None, None, None, 'master', 'SELECT pg_catalog.pg_is_in_recovery()', {}) self.assertTrue('False' in str(rows)) with patch.object(MockCursor, 'execute', Mock(side_effect=OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_catalog.pg_is_in_recovery()', {}) with patch('patroni.ctl.get_cursor', Mock(return_value=None)): rows = query_member(None, None, None, None, 'SELECT pg_catalog.pg_is_in_recovery()', {}) self.assertTrue('No connection to' in str(rows)) rows = query_member(None, None, None, 'replica', 'SELECT pg_catalog.pg_is_in_recovery()', {}) self.assertTrue('No connection to' in str(rows)) with patch('patroni.ctl.get_cursor', Mock(side_effect=OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_catalog.pg_is_in_recovery()', {})
def test_query_member(self): rows = query_member(None, None, None, 'master', 'SELECT pg_is_in_recovery()') assert 'False' in str(rows) rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') assert rows == (None, None) with patch('patroni.ctl.get_cursor', Mock(return_value=None)): rows = query_member(None, None, None, None, 'SELECT pg_is_in_recovery()') assert 'No connection to' in str(rows) rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') assert 'No connection to' in str(rows) with patch('patroni.ctl.get_cursor', Mock(side_effect=psycopg2.OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') with patch('test_postgresql.MockCursor.execute', Mock(side_effect=psycopg2.OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()')
def test_query_member(self): rows = query_member(None, None, None, "master", "SELECT pg_is_in_recovery()") assert "False" in str(rows) rows = query_member(None, None, None, "replica", "SELECT pg_is_in_recovery()") assert rows == (None, None) with patch("patroni.ctl.get_cursor", Mock(return_value=None)): rows = query_member(None, None, None, None, "SELECT pg_is_in_recovery()") assert "No connection to" in str(rows) rows = query_member(None, None, None, "replica", "SELECT pg_is_in_recovery()") assert "No connection to" in str(rows) with patch("patroni.ctl.get_cursor", Mock(side_effect=psycopg2.OperationalError("bla"))): rows = query_member(None, None, None, "replica", "SELECT pg_is_in_recovery()") with patch("test_postgresql.MockCursor.execute", Mock(side_effect=psycopg2.OperationalError("bla"))): rows = query_member(None, None, None, "replica", "SELECT pg_is_in_recovery()")
def test_query_member(self): with patch('patroni.ctl.get_cursor', Mock(return_value=MockConnect().cursor())): rows = query_member(None, None, None, 'master', 'SELECT pg_is_in_recovery()') self.assertTrue('False' in str(rows)) rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') self.assertEquals(rows, (None, None)) with patch('test_postgresql.MockCursor.execute', Mock(side_effect=OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') with patch('patroni.ctl.get_cursor', Mock(return_value=None)): rows = query_member(None, None, None, None, 'SELECT pg_is_in_recovery()') self.assertTrue('No connection to' in str(rows)) rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()') self.assertTrue('No connection to' in str(rows)) with patch('patroni.ctl.get_cursor', Mock(side_effect=OperationalError('bla'))): rows = query_member(None, None, None, 'replica', 'SELECT pg_is_in_recovery()')