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
|
// Copyright (c) 2018, Sylabs Inc. All rights reserved.
// This software is licensed under a 3-clause BSD license. Please consult the
// LICENSE.md file distributed with the sources of this project regarding your
// rights to use or distribute this software.
package sources
import (
"path/filepath"
"testing"
"github.com/sylabs/singularity/v4/internal/pkg/test"
"github.com/sylabs/singularity/v4/internal/pkg/util/fs"
)
func testWithGoodDir(t *testing.T, f func(d string) error) {
if err := f(t.TempDir()); err != nil {
t.Fatalf("Unexpected failure: %v", err)
}
}
func testWithBadDir(t *testing.T, f func(d string) error) {
if err := f("/this/will/be/a/problem"); err == nil {
t.Fatalf("Unexpected success with bad directory")
}
}
func TestMakeDirs(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
testWithGoodDir(t, makeDirs)
testWithBadDir(t, makeDirs)
}
func TestMakeSymlinks(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
testWithGoodDir(t, makeSymlinks)
testWithBadDir(t, makeSymlinks)
}
func TestMakeFiles(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
testWithGoodDir(t, func(d string) error {
if err := makeDirs(d); err != nil {
return err
}
return makeFiles(d)
})
testWithBadDir(t, makeFiles)
// #4532 - Check that we can succeed with an existing file that doesn't have
// write permission.
testWithGoodDir(t, func(d string) error {
if err := makeDirs(d); err != nil {
return err
}
err := fs.EnsureFileWithPermission(filepath.Join(d, "etc", "hosts"), 0o400)
if err != nil {
t.Fatalf("Failed to make test hosts file: %s", err)
}
return makeFiles(d)
})
}
func TestMakeBaseEnv(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
testWithGoodDir(t, makeBaseEnv)
testWithBadDir(t, makeBaseEnv)
}
|
