Skip to content

Commit

Permalink
add test files
Browse files Browse the repository at this point in the history
  • Loading branch information
similato87 committed May 2, 2024
1 parent a0794bf commit 1bb4887
Show file tree
Hide file tree
Showing 3 changed files with 38 additions and 0 deletions.
13 changes: 13 additions & 0 deletions tests/core/improve_function_test_cases/theo_case_chat
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
This script checks if it is being run as the main module and then starts the Uvicorn server with the FastAPI app. The server listens on all interfaces (`0.0.0.0`) and port `8000`.

Additionally, we need to update the Dockerfile to ensure that the new script is correctly utilized when the Docker container is run. We will change the CMD directive to use `run.py` instead of directly calling `uvicorn`.

```diff
--- dockerfile
+++ dockerfile
@@ -19,3 +19,3 @@
-# Run app.py when the container launches
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
+# Run run.py when the container launches
+CMD ["python", "run.py"]
```
20 changes: 20 additions & 0 deletions tests/core/improve_function_test_cases/theo_case_code
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
# Use an official Python runtime as a parent image
FROM python:3.9-slim

# Set the working directory in the container
WORKDIR /usr/src/app

# Copy the current directory contents into the container at /usr/src/app
COPY . /usr/src/app

# Install any needed packages specified in requirements.txt
RUN pip install --no-cache-dir -r requirements.txt

# Make port 80 available to the world outside this container
EXPOSE 80

# Define environment variable
ENV NAME World

# Run app.py when the container launches
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "80"]
5 changes: 5 additions & 0 deletions tests/core/test_salvage_correct_hunks.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,11 @@ def test_create_two_new_files():
)


def test_theo_case():
files = FilesDict({"dockerfile": get_file_content("theo_case_code")})
salvage_correct_hunks(message_builder("theo_case_chat"), files, [], memory)


def test_clean_up_folder(clean_up_folder):
# The folder should be deleted after the test is run
assert True
Expand Down

0 comments on commit 1bb4887

Please sign in to comment.