tests.test_auth

 1import pytest
 2from hanabiapp.models.user import User
 3from flask_login import current_user
 4
 5INVALID_MESSAGE = 'Invalid username or password'
 6
 7def test_valid_login(client, auth):
 8    """
 9    正しいユーザー情報でログインできることをテストする関数．
10
11    - `/login` ページが正常に取得できることを確認．
12    - `auth.login()` を実行し，ログイン後に `/home` へリダイレクトされることを確認．
13    - `/home` にアクセスした際，認証済みユーザーであることを確認．
14    """
15    assert client.get('/login').status_code == 200
16    response = auth.login()
17    assert response.headers['Location'] == '/home'
18    with client:
19        response = client.get('/home', follow_redirects=True)
20        assert response.status_code == 200
21        assert current_user.is_authenticated
22        assert current_user.username == 'test-user'
23
24@pytest.mark.parametrize(('username', 'password', 'message'), [
25    ('wrong-user', 'test-password', INVALID_MESSAGE),
26    ('test-user', 'wrong-password', INVALID_MESSAGE),
27    ('', 'test-password', INVALID_MESSAGE),
28    ('test-user', '', INVALID_MESSAGE),
29    ('', '', INVALID_MESSAGE)
30])
31def test_invalid_login(auth, username, password, message):
32    """
33    無効なユーザー情報でログインした際にエラーメッセージが表示されることをテストする関数．
34
35    Args:
36        auth (AuthActions): 認証処理を管理するテスト用オブジェクト
37        username (str): テストするユーザー名
38        password (str): テストするパスワード
39        message (str): 期待するエラーメッセージ
40    """
41    response = auth.login(username, password)
42    assert response.status_code == 401
43    assert message in response.data.decode('utf-8')
44
45def test_authenticated_user_login(client, auth):
46    """
47    すでにログイン済みのユーザーが `/login` にアクセスした際に `/home` にリダイレクトされることをテストする関数．
48    """
49    auth.login()
50    response = client.get('/login')
51    assert response.status_code == 302
52    assert response.headers['Location'] == '/home'
53
54def test_logout(client, auth):
55    """
56    ユーザーが正常にログアウトできることをテストする関数．
57
58    - ログイン後，ログアウトを実行し，ユーザーが認証解除されることを確認．
59    """
60    auth.login()
61    with client:
62        auth.logout()
63        assert not current_user.is_authenticated
64
65def test_logout_without_login(client):
66    """
67    未ログイン状態で `/logout` にアクセスした場合に `/login` にリダイレクトされることをテストする関数．
68    """
69    response = client.get('/logout')
70    assert response.status_code == 302
71    assert response.headers['Location'] == '/login'
72
73@pytest.mark.parametrize('endpoint', [
74    '/home',
75    '/consent'
76])
77def test_login_reqired_page(client, auth, endpoint):
78    """
79    認証が必要なページに未ログイン状態でアクセスした際に `/login` にリダイレクトされることをテストする関数．
80
81    - ログインなしで `endpoint` にアクセスすると `/login` にリダイレクトされることを確認．
82    - ログイン後に `endpoint` へアクセスすると正常に表示されることを確認．
83
84    Args:
85        client (FlaskClient): テスト用のクライアント
86        auth (AuthActions): 認証処理を管理するテスト用オブジェクト
87        endpoint (str): テスト対象のエンドポイント
88    """
89    response = client.get(endpoint)
90    assert response.status_code == 302
91    assert response.headers['Location'].startswith('/login')
92    assert 'next=' in response.headers['Location']
93    
94    auth.login()
95    response = client.get(endpoint)
96    assert response.status_code == 200