dest := new(fakes.FakeArtifactDestination)
err := putArtifactSource.StreamTo(dest)
Expect(err).NotTo(HaveOccurred())
Expect(fakeSource.StreamToCallCount()).To(Equal(1))
sourceDest := fakeSource.StreamToArgsForCall(0)
someStream := new(bytes.Buffer)
err = sourceDest.StreamIn("foo", someStream)
Expect(err).NotTo(HaveOccurred())
Expect(dest.StreamInCallCount()).To(Equal(1))
destPath, stream := dest.StreamInArgsForCall(0)
Expect(destPath).To(Equal("some-source/foo"))
Expect(stream).To(Equal(someStream))
Expect(fakeOtherSource.StreamToCallCount()).To(Equal(1))
otherSourceDest := fakeOtherSource.StreamToArgsForCall(0)
someOtherStream := new(bytes.Buffer)
err = otherSourceDest.StreamIn("foo", someOtherStream)
Expect(err).NotTo(HaveOccurred())
Expect(dest.StreamInCallCount()).To(Equal(2))
otherDestPath, otherStream := dest.StreamInArgsForCall(1)
Expect(otherDestPath).To(Equal("some-other-source/foo"))
Expect(otherStream).To(Equal(someOtherStream))
Expect(streamErr).NotTo(HaveOccurred())
})
It("streams both sources to the destination under subdirectories", func() {
someStream := new(bytes.Buffer)
Expect(firstSource.StreamToCallCount()).To(Equal(1))
Expect(secondSource.StreamToCallCount()).To(Equal(1))
firstDestination := firstSource.StreamToArgsForCall(0)
secondDestination := secondSource.StreamToArgsForCall(0)
Expect(firstDestination.StreamIn("foo", someStream)).To(Succeed())
Expect(fakeDestination.StreamInCallCount()).To(Equal(1))
destDir, stream := fakeDestination.StreamInArgsForCall(0)
Expect(destDir).To(Equal("first-source/foo"))
Expect(stream).To(Equal(someStream))
Expect(secondDestination.StreamIn("foo", someStream)).To(Succeed())
Expect(fakeDestination.StreamInCallCount()).To(Equal(2))
destDir, stream = fakeDestination.StreamInArgsForCall(1)
Expect(destDir).To(Equal("second-source/foo"))
Expect(stream).To(Equal(someStream))
})
Context("when the any of the sources fails to stream", func() {
disaster := errors.New("nope")
BeforeEach(func() {
)
BeforeEach(func() {
streamedOut = gbytes.NewBuffer()
fakeVersionedSource.StreamOutReturns(streamedOut, nil)
})
It("streams the resource to the destination", func() {
err := artifactSource.StreamTo(fakeDestination)
Ω(err).ShouldNot(HaveOccurred())
Ω(fakeVersionedSource.StreamOutCallCount()).Should(Equal(1))
Ω(fakeVersionedSource.StreamOutArgsForCall(0)).Should(Equal("."))
Ω(fakeDestination.StreamInCallCount()).Should(Equal(1))
dest, src := fakeDestination.StreamInArgsForCall(0)
Ω(dest).Should(Equal("."))
Ω(src).Should(Equal(streamedOut))
})
Context("when streaming out of the versioned source fails", func() {
disaster := errors.New("nope")
BeforeEach(func() {
fakeVersionedSource.StreamOutReturns(nil, disaster)
})
It("returns the error", func() {
Ω(artifactSource.StreamTo(fakeDestination)).Should(Equal(disaster))
})
})
dest := new(fakes.FakeArtifactDestination)
err := putArtifactSource.StreamTo(dest)
Ω(err).ShouldNot(HaveOccurred())
Ω(fakeSource.StreamToCallCount()).Should(Equal(1))
sourceDest := fakeSource.StreamToArgsForCall(0)
someStream := new(bytes.Buffer)
err = sourceDest.StreamIn("foo", someStream)
Ω(err).ShouldNot(HaveOccurred())
Ω(dest.StreamInCallCount()).Should(Equal(1))
destPath, stream := dest.StreamInArgsForCall(0)
Ω(destPath).Should(Equal("some-source/foo"))
Ω(stream).Should(Equal(someStream))
})
It("puts the resource with the io config forwarded", func() {
Ω(fakeResource.PutCallCount()).Should(Equal(1))
ioConfig, _, _, _ := fakeResource.PutArgsForCall(0)
Ω(ioConfig.Stdout).Should(Equal(stdoutBuf))
Ω(ioConfig.Stderr).Should(Equal(stderrBuf))
})
It("runs the get resource action", func() {
Ω(fakeVersionedSource.RunCallCount()).Should(Equal(1))
})
