playground tests

.devcontainer/Containerfile

@@ -0,0 +1,9 @@
+FROM registry.access.redhat.com/ubi9/python-311:1-52
+
+USER root
+
+COPY playground/tests/requirements.txt .
+
+RUN pip3 install -r requirements.txt && \
+    dnf install -y podman buildah && \
+    dnf clean all

.devcontainer/devcontainer.json

@@ -0,0 +1,8 @@
+{
+    "name": "playground",
+    "build": {
+        "dockerfile": "Containerfile",
+        "context": ".."
+    },
+    "privileged": true
+}

.github/workflows/playground.yaml

@@ -0,0 +1,53 @@
+name: playground
+
+on:
+  pull_request:
+    branches:
+      - main
+  push:
+    branches:
+      - main
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository_owner	}}/playground
+
+jobs:
+  build-and-push-image:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - uses: actions/[email protected]
+
+      - name: Login to ghcr
+        uses: docker/[email protected]
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Docker meta
+        id: meta
+        uses: docker/[email protected]
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+
+      - name: Build and push
+        uses: docker/[email protected]
+        with:
+          context: playground
+          file: playground/Containerfile
+          push: ${{ github.event_name != 'pull_request' }}
+          tags: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest
+          labels: ${{ steps.meta.outputs.labels }}
+
+      - name: Set up Python
+        uses: actions/[email protected]
+
+      - run: pip install -r playground/tests/requirements.txt
+
+      - name: Run tests
+        run: pytest

.gitignore

@@ -1,3 +1,4 @@
 *.gguf
 *.bin
 *_pycache_*
+port_check.lock

playground/Containerfile

@@ -1,8 +1,7 @@
 FROM registry.access.redhat.com/ubi9/python-39:latest
 WORKDIR /locallm
 COPY requirements.txt /locallm/requirements.txt
-RUN pip install --upgrade pip
-RUN pip install --no-cache-dir --upgrade -r /locallm/requirements.txt
+RUN pip install --no-cache-dir -r /locallm/requirements.txt
 COPY run.sh run.sh
 EXPOSE 8001
 ENTRYPOINT [ "sh", "run.sh" ]

playground/Makefile

@@ -0,0 +1,3 @@
+.PHONY: build
+build:
+	buildah bud -f Containerfile --no-cache -t playground .

playground/requirements.txt

@@ -1 +1,2 @@
-llama-cpp-python[server]
+llama-cpp-python[server]==0.2.57
+pip==24.0

playground/tests/conftest.py

@@ -0,0 +1,9 @@
+import pytest_container
+
+TW = pytest_container.Container(
+    url="containers-storage:localhost/${os.environ['IMAGE_NAME']}",
+    forwarded_ports=[pytest_container.PortForwarding(container_port=8001)]
+)
+
+def pytest_generate_tests(metafunc):
+    pytest_container.auto_container_parametrize(metafunc)

playground/tests/requirements.txt

@@ -0,0 +1,4 @@
+pip==24.0
+pytest-container==0.3.0
+pytest-testinfra==10.1.0
+pytest==8.1.1

playground/tests/test_alive.py

@@ -0,0 +1,7 @@
+import pytest_container
+from .conftest import TW
+
+CONTAINER_IMAGES = [TW]
+
+def test_alive(auto_container: pytest_container.container.ContainerData, host):
+    res = host.run_expect([0],f"curl localhost:{auto_container.forwarded_ports[0].host_port}",).stdout.strip()