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

test: updated test cases

Browse Source
This commit is contained in:
Marvin Zhang
2024-10-20 17:55:57 +08:00
parent b5e888f3e1
commit cecf4dfd1d
20 changed files with 79 additions and 321 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,27 +2,39 @@ package log
import (
"fmt"
"github.com/stretchr/testify/require"
"go.mongodb.org/mongo-driver/bson/primitive"
"os"
"path/filepath"
"strings"
"testing"
"github.com/spf13/viper"
"github.com/stretchr/testify/require"
"go.mongodb.org/mongo-driver/bson/primitive"
)
var testLogDir string
func setupFileDriverTest() {
cleanupFileDriverTest()
_ = os.MkdirAll("./tmp", os.ModePerm)
var err error
testLogDir, err = os.MkdirTemp("", "crawlab-test-logs")
if err != nil {
panic(err)
}
// Set the log path in viper configuration
viper.Set("log.path", testLogDir)
}
func cleanupFileDriverTest() {
_ = os.RemoveAll("./tmp")
_ = os.RemoveAll(testLogDir)
// Reset the log path in viper configuration
viper.Set("log.path", "")
}
func TestFileDriver_WriteLine(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver(nil)
d, err := newFileLogDriver()
require.Nil(t, err)
defer d.Close()
@@ -31,7 +43,7 @@ func TestFileDriver_WriteLine(t *testing.T) {
err = d.WriteLine(id.Hex(), "it works")
require.Nil(t, err)
logFilePath := fmt.Sprintf("/var/log/crawlab/%s/log.txt", id.Hex())
logFilePath := filepath.Join(testLogDir, id.Hex(), "log.txt")
require.FileExists(t, logFilePath)
text, err := os.ReadFile(logFilePath)
require.Nil(t, err)
@@ -42,7 +54,7 @@ func TestFileDriver_WriteLines(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver(nil)
d, err := newFileLogDriver()
require.Nil(t, err)
defer d.Close()
@@ -53,7 +65,7 @@ func TestFileDriver_WriteLines(t *testing.T) {
require.Nil(t, err)
}
logFilePath := fmt.Sprintf("/var/log/crawlab/%s/log.txt", id.Hex())
logFilePath := filepath.Join(testLogDir, id.Hex(), "log.txt")
require.FileExists(t, logFilePath)
text, err := os.ReadFile(logFilePath)
require.Nil(t, err)
@@ -66,7 +78,7 @@ func TestFileDriver_Find(t *testing.T) {
setupFileDriverTest()
t.Cleanup(cleanupFileDriverTest)
d, err := newFileLogDriver(nil)
d, err := newFileLogDriver()
require.Nil(t, err)
defer d.Close()