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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
package datafs
import (
"io/fs"
"net/url"
"os"
"testing"
"testing/fstest"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestEnvFS_Open(t *testing.T) {
fsys, err := newEnvFS(nil)
require.NoError(t, err)
assert.IsType(t, &envFS{}, fsys)
f, err := fsys.Open("foo")
require.NoError(t, err)
assert.IsType(t, &envFile{}, f)
}
func TestEnvFile_Read(t *testing.T) {
content := `hello world`
t.Setenv("HELLO_WORLD", "hello world")
f := &envFile{name: "HELLO_WORLD"}
b := make([]byte, len(content))
n, err := f.Read(b)
require.NoError(t, err)
assert.Equal(t, len(content), n)
assert.Equal(t, content, string(b))
fsys := fstest.MapFS{}
fsys["foo/bar/baz.txt"] = &fstest.MapFile{Data: []byte("\nhello world\n")}
t.Setenv("FOO_FILE", "/foo/bar/baz.txt")
f = &envFile{name: "FOO", locfs: fsys}
b = make([]byte, len(content))
t.Logf("b len is %d", len(b))
n, err = f.Read(b)
t.Logf("b len is %d", len(b))
require.NoError(t, err)
assert.Equal(t, len(content), n)
assert.Equal(t, content, string(b))
}
func TestEnvFile_Stat(t *testing.T) {
content := []byte(`hello world`)
t.Setenv("HELLO_WORLD", "hello world")
f := &envFile{name: "HELLO_WORLD"}
fi, err := f.Stat()
require.NoError(t, err)
assert.Equal(t, int64(len(content)), fi.Size())
fsys := fstest.MapFS{}
fsys["foo/bar/baz.txt"] = &fstest.MapFile{Data: []byte("\nhello world\n")}
t.Setenv("FOO_FILE", "/foo/bar/baz.txt")
f = &envFile{name: "FOO", locfs: fsys}
fi, err = f.Stat()
require.NoError(t, err)
assert.Equal(t, int64(len(content)), fi.Size())
}
func TestEnvFS(t *testing.T) {
t.Cleanup(func() { environ = os.Environ })
u, _ := url.Parse("env:")
lfsys := fstest.MapFS{}
lfsys["foo/bar/baz.txt"] = &fstest.MapFile{Data: []byte("\nhello file\n")}
fsys, err := newEnvFS(u)
require.NoError(t, err)
assert.IsType(t, &envFS{}, fsys)
envfs, ok := fsys.(*envFS)
assert.True(t, ok)
envfs.locfs = lfsys
t.Setenv("FOO_FILE", "/foo/bar/baz.txt")
b, err := fs.ReadFile(fsys, "FOO")
require.NoError(t, err)
assert.Equal(t, "hello file", string(b))
t.Setenv("FOO", "hello world")
b, err = fs.ReadFile(fsys, "FOO")
require.NoError(t, err)
assert.Equal(t, "hello world", string(b))
require.NoError(t, fstest.TestFS(fsys, "FOO", "FOO_FILE"))
}
func TestEnvFile_ReadDir(t *testing.T) {
t.Cleanup(func() { environ = os.Environ })
t.Run("name must be .", func(t *testing.T) {
f := &envFile{name: "foo"}
_, err := f.ReadDir(-1)
require.Error(t, err)
})
t.Run("empty env should return empty dir", func(t *testing.T) {
f := &envFile{name: "."}
environ = func() []string { return []string{} }
des, err := f.ReadDir(-1)
require.NoError(t, err)
assert.Empty(t, des)
})
t.Run("non-empty env should return dir with entries", func(t *testing.T) {
f := &envFile{name: "."}
environ = func() []string { return []string{"FOO=bar", "BAR=quux"} }
des, err := f.ReadDir(-1)
require.NoError(t, err)
require.Len(t, des, 2)
assert.Equal(t, "FOO", des[0].Name())
assert.Equal(t, "BAR", des[1].Name())
})
t.Run("deal with odd Windows env vars like '=C:=C:\tmp'", func(t *testing.T) {
f := &envFile{name: "."}
environ = func() []string { return []string{"FOO=bar", "=C:=C:\\tmp", "BAR=quux"} }
des, err := f.ReadDir(-1)
require.NoError(t, err)
require.Len(t, des, 2)
assert.Equal(t, "FOO", des[0].Name())
assert.Equal(t, "BAR", des[1].Name())
})
}
|
