-
Notifications
You must be signed in to change notification settings - Fork 4.9k
/
file_test.go
113 lines (85 loc) · 2.78 KB
/
file_test.go
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
package input
import (
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
)
// TODO: Tests to be implemented
// * Check file renaming
// * Check file ids for moved files (windows)
func TestIsSameFile(t *testing.T) {
absPath, err := filepath.Abs("../tests/files/")
assert.NotNil(t, absPath)
assert.Nil(t, err)
fileInfo1, err := os.Stat(absPath + "/logs/test.log")
fileInfo2, err := os.Stat(absPath + "/logs/system.log")
assert.Nil(t, err)
assert.NotNil(t, fileInfo1)
assert.NotNil(t, fileInfo2)
file1 := &File{
FileInfo: fileInfo1,
}
file2 := &File{
FileInfo: fileInfo2,
}
file3 := &File{
FileInfo: fileInfo2,
}
assert.False(t, file1.IsSameFile(file2))
assert.False(t, file2.IsSameFile(file1))
assert.True(t, file1.IsSameFile(file1))
assert.True(t, file2.IsSameFile(file2))
assert.True(t, file3.IsSameFile(file2))
assert.True(t, file2.IsSameFile(file3))
}
func TestSafeFileRotateExistingFile(t *testing.T) {
tempdir, err := ioutil.TempDir("", "")
assert.NoError(t, err)
defer func() {
assert.NoError(t, os.RemoveAll(tempdir))
}()
// create an existing .filebeat file
err = ioutil.WriteFile(filepath.Join(tempdir, ".filebeat"),
[]byte("existing filebeat"), 0x777)
assert.NoError(t, err)
// create a new .filebeat.new file
err = ioutil.WriteFile(filepath.Join(tempdir, ".filebeat.new"),
[]byte("new filebeat"), 0x777)
assert.NoError(t, err)
// rotate .filebeat.new into .filebeat
err = SafeFileRotate(filepath.Join(tempdir, ".filebeat"),
filepath.Join(tempdir, ".filebeat.new"))
assert.NoError(t, err)
contents, err := ioutil.ReadFile(filepath.Join(tempdir, ".filebeat"))
assert.NoError(t, err)
assert.Equal(t, []byte("new filebeat"), contents)
// do it again to make sure we deal with deleting the old file
err = ioutil.WriteFile(filepath.Join(tempdir, ".filebeat.new"),
[]byte("new filebeat 1"), 0x777)
assert.NoError(t, err)
err = SafeFileRotate(filepath.Join(tempdir, ".filebeat"),
filepath.Join(tempdir, ".filebeat.new"))
assert.NoError(t, err)
contents, err = ioutil.ReadFile(filepath.Join(tempdir, ".filebeat"))
assert.NoError(t, err)
assert.Equal(t, []byte("new filebeat 1"), contents)
// and again for good measure
err = ioutil.WriteFile(filepath.Join(tempdir, ".filebeat.new"),
[]byte("new filebeat 2"), 0x777)
assert.NoError(t, err)
err = SafeFileRotate(filepath.Join(tempdir, ".filebeat"),
filepath.Join(tempdir, ".filebeat.new"))
assert.NoError(t, err)
contents, err = ioutil.ReadFile(filepath.Join(tempdir, ".filebeat"))
assert.NoError(t, err)
assert.Equal(t, []byte("new filebeat 2"), contents)
}
func TestFileEventToMapStr(t *testing.T) {
// Test 'fields' is not present when it is nil.
event := FileEvent{}
mapStr := event.ToMapStr()
_, found := mapStr["fields"]
assert.False(t, found)
}