src/go.chromium.org/tast/core/internal/devserver/fake.go

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

package devserver

import (
	"bytes"
	"context"
	"io"
	"net/url"
	"os"

	"go.chromium.org/tast/core/errors"
)

// FakeClient is a fake implementation of devserver.Client suitable for unit tests.
type FakeClient struct {
	files map[string][]byte // GS URL -> content
}

var _ Client = &FakeClient{}

// NewFakeClient constructs a FakeClient. files is a map from GS URL to content.
func NewFakeClient(files map[string][]byte) *FakeClient {
	return &FakeClient{files}
}

// TearDown does nothing.
func (c *FakeClient) TearDown() error {
	return nil
}

// Open simulates a download from Google Cloud Storage.
func (c *FakeClient) Open(ctx context.Context, gsURL string) (io.ReadCloser, error) {
	data, ok := c.files[gsURL]
	if !ok {
		return nil, os.ErrNotExist
	}
	return io.NopCloser(bytes.NewReader(data)), nil
}

// Stage simulates a getting a url to Google Cloud Storage.
func (c *FakeClient) Stage(ctx context.Context, gsURL string) (*url.URL, error) {
	return nil, errors.New("FakeClient.Stage not implemented")
}