Type: "custom-type", Source: atc.Source{"some-custom": "source"}, }, })) Expect(delegate).To(Equal(putDelegate)) // TODO: Can we test the map values? Expect(sources).To(HaveKey("some-source")) Expect(sources).To(HaveKey("some-other-source")) Expect(sources).To(HaveKey("some-mounted-source")) }) It("puts the resource with the correct source and params, and the full repository as the artifact source", func() { Expect(fakeResource.PutCallCount()).To(Equal(1)) _, putSource, putParams, putArtifactSource := fakeResource.PutArgsForCall(0) Expect(putSource).To(Equal(resourceConfig.Source)) Expect(putParams).To(Equal(params)) 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())