blob: 309bdecce1559a93bc3acdc321735ac8c9dbe41e [file] [log] [blame]
// 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 (
// mockConnector is used in Fuzzer and Instance tests, and returned by mockLauncher.Start()
type mockConnector struct {
connected bool
shouldFailToConnectCount uint
shouldFailToExecuteCount uint
shouldFailToGetSysLog bool
// Store history of Get/Put paths to enable basic checks
PathsGot []string
PathsPut []string
// Store history of commands run on this connection
CmdHistory []string
func (c *mockConnector) Connect() error {
if c.connected {
return fmt.Errorf("Connect called when already connected")
if c.shouldFailToConnectCount > 0 {
c.shouldFailToConnectCount -= 1
return fmt.Errorf("Intentionally broken Connector")
c.connected = true
return nil
func (c *mockConnector) Close() {
c.connected = false
func (c *mockConnector) Command(name string, args ...string) InstanceCmd {
shouldFail := c.shouldFailToExecuteCount > 0
if shouldFail {
c.shouldFailToExecuteCount -= 1
return &mockInstanceCmd{connector: c, name: name, args: args, shouldFail: shouldFail}
func (c *mockConnector) Get(targetSrc string, hostDst string) error {
c.PathsGot = append(c.PathsGot, targetSrc)
return nil
func (c *mockConnector) Put(hostSrc string, targetDst string) error {
c.PathsPut = append(c.PathsPut, targetDst)
return nil
func (c *mockConnector) GetSysLog(pid int) (string, error) {
if c.shouldFailToGetSysLog {
return "syslog failure", fmt.Errorf("Intentionally broken Connector")
// TODO( more realistic test data
lines := []string{
fmt.Sprintf("syslog for %d", pid),
"[1234.5][klog] INFO: {{{0x41}}}",
return strings.Join(lines, "\n"), nil