Update tracker test

This commit is contained in:
Alan Bernstein 2020-10-21 11:07:33 -05:00 committed by Alan Bernstein
parent c94d242097
commit 65de5df2a9
2 changed files with 7 additions and 4 deletions

View file

@ -143,7 +143,7 @@ func (t *queryTracker) ActiveQueries() []ActiveQueryStatus {
func (t *queryTracker) PastQueries() []PastQueryStatus {
queries := make([]pastQuery, 0, len(t.history))
for pq, _ := range t.history {
for pq := range t.history {
queries = append(queries, pq)
}
// TODO use sort.Sort

View file

@ -14,7 +14,10 @@
package pilosa
import "testing"
import (
"testing"
"time"
)
func TestQueryTracker(t *testing.T) {
tracker := newQueryTracker()
@ -24,7 +27,7 @@ func TestQueryTracker(t *testing.T) {
t.Fatalf("expected no active queries; found %v", queries)
}
qs := tracker.Start("test query")
qs := tracker.Start("test query", "node0")
var queries []ActiveQueryStatus
for len(queries) < 1 {
@ -34,7 +37,7 @@ func TestQueryTracker(t *testing.T) {
t.Fatalf("unexpected queries: %v", queries)
}
tracker.Finish(qs)
tracker.Finish(qs, time.Now())
for len(queries) > 0 {
queries = tracker.ActiveQueries()