tmpDir := os.TempDir()
var err error
tmpFile, err = ioutil.TempFile(tmpDir, "tmp_file")
Expect(err).NotTo(HaveOccurred())
Expect(ioutil.WriteFile(tmpFile.Name(), []byte("some contents"), 0600)).To(Succeed())
})
AfterEach(func() {
Expect(os.Remove(tmpFile.Name())).To(Succeed())
})
It("uploads the file to the bucket", func() {
Expect(dropletRunner.UploadBits("droplet-name", tmpFile.Name())).To(Succeed())
Expect(fakeBlobStore.UploadCallCount()).To(Equal(1))
path, contents := fakeBlobStore.UploadArgsForCall(0)
Expect(path).To(Equal("droplet-name/bits.zip"))
Expect(ioutil.ReadAll(contents)).To(Equal([]byte("some contents")))
})
It("returns an error when we fail to open the droplet bits", func() {
err := dropletRunner.UploadBits("droplet-name", "some non-existent file")
Expect(err).To(MatchError("open some non-existent file: no such file or directory"))
})
It("returns an error when the upload fails", func() {
fakeBlobStore.UploadReturns(errors.New("some error"))
err := dropletRunner.UploadBits("droplet-name", tmpFile.Name())
Expect(err).To(MatchError("some error"))
