34 lines
716 B
Python
34 lines
716 B
Python
import pytest
|
|
from fasthtml.fastapp import fast_app
|
|
|
|
from myfasthtml.auth.utils import create_auth_beforeware
|
|
from myfasthtml.core.testclient import MyTestClient
|
|
|
|
def test_non_protected_route():
|
|
app, rt = fast_app()
|
|
user = MyTestClient(app)
|
|
|
|
@rt('/')
|
|
def index(): return "Welcome"
|
|
|
|
@rt('/login')
|
|
def index(): return "Sign In"
|
|
|
|
user.open("/")
|
|
user.should_see("Welcome")
|
|
|
|
|
|
def test_all_routes_are_protected():
|
|
beforeware = create_auth_beforeware()
|
|
app, rt = fast_app(before=beforeware)
|
|
user = MyTestClient(app)
|
|
|
|
@rt('/')
|
|
def index(): return "Welcome"
|
|
|
|
@rt('/login')
|
|
def index(): return "Sign In"
|
|
|
|
user.open("/")
|
|
user.should_see("Sign In")
|