mirror of
https://github.com/ClusterCockpit/cc-backend
synced 2026-01-15 17:21:46 +01:00
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
// Copyright (C) NHR@FAU, University Erlangen-Nuremberg.
|
|
// All rights reserved. This file is part of cc-backend.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ClusterCockpit/cc-backend/internal/graph/model"
|
|
)
|
|
|
|
func TestBuildJobStatsQuery(t *testing.T) {
|
|
r := setup(t)
|
|
q := r.buildStatsQuery(nil, "USER")
|
|
|
|
sql, _, err := q.ToSql()
|
|
noErr(t, err)
|
|
|
|
fmt.Printf("SQL: %s\n", sql)
|
|
}
|
|
|
|
func TestJobStats(t *testing.T) {
|
|
r := setup(t)
|
|
|
|
// First, count the actual jobs in the database (excluding test jobs)
|
|
var expectedCount int
|
|
err := r.DB.QueryRow(`SELECT COUNT(*) FROM job WHERE cluster != 'testcluster'`).Scan(&expectedCount)
|
|
noErr(t, err)
|
|
|
|
filter := &model.JobFilter{}
|
|
// Exclude test jobs created by other tests
|
|
testCluster := "testcluster"
|
|
filter.Cluster = &model.StringInput{Neq: &testCluster}
|
|
|
|
stats, err := r.JobsStats(getContext(t), []*model.JobFilter{filter})
|
|
noErr(t, err)
|
|
|
|
if stats[0].TotalJobs != expectedCount {
|
|
t.Fatalf("Want %d, Got %d", expectedCount, stats[0].TotalJobs)
|
|
}
|
|
}
|