snapshot of https://github.com/traggo/server.git (master, 2026-09-16; blobs > 131072B omitted)
Some checks failed
build / build (push) Failing after 3s
Some checks failed
build / build (push) Failing after 3s
This commit is contained in:
upstream-mirror
322
statistics/summary_test.go
Normal file
322
statistics/summary_test.go
Normal file
@@ -0,0 +1,322 @@
|
||||
package statistics
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/traggo/server/generated/gqlmodel"
|
||||
"github.com/traggo/server/model"
|
||||
"github.com/traggo/server/test"
|
||||
"github.com/traggo/server/test/fake"
|
||||
)
|
||||
|
||||
func TestSummary_fails(t *testing.T) {
|
||||
db := test.InMemoryDB(t)
|
||||
defer db.Close()
|
||||
|
||||
resolver := ResolverForStatistics{DB: db.DB}
|
||||
|
||||
_, err := resolver.Stats(fake.User(1), []*gqlmodel.Range{rangex("2019-06-11T10:00:00Z", "2019-06-11T09:00:00Z")}, []string{"type"}, nil, nil)
|
||||
require.Error(t, err, "range start must be before range end ")
|
||||
}
|
||||
|
||||
func TestStats(t *testing.T) {
|
||||
data := []*testEntry{
|
||||
name("simple").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"))).
|
||||
Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:00:10Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:00:10Z",
|
||||
entry("type", "review", 10*time.Second))),
|
||||
name("boundaries 1").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"))).
|
||||
Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:00:05Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:00:05Z",
|
||||
entry("type", "review", 5*time.Second))),
|
||||
name("boundaries 2").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"))).
|
||||
Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:05Z", "2019-06-11T10:00:10Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:05Z", "2019-06-11T10:00:10Z",
|
||||
entry("type", "review", 5*time.Second))),
|
||||
name("boundaries 3").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"))).
|
||||
Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z",
|
||||
entry("type", "review", 2*time.Second))),
|
||||
name("exclude 1").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"), tag("issue", "13"))).
|
||||
Keys("type").
|
||||
Exclude(tag("issue", "13")).
|
||||
Ranges(rangex("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z", []*gqlmodel.StatisticsEntry{}...)),
|
||||
name("empty range").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"), tag("issue", "13"))).
|
||||
Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z"),
|
||||
rangex("2019-06-12T10:00:04Z", "2019-06-12T10:00:06Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:04Z", "2019-06-11T10:00:06Z", entry("type", "review", 2*time.Second)),
|
||||
result("2019-06-12T10:00:04Z", "2019-06-12T10:00:06Z", entry("type", "review", 0*time.Second))),
|
||||
name("include 1").
|
||||
Load(ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"), tag("issue", "13"))).
|
||||
Keys("type").
|
||||
Include(tag("issue", "13")).
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:00:10Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:00:10Z",
|
||||
entry("type", "review", 10*time.Second))),
|
||||
name("multiple 1").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 10*time.Second, tag("type", "review"), tag("issue", "14")),
|
||||
).Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:02:00Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:02:00Z",
|
||||
entry("type", "review", 20*time.Second))),
|
||||
name("multiple 2").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 10*time.Second, tag("type", "review"), tag("issue", "14")),
|
||||
).Keys("type").
|
||||
Exclude(tag("issue", "14")).
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:02:00Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:02:00Z",
|
||||
entry("type", "review", 10*time.Second))),
|
||||
name("fill empty range").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("type", "review")),
|
||||
).Keys("type").
|
||||
Ranges(
|
||||
rangex("2019-06-10T09:58:00Z", "2019-06-10T09:59:59Z"),
|
||||
rangex("2019-06-10T10:00:00Z", "2019-06-11T10:02:00Z")).
|
||||
Expected(
|
||||
result("2019-06-10T09:58:00Z", "2019-06-10T09:59:59Z",
|
||||
entry("type", "review", 0*time.Second)),
|
||||
result("2019-06-10T10:00:00Z", "2019-06-11T10:02:00Z",
|
||||
entry("type", "review", 10*time.Second))),
|
||||
name("multiple 3").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 60*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:02:00Z", 7*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:03:00Z", 14*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:04:00Z", 16*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "14")),
|
||||
ts("2019-06-11T10:05:00Z", 33*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:05:00Z", 12*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:06:00Z", 90*time.Second, tag("type", "break")),
|
||||
ts("2019-06-11T10:08:00Z", 33*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:09:00Z", 21*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:10:00Z", 18*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:11:00Z", 56*time.Second, tag("proj", "traggo"), tag("type", "review"), tag("issue", "11")),
|
||||
ts("2019-06-11T10:12:00Z", 44*time.Second, tag("type", "break")),
|
||||
).Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z",
|
||||
entry("type", "break", 134*time.Second),
|
||||
entry("type", "review", 133*time.Second),
|
||||
entry("type", "support", 45*time.Second),
|
||||
entry("type", "work", 102*time.Second))),
|
||||
name("multiple 4").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 60*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:02:00Z", 7*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:03:00Z", 14*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:04:00Z", 16*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "14")),
|
||||
ts("2019-06-11T10:05:00Z", 33*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:05:00Z", 12*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:06:00Z", 90*time.Second, tag("type", "break")),
|
||||
ts("2019-06-11T10:08:00Z", 32*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:09:00Z", 21*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:10:00Z", 18*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:11:00Z", 56*time.Second, tag("proj", "traggo"), tag("type", "review"), tag("issue", "11")),
|
||||
ts("2019-06-11T10:12:00Z", 44*time.Second, tag("type", "break")),
|
||||
).Keys("type", "issue").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z",
|
||||
entry("issue", "10", 65*time.Second),
|
||||
entry("issue", "11", 56*time.Second),
|
||||
entry("issue", "12", 78*time.Second),
|
||||
entry("issue", "13", 29*time.Second),
|
||||
entry("issue", "14", 16*time.Second),
|
||||
entry("issue", "15", 35*time.Second),
|
||||
entry("type", "break", 134*time.Second),
|
||||
entry("type", "review", 133*time.Second),
|
||||
entry("type", "support", 45*time.Second),
|
||||
entry("type", "work", 101*time.Second))),
|
||||
name("multiple 5").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 60*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:02:00Z", 7*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:03:00Z", 14*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:04:00Z", 16*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "14")),
|
||||
ts("2019-06-11T10:05:00Z", 33*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:05:00Z", 12*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:06:00Z", 90*time.Second, tag("type", "break")),
|
||||
ts("2019-06-11T10:08:00Z", 32*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:09:00Z", 21*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:10:00Z", 18*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:11:00Z", 56*time.Second, tag("proj", "traggo"), tag("type", "review"), tag("issue", "11")),
|
||||
ts("2019-06-11T10:12:00Z", 44*time.Second, tag("type", "break")),
|
||||
).Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z")).
|
||||
Include(tag("issue", "13"), tag("issue", "12")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z",
|
||||
entry("type", "review", 77*time.Second),
|
||||
entry("type", "support", 12*time.Second),
|
||||
entry("type", "work", 18*time.Second))),
|
||||
name("multiple 6").
|
||||
Load(
|
||||
ts("2019-06-11T10:00:00Z", 10*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:01:00Z", 60*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:02:00Z", 7*time.Second, tag("proj", "gotify"), tag("type", "review"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:03:00Z", 14*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:04:00Z", 16*time.Second, tag("proj", "gotify"), tag("type", "work"), tag("issue", "14")),
|
||||
ts("2019-06-11T10:05:00Z", 33*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:05:00Z", 12*time.Second, tag("proj", "gotify"), tag("type", "support"), tag("issue", "13")),
|
||||
ts("2019-06-11T10:06:00Z", 90*time.Second, tag("type", "break")),
|
||||
ts("2019-06-11T10:08:00Z", 32*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "10")),
|
||||
ts("2019-06-11T10:09:00Z", 21*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "15")),
|
||||
ts("2019-06-11T10:10:00Z", 18*time.Second, tag("proj", "traggo"), tag("type", "work"), tag("issue", "12")),
|
||||
ts("2019-06-11T10:11:00Z", 56*time.Second, tag("proj", "traggo"), tag("type", "review"), tag("issue", "11")),
|
||||
ts("2019-06-11T10:12:00Z", 44*time.Second, tag("type", "break")),
|
||||
).Keys("type").
|
||||
Ranges(rangex("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z")).
|
||||
Exclude(tag("issue", "10"), tag("issue", "11"), tag("issue", "14"), tag("issue", "15")).
|
||||
Expected(
|
||||
result("2019-06-11T10:00:00Z", "2019-06-11T10:15:00Z",
|
||||
entry("type", "break", 134*time.Second),
|
||||
entry("type", "review", 77*time.Second),
|
||||
entry("type", "support", 12*time.Second),
|
||||
entry("type", "work", 18*time.Second))),
|
||||
}
|
||||
|
||||
for _, entry := range data {
|
||||
t.Run(entry.name, func(t *testing.T) {
|
||||
db := test.InMemoryDB(t)
|
||||
defer db.Close()
|
||||
|
||||
db.User(1)
|
||||
for _, timespan := range entry.timespans {
|
||||
db.Create(×pan)
|
||||
}
|
||||
|
||||
resolver := ResolverForStatistics{DB: db.DB}
|
||||
|
||||
stats, err := resolver.Stats(fake.User(1), entry.ranges, entry.keys, entry.exclude, entry.include)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, entry.expected, stats)
|
||||
require.Len(t, stats, len(entry.expected))
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func ts(start string, d time.Duration, tags ...model.TimeSpanTag) model.TimeSpan {
|
||||
end := test.Time(start).Add(d)
|
||||
return model.TimeSpan{
|
||||
StartUserTime: test.Time(start),
|
||||
EndUserTime: &end,
|
||||
UserID: 1,
|
||||
Tags: tags,
|
||||
}
|
||||
}
|
||||
|
||||
type testEntry struct {
|
||||
timespans []model.TimeSpan
|
||||
keys []string
|
||||
exclude []*gqlmodel.InputTimeSpanTag
|
||||
include []*gqlmodel.InputTimeSpanTag
|
||||
name string
|
||||
ranges []*gqlmodel.Range
|
||||
expected []*gqlmodel.RangedStatisticsEntries
|
||||
}
|
||||
|
||||
func name(name string) *testEntry {
|
||||
return &testEntry{
|
||||
name: name,
|
||||
}
|
||||
}
|
||||
|
||||
func (e *testEntry) Expected(expected ...*gqlmodel.RangedStatisticsEntries) *testEntry {
|
||||
e.expected = expected
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *testEntry) Keys(keys ...string) *testEntry {
|
||||
e.keys = keys
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *testEntry) Load(timespans ...model.TimeSpan) *testEntry {
|
||||
e.timespans = timespans
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *testEntry) Ranges(ranges ...*gqlmodel.Range) *testEntry {
|
||||
e.ranges = ranges
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *testEntry) Include(includes ...model.TimeSpanTag) *testEntry {
|
||||
for _, entry := range includes {
|
||||
e.include = append(e.include, &gqlmodel.InputTimeSpanTag{
|
||||
Key: entry.Key,
|
||||
Value: entry.StringValue,
|
||||
})
|
||||
}
|
||||
return e
|
||||
}
|
||||
func (e *testEntry) Exclude(excludes ...model.TimeSpanTag) *testEntry {
|
||||
for _, entry := range excludes {
|
||||
e.exclude = append(e.exclude, &gqlmodel.InputTimeSpanTag{
|
||||
Key: entry.Key,
|
||||
Value: entry.StringValue,
|
||||
})
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
func tag(key string, value string) model.TimeSpanTag {
|
||||
return model.TimeSpanTag{
|
||||
Key: key,
|
||||
StringValue: value,
|
||||
}
|
||||
}
|
||||
|
||||
func rangex(start, stop string) *gqlmodel.Range {
|
||||
return &gqlmodel.Range{
|
||||
Start: test.ModelTime(start),
|
||||
End: test.ModelTime(stop),
|
||||
}
|
||||
}
|
||||
|
||||
func result(start, stop string, entries ...*gqlmodel.StatisticsEntry) *gqlmodel.RangedStatisticsEntries {
|
||||
return &gqlmodel.RangedStatisticsEntries{
|
||||
Start: test.ModelTimeUTC(start),
|
||||
End: test.ModelTimeUTC(stop),
|
||||
Entries: entries,
|
||||
}
|
||||
}
|
||||
|
||||
func entry(key string, value string, duration time.Duration) *gqlmodel.StatisticsEntry {
|
||||
return &gqlmodel.StatisticsEntry{
|
||||
Key: key,
|
||||
Value: value,
|
||||
TimeSpendInSeconds: duration.Seconds(),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user