Step 3: Implement a File Server Testing Component

Let’s dive in on how to implement a testing component for a server.

This component needs to test two things:

• The /upload_file/ endpoint by creating a file and sending its content to it.

• The / endpoint listing files, by validating the that previously uploaded file is present in the response.

class TestFileServer(LightningWork):
    def __init__(self, drive: Drive):
        super().__init__(cache_calls=True)
        self.drive = drive

    def run(self, file_server_url: str, first=True):
        if first:
            with open("test.txt", "w") as f:
                f.write("Some text.")

            response = requests.post(
                file_server_url + "/upload_file/",
                files={'file': open("test.txt", 'rb')}
            )
            assert response.status_code == 200
        else:
            response = requests.get(file_server_url)

Tutorial

Step 1: Implement the File Server general structure

Put together the shape of the Component

Basic

Step 2: Implement the File Server upload and list files methods

Add the core functionalities to the Component

Basic

Step 4: Implement tests for the File Server component with pytest

Create an App to validate the upload and list files endpoints

Intermediate