42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
from flask.testing import FlaskClient
|
||
|
|
from tests.base_test import TestBase
|
||
|
|
|
||
|
|
|
||
|
|
class TestPages(TestBase):
|
||
|
|
""" Test class that encapsulates tests for
|
||
|
|
the main pages on the site
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_homepage(self, test_client: FlaskClient):
|
||
|
|
""" Tests that the main homepage loads correctly
|
||
|
|
once the '/' endpoint is hit
|
||
|
|
"""
|
||
|
|
response = test_client.get('/')
|
||
|
|
assert response.status_code == 302
|
||
|
|
|
||
|
|
response = test_client.get('/products')
|
||
|
|
assert response.status_code == 308
|
||
|
|
|
||
|
|
def test_products(self, test_client: FlaskClient):
|
||
|
|
""" Tests that a product page is displayed when
|
||
|
|
hitting one of the product endpoints
|
||
|
|
"""
|
||
|
|
response = test_client.get('/products/2')
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
response = test_client.get('/products/50')
|
||
|
|
assert response.status_code == 302
|
||
|
|
|
||
|
|
response = test_client.get('/products/Books')
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
def test_admin(self, test_client: FlaskClient):
|
||
|
|
""" Tests that the admin pages can be reached and redirect
|
||
|
|
upon reaching
|
||
|
|
"""
|
||
|
|
response = test_client.get('/admin/users/')
|
||
|
|
assert response.status_code == 302
|
||
|
|
|
||
|
|
response = test_client.get('/admin/products/')
|
||
|
|
assert response.status_code == 302
|