git-sync/crawlab
1
0
Fork 0
mirror of https://github.com/crawlab-team/crawlab.git synced 2026-01-22 17:31:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: test issues

Browse Source
This commit is contained in:
Marvin Zhang
2024-11-05 13:54:05 +08:00
parent 6025f20b7d
commit 49588fb2e5
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
core/task/log/file_driver_test.go
View File

@@ -34,13 +34,12 @@ func TestFileDriver_WriteLine(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver()
require.Nil(t, err)
d := newFileLogDriver()
defer d.Close()
id := primitive.NewObjectID()
err = d.WriteLine(id.Hex(), "it works")
err := d.WriteLine(id.Hex(), "it works")
require.Nil(t, err)
logFilePath := filepath.Join(testLogDir, id.Hex(), "log.txt")
@@ -54,14 +53,13 @@ func TestFileDriver_WriteLines(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver()
require.Nil(t, err)
d := newFileLogDriver()
defer d.Close()
id := primitive.NewObjectID()
for i := 0; i < 100; i++ {
err = d.WriteLine(id.Hex(), "it works")
err := d.WriteLine(id.Hex(), "it works")
require.Nil(t, err)
}
@@ -78,8 +76,7 @@ func TestFileDriver_Find(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver()
require.Nil(t, err)
d := newFileLogDriver()
defer d.Close()
id := primitive.NewObjectID()
@@ -91,14 +88,14 @@ func TestFileDriver_Find(t *testing.T) {
line := fmt.Sprintf("line: %d", i*batch+j+1)
lines = append(lines, line)
}
err = d.WriteLines(id.Hex(), lines)
err := d.WriteLines(id.Hex(), lines)
require.Nil(t, err)
lines = []string{}
}
driver := d
lines, err = driver.Find(id.Hex(), "", 0, 10)
lines, err := driver.Find(id.Hex(), "", 0, 10)
require.Nil(t, err)
require.Equal(t, 10, len(lines))
require.Equal(t, "line: 1", lines[0])