28 lines
959 B
Python
28 lines
959 B
Python
|
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
|
||
|
|
})
|