67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
// Copyright 2016 Google Inc. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
package fake
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
type DmsetupCommand struct {
|
|
Name string
|
|
Result string
|
|
Err error
|
|
}
|
|
|
|
// NewFakeDmsetupClient returns a new fake DmsetupClient.
|
|
func NewFakeDmsetupClient(t *testing.T, commands ...DmsetupCommand) *FakeDmsetupClient {
|
|
if len(commands) == 0 {
|
|
commands = make([]DmsetupCommand, 0)
|
|
}
|
|
return &FakeDmsetupClient{t: t, commands: commands}
|
|
}
|
|
|
|
// FakeDmsetupClient is a thread-unsafe fake implementation of the
|
|
// DmsetupClient interface
|
|
type FakeDmsetupClient struct {
|
|
t *testing.T
|
|
commands []DmsetupCommand
|
|
}
|
|
|
|
func (c *FakeDmsetupClient) Table(deviceName string) ([]byte, error) {
|
|
return c.dmsetup("table")
|
|
}
|
|
|
|
func (c *FakeDmsetupClient) Message(deviceName string, sector int, message string) ([]byte, error) {
|
|
return c.dmsetup("message")
|
|
}
|
|
|
|
func (c *FakeDmsetupClient) Status(deviceName string) ([]byte, error) {
|
|
return c.dmsetup("status")
|
|
}
|
|
|
|
func (c *FakeDmsetupClient) AddCommand(name string, result string, err error) {
|
|
c.commands = append(c.commands, DmsetupCommand{name, result, err})
|
|
}
|
|
|
|
func (c *FakeDmsetupClient) dmsetup(inputCommand string) ([]byte, error) {
|
|
var nextCommand DmsetupCommand
|
|
nextCommand, c.commands = c.commands[0], c.commands[1:]
|
|
if nextCommand.Name != inputCommand {
|
|
c.t.Fatalf("unexpected dmsetup command; expected: %q, got %q", nextCommand.Name, inputCommand)
|
|
// should be unreachable in a test context.
|
|
}
|
|
|
|
return []byte(nextCommand.Result), nextCommand.Err
|
|
}
|