pipenv-install/venv_locator_test.go at main · paketo-buildpacks/pipenv-install · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package pipenvinstall_test

import (
	"os"
	"path/filepath"
	"testing"

	pipenvinstall "github.com/paketo-buildpacks/pipenv-install"
	"github.com/sclevine/spec"

	. "github.com/onsi/gomega"
)

func testVenvLocator(t *testing.T, context spec.G, it spec.S) {
	var (
		Expect = NewWithT(t).Expect

		layerPath string

		process pipenvinstall.VenvLocator
	)

	it.Before(func() {
		var err error
		layerPath, err = os.MkdirTemp("", "layer")
		Expect(err).NotTo(HaveOccurred())

		Expect(os.Mkdir(filepath.Join(layerPath, "some-virtualenv-dir"), os.ModePerm)).To(Succeed())
		Expect(os.WriteFile(filepath.Join(layerPath, "some-virtualenv-dir", "pyvenv.cfg"), nil, os.ModePerm)).To(Succeed())

		Expect(os.Mkdir(filepath.Join(layerPath, "some-other-dir"), os.ModePerm)).To(Succeed())

		process = pipenvinstall.NewVenvLocator()
	})

	it.After(func() {
		Expect(os.RemoveAll(layerPath)).To(Succeed())
	})

	context("LocateVenvDir", func() {
		it("returns the full path to the packages", func() {
			venvDir, err := process.LocateVenvDir(layerPath)
			Expect(err).NotTo(HaveOccurred())

			Expect(venvDir).To(Equal(filepath.Join(layerPath, "some-virtualenv-dir")))
		})

		context("failure cases", func() {
			context("when reading the root directory fails", func() {
				it.Before(func() {
					Expect(os.Chmod(layerPath, 0000)).To(Succeed())
				})

				it.After(func() {
					Expect(os.Chmod(layerPath, os.ModePerm)).To(Succeed())
				})

				it("returns an error", func() {
					_, err := process.LocateVenvDir(layerPath)
					Expect(err).To(MatchError(ContainSubstring("permission denied")))
				})
			})

			context("when reading a subdirectory fails", func() {
				it.Before(func() {
					Expect(os.Chmod(filepath.Join(layerPath, "some-virtualenv-dir"), 0000)).To(Succeed())
				})

				it.After(func() {
					Expect(os.Chmod(filepath.Join(layerPath, "some-virtualenv-dir"), os.ModePerm)).To(Succeed())
				})

				it("returns an error", func() {
					_, err := process.LocateVenvDir(layerPath)
					Expect(err).To(MatchError(ContainSubstring("lookup failed")))
				})
			})

			context("when there is no virtual env directory", func() {
				var emptyLayerPath string
				it.Before(func() {
					var err error
					emptyLayerPath, err = os.MkdirTemp("", "layer")
					Expect(err).NotTo(HaveOccurred())
				})

				it.After(func() {
					Expect(os.RemoveAll(emptyLayerPath)).To(Succeed())
				})

				it("returns an error", func() {
					_, err := process.LocateVenvDir(emptyLayerPath)
					Expect(err).To(MatchError(ContainSubstring("virtual env directory not found")))
				})
			})
		})
	})
}