}) }) Describe("CleanUp", func() { It("removes tarball path", func() { fs := fakesys.NewFakeFileSystem() compressor := NewTarballCompressor(cmdRunner, fs) err := fs.WriteFileString("/fake-tarball.tar", "") Expect(err).ToNot(HaveOccurred()) err = compressor.CleanUp("/fake-tarball.tar") Expect(err).ToNot(HaveOccurred()) Expect(fs.FileExists("/fake-tarball.tar")).To(BeFalse()) }) It("returns error if removing tarball path fails", func() { fs := fakesys.NewFakeFileSystem() compressor := NewTarballCompressor(cmdRunner, fs) fs.RemoveAllStub = func(_ string) error { return errors.New("fake-remove-all-err") } err := compressor.CleanUp("/fake-tarball.tar") Expect(err).To(MatchError("fake-remove-all-err")) }) }) })