func newFakeStreamer() *fake_log_streamer.FakeLogStreamer { fakeStreamer := new(fake_log_streamer.FakeLogStreamer) stdoutBuffer := gbytes.NewBuffer() stderrBuffer := gbytes.NewBuffer() fakeStreamer.StdoutReturns(stdoutBuffer) fakeStreamer.StderrReturns(stderrBuffer) return fakeStreamer }
var fakeStreamer *fake_log_streamer.FakeLogStreamer var startMessage, successMessage, failureMessage string var logger *lagertest.TestLogger var stderrBuffer *bytes.Buffer var stdoutBuffer *bytes.Buffer BeforeEach(func() { stderrBuffer = new(bytes.Buffer) stdoutBuffer = new(bytes.Buffer) errorToReturn = nil startMessage, successMessage, failureMessage = "", "", "" cancelled = false fakeStreamer = new(fake_log_streamer.FakeLogStreamer) fakeStreamer.StderrReturns(stderrBuffer) fakeStreamer.StdoutReturns(stdoutBuffer) subStep = &fakes.FakeStep{ PerformStub: func() error { fakeStreamer.Stdout().Write([]byte("RUNNING\n")) return errorToReturn }, CancelStub: func() { cancelled = true }, } logger = lagertest.NewTestLogger("test") }) JustBeforeEach(func() {
runAction = models.RunAction{ Path: "sudo", Args: []string{"reboot"}, Dir: "/some-dir", Env: []*models.EnvironmentVariable{ {Name: "A", Value: "1"}, {Name: "B", Value: "2"}, }, ResourceLimits: &models.ResourceLimits{ Nofile: &fileDescriptorLimit, }, User: "******", } fakeStreamer = new(fake_log_streamer.FakeLogStreamer) fakeStreamer.StdoutReturns(noOpWriter{}) gardenClient = fakes.NewGardenClient() logger = lagertest.NewTestLogger("test") spawnedProcess = new(gfakes.FakeProcess) runError = nil gardenClient.Connection.RunStub = func(string, garden.ProcessSpec, garden.ProcessIO) (garden.Process, error) { return spawnedProcess, runError } externalIP = "external-ip" portMappings = nil exportNetworkEnvVars = false fakeClock = fakeclock.NewFakeClock(time.Unix(123, 456))