TODO languages with tests in implementation files

Author: bauersimon

model/llm/llm.go

@@ -145,7 +145,7 @@ type llmWriteTestSourceFilePromptContext struct {
 
 // llmWriteTestForFilePromptTemplate is the template for generating an LLM test generation prompt.
 var llmWriteTestForFilePromptTemplate = template.Must(template.New("model-llm-write-test-for-file-prompt").Parse(bytesutil.StringTrimIndentations(`
-	Given the following {{ .Language.Name }} code file "{{ .FilePath }}" with package "{{ .ImportPath }}", provide a test file for this code{{ with .TestFramework }} with {{ . }} as a test framework{{ end }}.
+	Given the following {{ .Language.Name }} code file "{{ .FilePath }}" {{- with .ImportPath }} with package "{{ . }}" {{- end }}, provide {{- if .Language.HasTestsInSource }} tests {{ else }} a test file {{ end -}} for this code{{ with .TestFramework }} with {{ . }} as a test framework{{ end }}.
 	The tests should produce 100 percent code coverage and must compile.
 	The response must contain only the test code in a fenced code block and nothing else.
 

model/llm/llm_test.go

@@ -21,6 +21,7 @@ import (
 	"github.com/symflower/eval-dev-quality/language"
 	"github.com/symflower/eval-dev-quality/language/golang"
 	"github.com/symflower/eval-dev-quality/language/java"
+	"github.com/symflower/eval-dev-quality/language/rust"
 	"github.com/symflower/eval-dev-quality/log"
 	"github.com/symflower/eval-dev-quality/model"
 	"github.com/symflower/eval-dev-quality/provider"
@@ -654,6 +655,68 @@ func TestFormatPromptContext(t *testing.T) {
 				` + "```" + `
 			`),
 		})
+
+		validate(t, &testCase{
+			Name: "No Import path",
+
+			Context: &llmWriteTestSourceFilePromptContext{
+				llmSourceFilePromptContext: llmSourceFilePromptContext{
+					Language: &golang.Language{},
+
+					Code: bytesutil.StringTrimIndentations(`
+						package increment
+
+						func increment(i int) int
+							return i + 1
+						}
+					`),
+					FilePath:   filepath.Join("path", "to", "increment.go"),
+					ImportPath: "",
+				},
+			},
+
+			ExpectedMessage: bytesutil.StringTrimIndentations(`
+				Given the following Go code file "path/to/increment.go", provide a test file for this code.
+				The tests should produce 100 percent code coverage and must compile.
+				The response must contain only the test code in a fenced code block and nothing else.
+
+				` + "```" + `golang
+				package increment
+
+				func increment(i int) int
+					return i + 1
+				}
+				` + "```" + `
+			`),
+		})
+
+		validate(t, &testCase{
+			Name: "Tests in source file",
+
+			Context: &llmWriteTestSourceFilePromptContext{
+				llmSourceFilePromptContext: llmSourceFilePromptContext{
+					Language: &rust.Language{},
+
+					Code: bytesutil.StringTrimIndentations(`
+						fn main() {
+						}
+					`),
+					FilePath:   filepath.Join("path", "to", "main.rs"),
+					ImportPath: "",
+				},
+			},
+
+			ExpectedMessage: bytesutil.StringTrimIndentations(`
+				Given the following Rust code file "path/to/main.rs", provide tests for this code.
+				The tests should produce 100 percent code coverage and must compile.
+				The response must contain only the test code in a fenced code block and nothing else.
+
+				` + "```" + `rust
+				fn main() {
+				}
+				` + "```" + `
+			`),
+		})
 	})
 
 	validate(t, &testCase{