auth_test.py

import sqlite3
import unittest
from unittest import mock

from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop

import realtimechess


class TestAuthDebug(AioHTTPTestCase):
	async def get_application(self):
		return realtimechess.make_app(True)

	@unittest_run_loop
	async def test_authenticated_checks_logged_in(self):
		response = await self.client.post("/newgame", data={})
		assert response.status == 403

	@unittest_run_loop
	async def test_login_needs_name(self):
		response = await self.client.post("/anonymous_login", data={})
		assert response.status == 400
		assert "Need name" in await response.text()

	@unittest_run_loop
	async def test_login_invalid_name(self):
		response = await self.client.post(
		    "/anonymous_login", data={"name": "as-d%!¤"})
		assert response.status == 400
		assert "Invalid name" in await response.text()

	@unittest_run_loop
	async def test_valid_password_but_user_not_present(self):
		cookies = {}
		cookies["name"] = "Petter"
		cookies["password"] = self.app["user_manager"]._password("Petter")
		self.client.session.cookie_jar.update_cookies(cookies)
		response = await self.client.get("/")
		assert response.status == 200
		assert "/?g=" in str(response.url)

	@unittest_run_loop
	async def test_invalid_password(self):
		cookies = {}
		cookies["name"] = "Petter2"
		cookies["password"] = "xyz"
		self.client.session.cookie_jar.update_cookies(cookies)
		response = await self.client.get("/")
		assert response.status == 200
		# Redirected to login page.
		assert "/loginpage" in str(response.url)

	@unittest_run_loop
	async def test_user_already_exists(self):
		data = {"name": "Petter3"}
		response = await self.client.post("/anonymous_login", data=data)
		assert response.status == 200

		self.client.session.cookie_jar.clear()
		response = await self.client.post("/anonymous_login", data=data)
		assert response.status == 200


class TestAuthNoDebug(AioHTTPTestCase):
	async def get_application(self):
		real_sqlite3_connect = sqlite3.connect
		with mock.patch("sqlite3.connect", autospec=True) as mock_connect:
			self.mock_connect = mock_connect
			self.mock_connect.return_value = real_sqlite3_connect(":memory:")
			return realtimechess.make_app(False)

	@unittest_run_loop
	async def test_real_db(self):
		self.mock_connect.assert_called_with("auth.db")

	@unittest_run_loop
	async def test_setdebug_not_present(self):
		response = await self.client.post("/setdebug", data={})
		assert response.status == 404

	@unittest_run_loop
	async def test_user_already_exists(self):
		data = {"name": "Petter3"}
		response = await self.client.post("/anonymous_login", data=data)
		assert response.status == 200

		self.client.session.cookie_jar.clear()
		response = await self.client.post("/anonymous_login", data=data)
		assert response.status == 401


if __name__ == '__main__':
	unittest.main()