mirror of
https://github.com/goharbor/harbor
synced 2025-04-18 17:45:46 +00:00
49 lines
977 B
Go
49 lines
977 B
Go
package sweeper
|
|
|
|
import (
|
|
"github.com/goharbor/harbor/src/common/dao"
|
|
"github.com/goharbor/harbor/src/common/utils/log"
|
|
"github.com/goharbor/harbor/src/jobservice/logger/backend"
|
|
"github.com/stretchr/testify/require"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
databases := []string{"postgresql"}
|
|
for _, database := range databases {
|
|
log.Infof("run test cases for database: %s", database)
|
|
|
|
result := 1
|
|
switch database {
|
|
case "postgresql":
|
|
dao.PrepareTestForPostgresSQL()
|
|
default:
|
|
log.Fatalf("invalid database: %s", database)
|
|
}
|
|
|
|
result = m.Run()
|
|
|
|
if result != 0 {
|
|
os.Exit(result)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// TestDBGetter
|
|
func TestDBGetter(t *testing.T) {
|
|
uuid := "uuid_for_unit_test_sweeper"
|
|
l, err := backend.NewDBLogger(uuid, "DEBUG", 4)
|
|
require.Nil(t, err)
|
|
|
|
l.Debug("JobLog Debug: TestDBLoggerSweeper")
|
|
l.Close()
|
|
|
|
PrepareDBSweep()
|
|
dbSweeper := NewDBSweeper(-1)
|
|
count, err := dbSweeper.Sweep()
|
|
require.Nil(t, err)
|
|
require.Equal(t, 1, count)
|
|
}
|