# 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)