builder/dockerfile/mockbackend_test.go
9b4aa762
 package dockerfile
 
 import (
bd5f92d2
 	"encoding/json"
9b4aa762
 	"io"
0380fbff
 	"runtime"
9b4aa762
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/backend"
 	"github.com/docker/docker/api/types/container"
 	"github.com/docker/docker/builder"
49211715
 	containerpkg "github.com/docker/docker/container"
bd5f92d2
 	"github.com/docker/docker/layer"
7a7357da
 	"github.com/docker/docker/pkg/containerfs"
9b4aa762
 	"golang.org/x/net/context"
 )
 
 // MockBackend implements the builder.Backend interface for unit testing
 type MockBackend struct {
b3bc7b28
 	containerCreateFunc func(config types.ContainerCreateConfig) (container.ContainerCreateCreatedBody, error)
 	commitFunc          func(string, *backend.ContainerCommitConfig) (string, error)
 	getImageFunc        func(string) (builder.Image, builder.ReleaseableLayer, error)
ce8e529e
 	makeImageCacheFunc  func(cacheFrom []string) builder.ImageCache
9b4aa762
 }
 
32ca1214
 func (m *MockBackend) ContainerAttachRaw(cID string, stdin io.ReadCloser, stdout, stderr io.Writer, stream bool, attached chan struct{}) error {
9b4aa762
 	return nil
 }
 
 func (m *MockBackend) ContainerCreate(config types.ContainerCreateConfig) (container.ContainerCreateCreatedBody, error) {
9f738cc5
 	if m.containerCreateFunc != nil {
 		return m.containerCreateFunc(config)
 	}
9b4aa762
 	return container.ContainerCreateCreatedBody{}, nil
 }
 
 func (m *MockBackend) ContainerRm(name string, config *types.ContainerRmConfig) error {
 	return nil
 }
 
9f738cc5
 func (m *MockBackend) Commit(cID string, cfg *backend.ContainerCommitConfig) (string, error) {
 	if m.commitFunc != nil {
 		return m.commitFunc(cID, cfg)
 	}
9b4aa762
 	return "", nil
 }
 
 func (m *MockBackend) ContainerKill(containerID string, sig uint64) error {
 	return nil
 }
 
 func (m *MockBackend) ContainerStart(containerID string, hostConfig *container.HostConfig, checkpoint string, checkpointDir string) error {
 	return nil
 }
 
49211715
 func (m *MockBackend) ContainerWait(ctx context.Context, containerID string, condition containerpkg.WaitCondition) (<-chan containerpkg.StateStatus, error) {
 	return nil, nil
9b4aa762
 }
 
 func (m *MockBackend) ContainerCreateWorkdir(containerID string) error {
 	return nil
 }
 
d1faf3df
 func (m *MockBackend) CopyOnBuild(containerID string, destPath string, srcRoot string, srcPath string, decompress bool) error {
9b4aa762
 	return nil
 }
 
6c28e8ed
 func (m *MockBackend) GetImageAndReleasableLayer(ctx context.Context, refOrID string, opts backend.GetImageAndLayerOptions) (builder.Image, builder.ReleaseableLayer, error) {
b3bc7b28
 	if m.getImageFunc != nil {
 		return m.getImageFunc(refOrID)
 	}
 
 	return &mockImage{id: "theid"}, &mockLayer{}, nil
9b4aa762
 }
 
ce8e529e
 func (m *MockBackend) MakeImageCache(cacheFrom []string) builder.ImageCache {
19f3b071
 	if m.makeImageCacheFunc != nil {
ce8e529e
 		return m.makeImageCacheFunc(cacheFrom)
19f3b071
 	}
 	return nil
 }
 
ce8e529e
 func (m *MockBackend) CreateImage(config []byte, parent string) (builder.Image, error) {
51360965
 	return nil, nil
bd5f92d2
 }
 
9b4aa762
 type mockImage struct {
 	id     string
 	config *container.Config
 }
 
 func (i *mockImage) ImageID() string {
 	return i.id
 }
 
 func (i *mockImage) RunConfig() *container.Config {
 	return i.config
 }
9f738cc5
 
0380fbff
 func (i *mockImage) OperatingSystem() string {
 	return runtime.GOOS
 }
 
bd5f92d2
 func (i *mockImage) MarshalJSON() ([]byte, error) {
 	type rawImage mockImage
 	return json.Marshal(rawImage(*i))
 }
 
9f738cc5
 type mockImageCache struct {
 	getCacheFunc func(parentID string, cfg *container.Config) (string, error)
 }
 
 func (mic *mockImageCache) GetCache(parentID string, cfg *container.Config) (string, error) {
 	if mic.getCacheFunc != nil {
 		return mic.getCacheFunc(parentID, cfg)
 	}
 	return "", nil
 }
b3bc7b28
 
 type mockLayer struct{}
 
 func (l *mockLayer) Release() error {
 	return nil
 }
 
7a7357da
 func (l *mockLayer) Mount() (containerfs.ContainerFS, error) {
 	return containerfs.NewLocalContainerFS("mountPath"), nil
b3bc7b28
 }
bd5f92d2
 
afd305c4
 func (l *mockLayer) Commit() (builder.ReleaseableLayer, error) {
51360965
 	return nil, nil
 }
 
bd5f92d2
 func (l *mockLayer) DiffID() layer.DiffID {
51360965
 	return layer.DiffID("abcdef")
bd5f92d2
 }