89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
package packer
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"testing"
|
|
)
|
|
|
|
// TestUi creates a simple UI for use in testing.
|
|
// It's not meant for "real" use.
|
|
func TestUi(t *testing.T) Ui {
|
|
var buf bytes.Buffer
|
|
return &BasicUi{
|
|
Reader: &buf,
|
|
Writer: ioutil.Discard,
|
|
ErrorWriter: ioutil.Discard,
|
|
PB: &NoopProgressTracker{},
|
|
}
|
|
}
|
|
|
|
type MockUi struct {
|
|
AskCalled bool
|
|
AskQuery string
|
|
ErrorCalled bool
|
|
ErrorMessage string
|
|
MachineCalled bool
|
|
MachineType string
|
|
MachineArgs []string
|
|
MessageCalled bool
|
|
MessageMessage string
|
|
SayCalled bool
|
|
SayMessage string
|
|
|
|
TrackProgressCalled bool
|
|
ProgressBarAddCalled bool
|
|
ProgressBarCloseCalled bool
|
|
}
|
|
|
|
func (u *MockUi) Ask(query string) (string, error) {
|
|
u.AskCalled = true
|
|
u.AskQuery = query
|
|
return "foo", nil
|
|
}
|
|
|
|
func (u *MockUi) Error(message string) {
|
|
u.ErrorCalled = true
|
|
u.ErrorMessage = message
|
|
}
|
|
|
|
func (u *MockUi) Machine(t string, args ...string) {
|
|
u.MachineCalled = true
|
|
u.MachineType = t
|
|
u.MachineArgs = args
|
|
}
|
|
|
|
func (u *MockUi) Message(message string) {
|
|
u.MessageCalled = true
|
|
u.MessageMessage = message
|
|
}
|
|
|
|
func (u *MockUi) Say(message string) {
|
|
u.SayCalled = true
|
|
u.SayMessage = message
|
|
}
|
|
|
|
func (u *MockUi) TrackProgress(_ string, _, _ int64, stream io.ReadCloser) (body io.ReadCloser) {
|
|
u.TrackProgressCalled = true
|
|
|
|
return &readCloser{
|
|
read: func(p []byte) (int, error) {
|
|
u.ProgressBarAddCalled = true
|
|
return stream.Read(p)
|
|
},
|
|
close: func() error {
|
|
u.ProgressBarCloseCalled = true
|
|
return stream.Close()
|
|
},
|
|
}
|
|
}
|
|
|
|
type readCloser struct {
|
|
read func([]byte) (int, error)
|
|
close func() error
|
|
}
|
|
|
|
func (c *readCloser) Close() error { return c.close() }
|
|
func (c *readCloser) Read(p []byte) (int, error) { return c.read(p) }
|