162 lines
2.7 KiB
Go
162 lines
2.7 KiB
Go
package base
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func TestSessionJSON(t *testing.T) {
|
|
session := &Session{
|
|
Pid: 1,
|
|
User: "test",
|
|
Db: "db",
|
|
Client: "127.0.0.1:5432",
|
|
State: "active",
|
|
Query: "select 1",
|
|
StateDuration: 1.5,
|
|
ApplicationName: "psql",
|
|
}
|
|
|
|
var got Session
|
|
if err := json.Unmarshal([]byte(session.JSON()), &got); err != nil {
|
|
t.Fatalf("invalid JSON: %v", err)
|
|
}
|
|
|
|
if got != *session {
|
|
t.Errorf("got %+v; want %+v", got, *session)
|
|
}
|
|
}
|
|
|
|
func TestSessionEqual(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
first *Session
|
|
second *Session
|
|
want bool
|
|
}{
|
|
{
|
|
"Empty sessions",
|
|
&Session{},
|
|
&Session{},
|
|
true,
|
|
},
|
|
{
|
|
"Identical process id",
|
|
&Session{Pid: 1},
|
|
&Session{Pid: 1},
|
|
true,
|
|
},
|
|
{
|
|
"Different process id",
|
|
&Session{Pid: 1},
|
|
&Session{Pid: 2},
|
|
false,
|
|
},
|
|
{
|
|
"Identical users",
|
|
&Session{User: "test"},
|
|
&Session{User: "test"},
|
|
true,
|
|
},
|
|
{
|
|
"Different users",
|
|
&Session{User: "test"},
|
|
&Session{User: "random"},
|
|
false,
|
|
},
|
|
{
|
|
"Identical databases",
|
|
&Session{Db: "test"},
|
|
&Session{Db: "test"},
|
|
true,
|
|
},
|
|
{
|
|
"Different databases",
|
|
&Session{Db: "test"},
|
|
&Session{Db: "random"},
|
|
false,
|
|
},
|
|
{
|
|
"Identical users and databases",
|
|
&Session{User: "test", Db: "test"},
|
|
&Session{User: "test", Db: "test"},
|
|
true,
|
|
},
|
|
{
|
|
"Different users and same databases",
|
|
&Session{User: "test_1", Db: "test"},
|
|
&Session{User: "test_2", Db: "test"},
|
|
false,
|
|
},
|
|
{
|
|
"Different databases and same user",
|
|
&Session{User: "test", Db: "test_1"},
|
|
&Session{User: "test", Db: "test_2"},
|
|
false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := tc.first.Equal(tc.second)
|
|
if got != tc.want {
|
|
t.Errorf("got %t; want %t", got, tc.want)
|
|
} else {
|
|
t.Logf("got %t; want %t", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSessionInSlice(t *testing.T) {
|
|
sessions := []*Session{
|
|
{User: "test"},
|
|
{User: "test_1"},
|
|
{User: "test_2"},
|
|
{User: "postgres"},
|
|
{Db: "test"},
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
input *Session
|
|
want bool
|
|
}{
|
|
{
|
|
"Empty session",
|
|
&Session{},
|
|
false,
|
|
},
|
|
{
|
|
"Session with user in slice",
|
|
&Session{User: "test"},
|
|
true,
|
|
},
|
|
{
|
|
"Session with user not in slice",
|
|
&Session{User: "random"},
|
|
false,
|
|
},
|
|
{
|
|
"Session with db in slice",
|
|
&Session{Db: "test"},
|
|
true,
|
|
},
|
|
{
|
|
"Session with db not in slice",
|
|
&Session{Db: "random"},
|
|
false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := tc.input.InSlice(sessions)
|
|
if got != tc.want {
|
|
t.Errorf("got %t; want %t", got, tc.want)
|
|
} else {
|
|
t.Logf("got %t; want %t", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|