cc-backend/internal/repository/repository_test.go

165 lines
3.5 KiB
Go
Raw Normal View History

2023-06-01 13:57:35 +02:00
// Copyright (C) 2022 NHR@FAU, University Erlangen-Nuremberg.
// All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package repository
import (
"context"
2023-06-01 13:57:35 +02:00
"testing"
"github.com/ClusterCockpit/cc-backend/internal/graph/model"
"github.com/ClusterCockpit/cc-backend/pkg/log"
"github.com/ClusterCockpit/cc-backend/pkg/schema"
2023-06-01 13:57:35 +02:00
_ "github.com/mattn/go-sqlite3"
)
func TestPragma(t *testing.T) {
t.Run("sets up a new DB", func(t *testing.T) {
db := setup(t)
for _, pragma := range []string{"synchronous", "journal_mode", "busy_timeout", "auto_vacuum", "foreign_keys"} {
t.Log("PRAGMA", pragma, getPragma(db, pragma))
}
})
}
func getPragma(db *JobRepository, name string) string {
var s string
if err := db.DB.QueryRow(`PRAGMA ` + name).Scan(&s); err != nil {
panic(err)
}
return s
}
func BenchmarkSelect1(b *testing.B) {
db := setup(b)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := db.DB.Exec(`select 1`)
noErr(b, err)
}
})
}
func BenchmarkDB_FindJobById(b *testing.B) {
var jobId int64 = 1677322
b.Run("FindJobById", func(b *testing.B) {
db := setup(b)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := db.FindById(jobId)
noErr(b, err)
}
})
})
}
func BenchmarkDB_FindJob(b *testing.B) {
var jobId int64 = 107266
var startTime int64 = 1657557241
var cluster = "fritz"
b.Run("FindJob", func(b *testing.B) {
db := setup(b)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := db.Find(&jobId, &cluster, &startTime)
noErr(b, err)
}
})
})
}
func BenchmarkDB_CountJobs(b *testing.B) {
filter := &model.JobFilter{}
filter.State = append(filter.State, "running")
cluster := "fritz"
filter.Cluster = &model.StringInput{Eq: &cluster}
2023-06-01 14:10:10 +02:00
user := "mppi133h"
filter.User = &model.StringInput{Eq: &user}
2023-06-01 13:57:35 +02:00
b.Run("CountJobs", func(b *testing.B) {
db := setup(b)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := db.CountJobs(getContext(b), []*model.JobFilter{filter})
2023-06-01 13:57:35 +02:00
noErr(b, err)
}
})
})
}
2023-06-01 15:27:19 +02:00
func BenchmarkDB_QueryJobs(b *testing.B) {
filter := &model.JobFilter{}
filter.State = append(filter.State, "running")
cluster := "fritz"
filter.Cluster = &model.StringInput{Eq: &cluster}
user := "mppi133h"
filter.User = &model.StringInput{Eq: &user}
page := &model.PageRequest{ItemsPerPage: 50, Page: 1}
order := &model.OrderByInput{Field: "startTime", Order: model.SortDirectionEnumDesc}
b.Run("QueryJobs", func(b *testing.B) {
db := setup(b)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := db.QueryJobs(getContext(b), []*model.JobFilter{filter}, page, order)
2023-06-01 15:27:19 +02:00
noErr(b, err)
}
})
})
}
func getContext(tb testing.TB) context.Context {
tb.Helper()
var roles []string
roles = append(roles, schema.GetRoleString(schema.RoleAdmin))
projects := make([]string, 0)
user := &schema.User{
Username: "demo",
Name: "The man",
Roles: roles,
Projects: projects,
AuthSource: schema.AuthViaLDAP,
}
ctx := context.Background()
return context.WithValue(ctx, ContextUserKey, user)
}
2023-06-01 13:57:35 +02:00
func setup(tb testing.TB) *JobRepository {
tb.Helper()
log.Init("warn", true)
dbfile := "testdata/job.db"
err := MigrateDB("sqlite3", dbfile)
noErr(tb, err)
Connect("sqlite3", dbfile)
return GetJobRepository()
}
func noErr(tb testing.TB, err error) {
tb.Helper()
if err != nil {
tb.Fatal("Error is not nil:", err)
}
}