daemon_test.go

package test

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"
	"os"
	"os/exec"
	"path"
	"strings"
	"testing"

	"github.com/google/pprof/profile"
	"github.com/moby/buildkit-bench/util/testutil"
	"github.com/stretchr/testify/require"
)

func TestDaemon(t *testing.T) {
	testutil.Run(t, testutil.TestFuncs(
		testDaemonVersion,
		testDaemonDebugHeap,
	))
}

func BenchmarkDaemon(b *testing.B) {
	testutil.Run(b, testutil.BenchFuncs(
		benchmarkDaemonVersion,
		benchmarkDaemonSize,
	))
}

func testDaemonVersion(t *testing.T, sb testutil.Sandbox) {
	buildkitdPath := path.Join(sb.BuildKitBinsDir(), sb.Name(), "buildkitd")

	output, err := exec.CommandContext(context.Background(), buildkitdPath, "--version").Output()
	require.NoError(t, err)

	versionParts := strings.Fields(string(output))
	require.Len(t, versionParts, 4)
	require.Equal(t, "buildkitd", versionParts[0])
	t.Log("repo:", versionParts[1])
	t.Log("version:", versionParts[2])
	t.Log("commit:", versionParts[3])
}

func testDaemonDebugHeap(t *testing.T, sb testutil.Sandbox) {
	client := &http.Client{}

	req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, fmt.Sprintf("http://%s/debug/pprof/heap?debug=1", sb.DebugAddress()), nil)
	require.NoError(t, err)
	resp, err := client.Do(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	p, err := profile.Parse(resp.Body)
	require.NoError(t, err)

	b, err := json.MarshalIndent(p, "", "  ")
	require.NoError(t, err)

	require.NotEmpty(t, p.SampleType, string(b))
	require.NotEmpty(t, p.Sample, string(b))
}

func benchmarkDaemonVersion(b *testing.B, sb testutil.Sandbox) {
	buildkitdPath := path.Join(sb.BuildKitBinsDir(), sb.Name(), "buildkitd")
	b.StartTimer()
	err := exec.CommandContext(context.Background(), buildkitdPath, "--version").Run()
	b.StopTimer()
	require.NoError(b, err)
}

func benchmarkDaemonSize(b *testing.B, sb testutil.Sandbox) {
	buildkitdPath := path.Join(sb.BuildKitBinsDir(), sb.Name(), "buildkitd")
	fi, err := os.Stat(buildkitdPath)
	require.NoError(b, err)
	testutil.ReportMetric(b, float64(fi.Size()), testutil.MetricBytes)
}