garnet/bin/cpuperf/print/tests/raw_printer_test.go

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package tests

import (
	"io/ioutil"
	"os"
	"path"
	"path/filepath"
	"runtime"
	"testing"
)

const (
	sessionFilename        = "raw-test.cpsession"
	expectedOutputFilename = "raw-expected-output.txt"
	printerProgram         = "cpuperf_print"
	outputTempFile         = "raw-printer-test."
)

func TestRawPrinter(t *testing.T) {
	testPath, err := filepath.Abs(os.Args[0])
	if err != nil {
		t.Fatal(err)
	}
	outDir := filepath.Dir(testPath)

	// TODO(IN-819): a required dance, so long as we are running tests out
	// of $root_build_dir/host_tests, given how the cpuperf_print tool is
	// symlinked when built with a variant (e.g., host_asan). When this is
	// no longer the case, we may set
	// printerProgramPath := path.Join(outDir, printerProgram)
	rootBuildDir := filepath.Dir(outDir)
	var hostBuildDir string
	switch runtime.GOARCH {
	case "amd64":
		hostBuildDir = path.Join(rootBuildDir, "host_x64")
	case "arm64":
		hostBuildDir = path.Join(rootBuildDir, "host_arm64")
	}
	printerProgramPath := path.Join(hostBuildDir, printerProgram)

	testDataDir := path.Join(outDir, "test_data", "cpuperf")
	sessionSpecPath := path.Join(testDataDir, sessionFilename)
	expectedOutputPath := path.Join(testDataDir, expectedOutputFilename)

	outputFile, err := ioutil.TempFile("", outputTempFile)
	if err != nil {
		t.Fatalf("Error creating output file: %s", err.Error())
	}
	defer os.Remove(outputFile.Name())
	defer outputFile.Close()

	// Pass --quiet so INFO lines, which contain source line numbers
	// and the output path prefix, won't cause erroneous failures.
	args := []string{"--session=" + sessionSpecPath, "--quiet"}
	err = runCommandWithOutputToFile(printerProgramPath, args,
		outputFile)
	if err != nil {
		t.Fatal(err)
	}

	err = compareFiles(expectedOutputPath, outputFile.Name())
	if err != nil {
		t.Fatalf("Error comparing output with expected output: %v", err)
	}
}