version_daemon_test.go

package cmd_test

import (
	"fmt"
	"io"
	"net/http"
	"os/exec"
	"strings"
	"testing"

	"github.com/filecoin-project/venus/pkg/constants"

	th "github.com/filecoin-project/venus/pkg/testhelpers"
	tf "github.com/filecoin-project/venus/pkg/testhelpers/testflags"
	manet "github.com/multiformats/go-multiaddr/net"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestVersion(t *testing.T) {
	tf.IntegrationTest(t)

	commit := getCodeCommit(t)
	verOut, err := exec.Command(th.MustGetFilecoinBinary(), "version").Output()
	require.NoError(t, err)

	version := string(verOut)
	assert.Contains(t, version, fmt.Sprintf("%s+git.%s", constants.BuildVersion, commit[0:7]))
}

func TestVersionOverHttp(t *testing.T) {
	tf.IntegrationTest(t)

	td := th.NewDaemon(t).Start()
	defer td.ShutdownSuccess()

	maddr, err := td.CmdAddr()
	require.NoError(t, err)

	_, host, err := manet.DialArgs(maddr) // nolint
	require.NoError(t, err)

	url := fmt.Sprintf("http://%s/api/version", host)
	req, err := http.NewRequest("POST", url, nil)
	require.NoError(t, err)

	token, _ := td.CmdToken()
	req.Header.Add("Authorization", "Bearer "+token)

	res, err := http.DefaultClient.Do(req)
	require.NoError(t, err)
	require.Equal(t, http.StatusOK, res.StatusCode)

	commit := getCodeCommit(t)[0:7]

	defer res.Body.Close() // nolint: errcheck
	body, err := io.ReadAll(res.Body)
	require.NoError(t, err)
	require.Contains(t, string(body), constants.BuildVersion+"+git."+commit)
}

func getCodeCommit(t *testing.T) string {
	var gitOut []byte
	var err error
	gitArgs := []string{"rev-parse", "--verify", "HEAD"}
	if gitOut, err = exec.Command("git", gitArgs...).Output(); err != nil {
		assert.NoError(t, err)
	}
	return strings.TrimSpace(string(gitOut))
}