1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
import psycopg2
from unittest.mock import patch
from odoo.tests import HttpCase
class TestWebController(HttpCase):
def test_health(self):
response = self.url_open('/web/health')
self.assertEqual(response.status_code, 200)
payload = response.json()
self.assertEqual(payload['status'], 'pass')
self.assertFalse(response.cookies.get('session_id'))
def test_health_db_server_status(self):
response = self.url_open('/web/health?db_server_status=1')
self.assertEqual(response.status_code, 200)
payload = response.json()
self.assertEqual(payload['status'], 'pass')
self.assertEqual(payload['db_server_status'], True)
self.assertFalse(response.cookies.get('session_id'))
def _raise_psycopg2_error(*args):
raise psycopg2.Error('boom')
with patch('odoo.sql_db.db_connect', new=_raise_psycopg2_error):
response = self.url_open('/web/health?db_server_status=1')
self.assertEqual(response.status_code, 500)
payload = response.json()
self.assertEqual(payload['status'], 'fail')
self.assertEqual(payload['db_server_status'], False)
|
