fake_container.go

package fake_container_pool

import (
	"io"
	"sync"
	"time"

	"github.com/cloudfoundry-incubator/garden"
	"github.com/cloudfoundry-incubator/garden/fakes"
)

type FakeContainer struct {
	*fakes.FakeContainer

	Spec garden.ContainerSpec

	SnapshotError  error
	SavedSnapshots []io.Writer
	snapshotMutex  *sync.RWMutex

	StartError error
	Started    bool

	CleanedUp bool
}

func NewFakeContainer(spec garden.ContainerSpec) *FakeContainer {
	return &FakeContainer{
		Spec: spec,

		FakeContainer: new(fakes.FakeContainer),

		snapshotMutex: new(sync.RWMutex),
	}
}

func (c *FakeContainer) ID() string {
	return c.Spec.Handle
}

func (c *FakeContainer) Handle() string {
	return c.Spec.Handle
}

func (c *FakeContainer) HasProperties(ps garden.Properties) bool {
	containerProps := c.Spec.Properties

	for key, val := range ps {
		cval, ok := containerProps[key]
		if !ok {
			return false
		}

		if cval != val {
			return false
		}
	}

	return true
}

func (c *FakeContainer) Start() error {
	c.Started = true
	return c.StartError
}

func (c *FakeContainer) Cleanup() {
	c.CleanedUp = true
}

func (c *FakeContainer) GraceTime() time.Duration {
	return c.Spec.GraceTime
}

func (c *FakeContainer) Snapshot(snapshot io.Writer) error {
	if c.SnapshotError != nil {
		return c.SnapshotError
	}

	c.snapshotMutex.Lock()
	defer c.snapshotMutex.Unlock()

	c.SavedSnapshots = append(c.SavedSnapshots, snapshot)

	return nil
}