mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
Joe Elliottandzeekay
d09d989270
* remove ssb target and move all-3.0 to all Signed-off-by: Joe Elliott <[email protected]> * changelog Signed-off-by: Joe Elliott <[email protected]> * wait a little longer Signed-off-by: Joe Elliott <[email protected]> --------- Signed-off-by: Joe Elliott <[email protected]>
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/tempo/pkg/util"
|
|
"github.com/grafana/tempo/tempodb/backend"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestApp_RunStop(t *testing.T) {
|
|
tempDir, err := os.MkdirTemp("", "tempo-test-app-*")
|
|
require.NoError(t, err)
|
|
|
|
defer func() {
|
|
err := os.RemoveAll(tempDir)
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
config := NewDefaultConfig()
|
|
config.Server.HTTPListenPort = util.MustGetFreePort()
|
|
config.Server.GRPCListenPort = util.MustGetFreePort() // not used in the test; set to ensure conflict-free start
|
|
config.StorageConfig.Trace.Backend = backend.Local
|
|
config.StorageConfig.Trace.Local.Path = filepath.Join(tempDir, "tempo")
|
|
config.StorageConfig.Trace.WAL.Filepath = filepath.Join(tempDir, "wal")
|
|
config.UsageReport.Enabled = false // speeds up the shutdown process
|
|
|
|
app, err := New(*config)
|
|
require.NoError(t, err)
|
|
|
|
// start Tempo
|
|
go func() {
|
|
require.NoError(t, app.Run())
|
|
}()
|
|
|
|
// check health endpoint is reachable
|
|
healthCheckURL := fmt.Sprintf("http://localhost:%d/ready", config.Server.HTTPListenPort)
|
|
require.Eventually(t, func() bool {
|
|
t.Log("Checking Tempo is up...")
|
|
// #nosec G107 -- nosemgrep: tainted-url-host
|
|
resp, httpErr := http.Get(healthCheckURL)
|
|
return httpErr == nil && resp.StatusCode == http.StatusOK
|
|
}, 30*time.Second, 1*time.Second)
|
|
|
|
// stop Tempo
|
|
app.Stop()
|
|
|
|
// check health endpoint is not reachable anymore
|
|
require.Eventually(t, func() bool {
|
|
t.Log("Checking Tempo is down...")
|
|
// #nosec G107 -- nosemgrep: tainted-url-host
|
|
_, httpErr := http.Get(healthCheckURL)
|
|
return httpErr != nil
|
|
}, 60*time.Second, 1*time.Second)
|
|
}
|