Function TestApp_RunStop

cmd/tempo/app/app_test.go:16–61

(t *testing.T)

)

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.Target = BackendScheduler
	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)
}