Testing Authenticated Endpoints

Welcome to the last stop of our journey through automated API testing with Python. In this lesson, we'll focus on testing authenticated API endpoints. While you've already learned to organize tests and handle CRUD operations using pytest, today we'll delve into how APIs manage secure access through different authentication methods — API Keys, Sessions, and JWT (JSON Web Tokens).

Authentication is crucial for securing API endpoints against unauthorized access. By understanding how to test these mechanisms, you'll ensure that your API maintains its integrity and protects sensitive data. We'll explore practical examples for each authentication method, giving you the tools to verify that only authorized users can interact with protected resources.

API Key authentication is one of the simplest ways to secure an API. It involves sending a unique key as part of the request headers, allowing access to protected endpoints. Let's look at an example of how you can set up and test API Key authentication using the following code:

import pytest
import requests

BASE_URL = "http://localhost:8000"
API_KEY = "123e4567-e89b-12d3-a456-426614174000"

class TestAPIKeyAuthentication:

    def test_API_KEY_authentication(self):
        # Arrange
        headers = {"X-API-Key": API_KEY}
        
        # Act
        response = requests.get(f"{BASE_URL}/todos", headers=headers)

        # Assert
        assert response.status_code == 200

Here, we define a test class TestAPIKeyAuthentication. Within the test method, test_API_KEY_authentication, we specify the headers, including the X-API-Key. We use the requests.get method to make a call to the /todos endpoint and pass the headers, ensuring the API key is included in the request. The test then asserts that the response status code is 200, indicating successful authentication and access to the endpoint.

We begin our session-based authentication by establishing a fixture for user credentials and a helper method for login. This enables each test to authenticate independently, providing a consistent environment to test both login functionality and access to protected endpoints:

import pytest
import requests

BASE_URL = "http://localhost:8000"

class TestSessionAuthentication:

    @pytest.fixture
    def auth_details(self):
        # Arrange - Providing user credentials
        return {
            "username": "testuser",
            "password": "testpass123"
        }

    # Helper method for sending a login POST request
    def login(self, session, auth_details):
        # Sends a login request and maintains session state
        return session.post(f"{BASE_URL}/auth/login", json=auth_details)

The auth_details fixture supplies the required credentials, and the login helper method facilitates independent authentication for each test.

This test leverages the login helper method to authenticate independently, validating that user credentials create a session successfully:

def test_login(self, auth_details):
    # Arrange - Initialize session
    session = requests.Session()

    # Act - Execute the login process using the helper method
    login_response = self.login(session, auth_details)
    
    # Assert - Verify that the login was successful
    assert login_response.status_code == 200

The use of the helper ensures that this test does not rely on any previous operations, providing isolated verification of login success via status code.

In this test, we independently authenticate using the helper method before confirming that an established session grants access to a protected endpoint:

def test_access_with_session(self, auth_details):
    # Arrange - Use the login helper method to authenticate
    session = requests.Session()
    self.login(session, auth_details)

    # Act - Request the protected resource using the session
    response = session.get(f"{BASE_URL}/todos")

    # Assert - Confirm access is granted with a successful status code
    assert response.status_code == 200

The helper method's strategic use ensures that each test individually verifies session-based access, unaffected by results of prior tests.

By using the login helper method, this test independently authenticates and then verifies the ability to terminate a session, ensuring logout functionality is effective:

def test_logout_with_session(self, auth_details):
    # Arrange - Use the login helper method to authenticate
    session = requests.Session()
    self.login(session, auth_details)
    
    # Act - Perform a logout operation to terminate the session
    logout_response = session.post(f"{BASE_URL}/auth/logout")

    # Assert - Verify logout success via status code
    assert logout_response.status_code == 200

This approach confirms isolation, as each session is independently managed, ensuring logout integrity without relying on previous tests.

For JWT-based authentication, we replicate this strategy with a fixture for credentials and a login helper method, enabling each test to independently obtain and manage JWTs:

import pytest
import requests

BASE_URL = "http://localhost:8000"

class TestJWTAuthentication:

    @pytest.fixture
    def auth_details(self):
        # Arrange - User credentials for JWT
        return {
            "username": "testuser",
            "password": "testpass123"
        }

    # Helper method for sending a login POST request
    def login(self, auth_details):
        # Obtain JWT tokens with login request
        return requests.post(f"{BASE_URL}/auth/login", json=auth_details)

This setup empowers each test to authenticate independently, retrieving tokens without relying on the order or success of other tests.

Using the login helper method, this test independently verifies that successful authentication results in token issuance, forming the basis for accessing protected resources:

def test_login(self, auth_details):
    # Act - Authenticate and retrieve JWT tokens
    login_response = self.login(auth_details)
    
    # Assert - Confirm receipt of tokens and check their existence
    assert login_response.status_code == 200
    tokens = login_response.json()
    assert "access_token" in tokens
    assert "refresh_token" in tokens

By independently logging in, the test ensures token validity regardless of prior tests, establishing the foundation for secure access.

This test independently authenticates using the helper method and verifies that a JWT access token provides entry to a protected endpoint:

def test_access_with_jwt(self, auth_details):
    # Arrange - Authenticate and obtain an access token using the helper method
    login_response = self.login(auth_details)
    access_token = login_response.json()['access_token']

    # Act - Use the access token to request a protected endpoint
    headers = {"Authorization": f"Bearer {access_token}"}
    response = requests.get(f"{BASE_URL}/todos", headers=headers)

    # Assert - Confirm the request was successful by checking the status code
    assert response.status_code == 200

By ensuring the test is independent, this approach validates the token's effectiveness in protecting resources, confirming stateless authentication.

Finally, this test independently authenticates to verify that access and refresh tokens can be invalidated, securing the system against unauthorized reuse:

def test_jwt_logout(self, auth_details):
    # Arrange - Authenticate and prepare tokens using the helper method
    login_response = self.login(auth_details)
    access_token = login_response.json()['access_token']
    refresh_token = login_response.json()['refresh_token']
    
    headers = {"Authorization": f"Bearer {access_token}"}

    # Act - Use access and refresh tokens to process logout
    logout_response = requests.post(f"{BASE_URL}/auth/logout", headers=headers, json={"refresh_token": refresh_token})

    # Assert - Verify logout success through confirmation of status code
    assert logout_response.status_code == 200

This test confirms the secure handling and invalidation of tokens, ensuring each test independently checks the integrity of the logout process.

Throughout this lesson, we've explored different methods of API authentication: API Keys, Sessions, and JWTs. Each method has distinct ways to test and secure access, and you've seen how to implement tests for them effectively.

When testing any form of authentication, it’s crucial to handle API credentials with care. Avoid hardcoding sensitive information in your codebase, and consider using environment variables or secure vaults in production environments.

As we conclude this lesson and the course, take pride in the journey you've completed. You've gained a comprehensive understanding of how to automate API tests using Python, mastering basic requests through to secure, authenticated endpoints. Continue to practice and explore, applying these skills to ensure robust, reliable APIs in your projects. Congratulations on reaching the end of this course!