Add bats test to cover docker-compose in serve

During #1839 review we decided that having bats test will be very
helpful. This PR adds docker-compose bats tests.

Also fixes the issue of generating compose file with static name.

Fixes: #1873
Signed-off-by: Abhishek Bongale <[email protected]>

Author: abhibongale

ramalama/cli.py

@@ -47,7 +47,7 @@
 shortnames = Shortnames()
 
 
-GENERATE_OPTIONS = ["quadlet", "kube", "quadlet/kube"]
+GENERATE_OPTIONS = ["quadlet", "kube", "quadlet/kube", "compose"]
 
 
 class ParsedGenerateInput:

ramalama/compose.py

@@ -184,7 +184,7 @@ def generate(self) -> PlainFile:
 
 def genfile(name: str, content: str) -> PlainFile:
     file_name = "docker-compose.yaml"
-    print(f"Generating Docker Compose file: {file_name}")
+    print(f"Generating Compose YAML file: {file_name}")
 
     file = PlainFile(file_name)
     file.content = content

test/system/040-serve.bats

@@ -444,6 +444,43 @@ verify_begin=".*run --rm"
     rm /tmp/$name.yaml
 }
 
+@test  "ramalama serve --generate=compose" {
+    model="smollm:135m"
+    name="docker-compose"
+    run_ramalama pull $model
+    run_ramalama serve --name=$name --port 1234 --generate=compose $model
+    is "$output" ".*Generating Compose YAML file: ${name}.yaml" "generate .yaml file"
+
+    run cat $name.yaml
+    is "$output" ".*command: .*serve.*" "Should contain serve command"
+    is "$output" ".*ports:" "Should contain ports section"
+    is "$output" ".*- \"1234:1234\"" "Should map the container port"
+
+    HIP_VISIBLE_DEVICES=99 run_ramalama serve --name=${name} --port 1234 --generate=compose $model
+    is "$output" ".*Generating Compose YAML file: ${name}.yaml" "generate .yaml file"
+
+    run cat $name.yaml
+    is "$output" ".*environment:" "Should contain environment section"
+    is "$output" ".*- HIP_VISIBLE_DEVICES=99" "Should contain the HIP_VISIBLE_DEVICES env var"
+
+    rm $name.yaml
+}
+
+@test "ramalama serve --generate=compose:/tmp" {
+    model=tiny
+    name="docker-compose"
+    run_ramalama pull ${model}
+    run_ramalama serve --name=$name --port 1234 --generate=compose:/tmp ${model}
+    is "$output" ".*Generating Compose YAML file: ${name}.yaml" "generate .yaml file in /tmp"
+
+    run cat /tmp/$name.yaml
+    is "$output" ".*command: .*serve.*" "Should contain serve command"
+    is "$output" ".*ports:" "Should contain ports section"
+    is "$output" ".*- \"1234:1234\"" "Should map the container port correctly"
+
+    rm /tmp/$name.yaml
+}
+
 @test "ramalama serve --api llama-stack" {
     skip_if_docker
     skip_if_nocontainer

test/unit/test_compose.py

@@ -200,7 +200,7 @@ def test_compose_genfile(monkeypatch):
 
     assert result.filename == "docker-compose.yaml"
     assert result.content == content
-    assert printed_output == ["Generating Docker Compose file: docker-compose.yaml"]
+    assert printed_output == ["Generating Compose YAML file: docker-compose.yaml"]
 
 
 def test_genfile_empty_content(monkeypatch):
@@ -215,7 +215,7 @@ def test_genfile_empty_content(monkeypatch):
 
     assert result.filename == "docker-compose.yaml"
     assert result.content == ""
-    assert printed_output == ["Generating Docker Compose file: docker-compose.yaml"]
+    assert printed_output == ["Generating Compose YAML file: docker-compose.yaml"]
 
 
 def test_compose_no_port_arg(monkeypatch):