from aiohttp import BasicAuth
from aiohttp.test_utils import unittest_run_loop

from actioner.settings import BASIC_AUTH
from tests.test_web import BaseWebTestCase


class HealthcheckTestCase(BaseWebTestCase):
    @unittest_run_loop
    async def test_heartbeat_requires_auth(self):
        response = await self.client.get("/healthcheck")
        self.assertEqual(response.status, 401)
        response = await self.client.get("/healthcheck", headers={
            'Authorization': BasicAuth(*BASIC_AUTH).encode()
        })
        self.assertEqual(response.status, 200)

    @unittest_run_loop
    async def test_heartbeat_response(self):
        response = await self.client.get("/healthcheck", headers={
            'Authorization': BasicAuth(*BASIC_AUTH).encode()
        })
        self.assertEqual(response.status, 200)
        self.assertEqual(await response.json(), {
            'github': 'RealOrangeOne',
            'todoist': 7471233
        })