ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2024-12-19 20:57:54 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

tests: try to get logs

Browse Source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-06-25 09:11:46 +02:00
parent e84b31935c
commit f41a519a2c
1 changed files with 36 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
tests/e2e-aio/e2e_suite_test.go
View File

@ -1,6 +1,7 @@
package e2e_test
import (
"bytes"
"context"
"fmt"
"os"
@ -38,7 +39,7 @@ var _ = BeforeSuite(func() {
var defaultConfig openai.ClientConfig
if apiEndpoint == "" {
startDockerImage()
startDockerImage("")
defaultConfig = openai.DefaultConfig(apiKey)
apiEndpoint = "http://localhost:" + apiPort + "/v1" // So that other tests can reference this value safely.
defaultConfig.BaseURL = apiEndpoint
@ -58,9 +59,41 @@ var _ = BeforeSuite(func() {
})
var _ = AfterSuite(func() {
// if the suite failed, logs will be printed
// to the console
if CurrentGinkgoTestDescription().Failed {
if resource != nil {
logs := bytes.NewBufferString("")
err := pool.Client.Logs(docker.LogsOptions{
Container: resource.Container.ID,
OutputStream: logs,
ErrorStream: logs,
Stdout: true,
Stderr: true,
Timestamps: true,
})
if err != nil {
fmt.Println("Could not take logs for failed suite", err.Error())
}
fmt.Println("Suite failed, printing logs")
fmt.Println(logs.String())
c, err := pool.Client.InspectContainer(resource.Container.ID)
if err != nil {
fmt.Println("Could not inspect container", err.Error())
}
fmt.Println("Container state")
fmt.Println("Running:", c.State.Running)
fmt.Println("ExitCode:", c.State.ExitCode)
fmt.Println("Error:", c.State.Error)
}
}
if resource != nil {
Expect(pool.Purge(resource)).To(Succeed())
}
//dat, err := os.ReadFile(resource.Container.LogPath)
//Expect(err).To(Not(HaveOccurred()))
//Expect(string(dat)).To(ContainSubstring("GRPC Service Ready"))
@ -71,8 +104,8 @@ var _ = AfterEach(func() {
//Expect(dbClient.Clear()).To(Succeed())
})
func startDockerImage() {
p, err := dockertest.NewPool("")
func startDockerImage(endpoint string) {
p, err := dockertest.NewPool(endpoint)
Expect(err).To(Not(HaveOccurred()))
Expect(p.Client.Ping()).To(Succeed())