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