tools/fuzz/test_utils_test.go

// Copyright 2020 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 fuzz

import (
	"bufio"
	"crypto/sha256"
	"fmt"
	"io"
	"math/rand"
	"os"
	"strings"
	"testing"

	"github.com/google/go-cmp/cmp/cmpopts"
)

const invalidPath = "/invalid/path"

func expectPathAbsent(t *testing.T, path string) {
	if _, err := os.Stat(path); !os.IsNotExist(err) {
		t.Fatalf("expected path %q to not exist, but it does", path)
	}
}

func getFlagFromArgs(t *testing.T, name string, args []string) string {
	for j, arg := range args[:len(args)-1] {
		if arg == name {
			return args[j+1]
		}
	}

	t.Fatalf("flag %s not found in args: %q", name, args)
	return ""
}

// Must be cleaned up by caller
func createTempfileWithContents(t *testing.T, contents string, extension string) string {
	file, err := os.CreateTemp("", "*."+extension)
	if err != nil {
		t.Fatalf("error creating tempfile: %s", err)
	}

	file.WriteString(contents)
	file.Close()

	return file.Name()
}

// This is used by both the mockBuild and the mock symbolize subprocess
func fakeSymbolize(in io.ReadCloser, out io.Writer) error {
	defer in.Close()

	scanner := bufio.NewScanner(in)
	for scanner.Scan() {
		line := scanner.Text()
		io.WriteString(out, strings.ReplaceAll(line, "{{{0x41}}}", "wow.c:1")+"\n")
	}
	if err := scanner.Err(); err != nil {
		return fmt.Errorf("failed during scan: %s", err)
	}
	return nil
}

func getFileHash(t *testing.T, path string) []byte {
	f, err := os.Open(path)
	if err != nil {
		t.Fatalf("error opening file: %s", err)
	}
	defer f.Close()

	hash := sha256.New()
	if _, err := io.Copy(hash, f); err != nil {
		t.Fatalf("error while hashing file %q: %s", path, err)
	}

	return hash.Sum(nil)
}

// Create a file, with random contents to allow for simple change detection
// Note: CPRNG seeds need to be propagated to subprocesses to avoid collisions
func touchRandomFile(t *testing.T, path string) {
	contents := make([]byte, 16)
	if _, err := rand.Read(contents); err != nil {
		t.Fatalf("Error getting random contents: %s", err)
	}
	if err := os.WriteFile(path, contents, 0o600); err != nil {
		t.Fatalf("Error touching file %q: %s", path, err)
	}
}

// Used to set cmp.Diff to be order-insensitive
var sortSlicesOpt = cmpopts.SortSlices(func(a, b string) bool { return a < b })