mirror of
https://github.com/percona/percona-toolkit.git
synced 2025-09-21 19:34:52 +00:00
Fixes for PT-61 & PT-62
This commit is contained in:
Carlos Salguero
290
src/go/pt-mongodb-query-digest/main_test.go
Normal file
290
src/go/pt-mongodb-query-digest/main_test.go
Normal file
@@ -0,0 +1,290 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"reflect"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/percona/percona-toolkit/src/go/mongolib/proto"
|
||||
|
||||
"gopkg.in/mgo.v2/dbtest"
|
||||
)
|
||||
|
||||
var Server dbtest.DBServer
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
// The tempdir is created so MongoDB has a location to store its files.
|
||||
// Contents are wiped once the server stops
|
||||
os.Setenv("CHECK_SESSIONS", "0")
|
||||
tempDir, _ := ioutil.TempDir("", "testing")
|
||||
Server.SetPath(tempDir)
|
||||
|
||||
dat, err := ioutil.ReadFile("test/sample/system.profile.json")
|
||||
if err != nil {
|
||||
fmt.Printf("cannot load fixtures: %s", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
var docs []proto.SystemProfile
|
||||
err = json.Unmarshal(dat, &docs)
|
||||
c := Server.Session().DB("samples").C("system_profile")
|
||||
for _, doc := range docs {
|
||||
c.Insert(doc)
|
||||
}
|
||||
|
||||
retCode := m.Run()
|
||||
|
||||
Server.Session().Close()
|
||||
Server.Session().DB("samples").DropDatabase()
|
||||
|
||||
// Stop shuts down the temporary server and removes data on disk.
|
||||
Server.Stop()
|
||||
|
||||
// call with result of m.Run()
|
||||
os.Exit(retCode)
|
||||
}
|
||||
|
||||
func TestCalcStats(t *testing.T) {
|
||||
it := Server.Session().DB("samples").C("system_profile").Find(nil).Sort("Ts").Iter()
|
||||
data := getData(it)
|
||||
s := calcStats(data[0].NScanned)
|
||||
|
||||
want := statistics{Pct: 0, Total: 159, Min: 79, Max: 80, Avg: 79.5, Pct95: 80, StdDev: 0.5, Median: 79.5}
|
||||
|
||||
if !reflect.DeepEqual(s, want) {
|
||||
t.Errorf("error in calcStats: got:\n%#v\nwant:\n%#v\n", s, want)
|
||||
}
|
||||
|
||||
wantTotals := stat{
|
||||
ID: "",
|
||||
Fingerprint: "",
|
||||
Namespace: "",
|
||||
Query: map[string]interface{}(nil),
|
||||
Count: 0,
|
||||
TableScan: false,
|
||||
NScanned: []float64{79, 80},
|
||||
NReturned: []float64{79, 80},
|
||||
QueryTime: []float64{27, 28},
|
||||
ResponseLength: []float64{109, 110},
|
||||
LockTime: nil,
|
||||
BlockedTime: nil,
|
||||
FirstSeen: time.Time{},
|
||||
LastSeen: time.Time{},
|
||||
}
|
||||
|
||||
totals := getTotals(data[0:1])
|
||||
|
||||
if !reflect.DeepEqual(totals, wantTotals) {
|
||||
t.Errorf("error in calcStats: got:\n%#v\nwant:\n:%#v\n", totals, wantTotals)
|
||||
}
|
||||
var wantTotalCount int = 2
|
||||
var wantTotalScanned, wantTotalReturned, wantTotalQueryTime, wantTotalBytes float64 = 159, 159, 55, 219
|
||||
|
||||
totalCount, totalScanned, totalReturned, totalQueryTime, totalBytes := calcTotals(data[0:1])
|
||||
|
||||
if totalCount != wantTotalCount {
|
||||
t.Errorf("invalid total count. Want %v, got %v\n", wantTotalCount, totalCount)
|
||||
}
|
||||
|
||||
if totalScanned != wantTotalScanned {
|
||||
t.Errorf("invalid total count. Want %v, got %v\n", wantTotalScanned, totalScanned)
|
||||
}
|
||||
if totalReturned != wantTotalReturned {
|
||||
t.Errorf("invalid total count. Want %v, got %v\n", wantTotalReturned, totalReturned)
|
||||
}
|
||||
if totalQueryTime != wantTotalQueryTime {
|
||||
t.Errorf("invalid total count. Want %v, got %v\n", wantTotalQueryTime, totalQueryTime)
|
||||
}
|
||||
if totalBytes != wantTotalBytes {
|
||||
t.Errorf("invalid total count. Want %v, got %v\n", wantTotalBytes, totalBytes)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetData(t *testing.T) {
|
||||
it := Server.Session().DB("samples").C("system_profile").Find(nil).Iter()
|
||||
tests := []struct {
|
||||
name string
|
||||
i iter
|
||||
want []stat
|
||||
}{
|
||||
{
|
||||
name: "test 1",
|
||||
i: it,
|
||||
want: []stat{
|
||||
stat{
|
||||
ID: "6c3fff4804febd156700a06f9a346162",
|
||||
Fingerprint: "find,limit",
|
||||
Namespace: "samples.col1",
|
||||
Query: map[string]interface{}{
|
||||
"find": "col1",
|
||||
"limit": float64(2),
|
||||
},
|
||||
Count: 2,
|
||||
TableScan: false,
|
||||
NScanned: []float64{79, 80},
|
||||
NReturned: []float64{79, 80},
|
||||
QueryTime: []float64{27, 28},
|
||||
ResponseLength: []float64{109, 110},
|
||||
LockTime: nil,
|
||||
BlockedTime: nil,
|
||||
FirstSeen: time.Date(2016, time.November, 8, 13, 46, 27, 0, time.UTC).Local(),
|
||||
LastSeen: time.Date(2016, time.November, 8, 13, 46, 27, 0, time.UTC).Local(),
|
||||
},
|
||||
|
||||
stat{
|
||||
ID: "fdcea004122ddb225bc56de417391e25",
|
||||
Fingerprint: "find",
|
||||
Namespace: "samples.col1",
|
||||
Query: map[string]interface{}{"find": "col1"},
|
||||
Count: 8,
|
||||
TableScan: false,
|
||||
NScanned: []float64{71, 72, 73, 74, 75, 76, 77, 78},
|
||||
NReturned: []float64{71, 72, 73, 74, 75, 76, 77, 78},
|
||||
QueryTime: []float64{19, 20, 21, 22, 23, 24, 25, 26},
|
||||
ResponseLength: []float64{101, 102, 103, 104, 105, 106, 107, 108},
|
||||
LockTime: nil,
|
||||
BlockedTime: nil,
|
||||
FirstSeen: time.Date(2016, time.November, 8, 13, 46, 27, 0, time.UTC).Local(),
|
||||
LastSeen: time.Date(2016, time.November, 8, 13, 46, 27, 0, time.UTC).Local(),
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
got := getData(tt.i)
|
||||
if !reflect.DeepEqual(got, tt.want) {
|
||||
t.Errorf("got\n%#v\nwant\n%#v", got, tt.want)
|
||||
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestUptime(t *testing.T) {
|
||||
session := Server.Session()
|
||||
time.Sleep(1500 * time.Millisecond)
|
||||
if uptime(session) <= 0 {
|
||||
t.Error("uptime is 0")
|
||||
}
|
||||
session.Close()
|
||||
|
||||
}
|
||||
|
||||
func TestFingerprint(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
query map[string]interface{}
|
||||
want string
|
||||
}{
|
||||
{
|
||||
query: map[string]interface{}{"query": map[string]interface{}{}, "orderby": map[string]interface{}{"ts": -1}},
|
||||
want: "orderby,query,ts",
|
||||
},
|
||||
{
|
||||
query: map[string]interface{}{"find": "system.profile", "filter": map[string]interface{}{}, "sort": map[string]interface{}{"$natural": 1}},
|
||||
want: "$natural,filter,find,sort",
|
||||
},
|
||||
{
|
||||
|
||||
query: map[string]interface{}{"collection": "system.profile", "batchSize": 0, "getMore": 18531768265},
|
||||
want: "batchSize,collection,getMore",
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := fingerprint(tt.query); got != tt.want {
|
||||
t.Errorf("fingerprint() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimesLen(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
a times
|
||||
want int
|
||||
}{
|
||||
{
|
||||
name: "Times.Len",
|
||||
a: []time.Time{time.Now()},
|
||||
want: 1,
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := tt.a.Len(); got != tt.want {
|
||||
t.Errorf("times.Len() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimesSwap(t *testing.T) {
|
||||
type args struct {
|
||||
i int
|
||||
j int
|
||||
}
|
||||
t1 := time.Now()
|
||||
t2 := t1.Add(1 * time.Minute)
|
||||
tests := []struct {
|
||||
name string
|
||||
a times
|
||||
args args
|
||||
}{
|
||||
{
|
||||
name: "Times.Swap",
|
||||
a: times{t1, t2},
|
||||
args: args{i: 0, j: 1},
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
tt.a.Swap(tt.args.i, tt.args.j)
|
||||
if tt.a[0] != t2 || tt.a[1] != t1 {
|
||||
t.Errorf("%s has (%v, %v) want (%v, %v)", tt.name, tt.a[0], tt.a[1], t2, t1)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimesLess(t *testing.T) {
|
||||
type args struct {
|
||||
i int
|
||||
j int
|
||||
}
|
||||
t1 := time.Now()
|
||||
t2 := t1.Add(1 * time.Minute)
|
||||
tests := []struct {
|
||||
name string
|
||||
a times
|
||||
args args
|
||||
want bool
|
||||
}{
|
||||
{
|
||||
name: "Times.Swap",
|
||||
a: times{t1, t2},
|
||||
args: args{i: 0, j: 1},
|
||||
want: true,
|
||||
},
|
||||
{
|
||||
name: "Times.Swap",
|
||||
a: times{t2, t1},
|
||||
args: args{i: 0, j: 1},
|
||||
want: false,
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := tt.a.Less(tt.args.i, tt.args.j); got != tt.want {
|
||||
t.Errorf("times.Less() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user