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
|
package stdlib_test
import (
"io/ioutil"
"os"
"testing"
"github.com/d5/tengo/v2"
"github.com/d5/tengo/v2/require"
)
func TestReadFile(t *testing.T) {
content := []byte("the quick brown fox jumps over the lazy dog")
tf, err := ioutil.TempFile("", "test")
require.NoError(t, err)
defer func() { _ = os.Remove(tf.Name()) }()
_, err = tf.Write(content)
require.NoError(t, err)
_ = tf.Close()
module(t, "os").call("read_file", tf.Name()).
expect(&tengo.Bytes{Value: content})
}
func TestReadFileArgs(t *testing.T) {
module(t, "os").call("read_file").expectError()
}
func TestFileStatArgs(t *testing.T) {
module(t, "os").call("stat").expectError()
}
func TestFileStatFile(t *testing.T) {
content := []byte("the quick brown fox jumps over the lazy dog")
tf, err := ioutil.TempFile("", "test")
require.NoError(t, err)
defer func() { _ = os.Remove(tf.Name()) }()
_, err = tf.Write(content)
require.NoError(t, err)
_ = tf.Close()
stat, err := os.Stat(tf.Name())
if err != nil {
t.Logf("could not get tmp file stat: %s", err)
return
}
module(t, "os").call("stat", tf.Name()).expect(&tengo.ImmutableMap{
Value: map[string]tengo.Object{
"name": &tengo.String{Value: stat.Name()},
"mtime": &tengo.Time{Value: stat.ModTime()},
"size": &tengo.Int{Value: stat.Size()},
"mode": &tengo.Int{Value: int64(stat.Mode())},
"directory": tengo.FalseValue,
},
})
}
func TestFileStatDir(t *testing.T) {
td, err := ioutil.TempDir("", "test")
require.NoError(t, err)
defer func() { _ = os.RemoveAll(td) }()
stat, err := os.Stat(td)
require.NoError(t, err)
module(t, "os").call("stat", td).expect(&tengo.ImmutableMap{
Value: map[string]tengo.Object{
"name": &tengo.String{Value: stat.Name()},
"mtime": &tengo.Time{Value: stat.ModTime()},
"size": &tengo.Int{Value: stat.Size()},
"mode": &tengo.Int{Value: int64(stat.Mode())},
"directory": tengo.TrueValue,
},
})
}
func TestOSExpandEnv(t *testing.T) {
curMaxStringLen := tengo.MaxStringLen
defer func() { tengo.MaxStringLen = curMaxStringLen }()
tengo.MaxStringLen = 12
_ = os.Setenv("TENGO", "FOO BAR")
module(t, "os").call("expand_env", "$TENGO").expect("FOO BAR")
_ = os.Setenv("TENGO", "FOO")
module(t, "os").call("expand_env", "$TENGO $TENGO").expect("FOO FOO")
_ = os.Setenv("TENGO", "123456789012")
module(t, "os").call("expand_env", "$TENGO").expect("123456789012")
_ = os.Setenv("TENGO", "1234567890123")
module(t, "os").call("expand_env", "$TENGO").expectError()
_ = os.Setenv("TENGO", "123456")
module(t, "os").call("expand_env", "$TENGO$TENGO").expect("123456123456")
_ = os.Setenv("TENGO", "123456")
module(t, "os").call("expand_env", "${TENGO}${TENGO}").
expect("123456123456")
_ = os.Setenv("TENGO", "123456")
module(t, "os").call("expand_env", "$TENGO $TENGO").expectError()
_ = os.Setenv("TENGO", "123456")
module(t, "os").call("expand_env", "${TENGO} ${TENGO}").expectError()
}
|
