tools/fuzz/test_utils.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"
	"io/ioutil"
	"math/rand"
	"os"
	"strings"
	"testing"
)

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 getTempdir(t *testing.T) string {
	tmpDir, err := ioutil.TempDir("", "clusterfuchsia_test")

	if err != nil {
		t.Fatalf("error creating temp dir: %s", err)
	}

	return tmpDir
}

// Must be cleaned up by caller
func createTempfileWithContents(t *testing.T, contents string, extension string) string {
	file, err := ioutil.TempFile("", "*."+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 := ioutil.WriteFile(path, contents, 0o600); err != nil {
		t.Fatalf("Error touching file %q: %s", path, err)
	}
}