fix: Filters from list and regex both match (#2)
Signed-off-by: Julien Riou <julien@riou.xyz>
This commit is contained in:
parent
763359c3d6
commit
9cff851021
6 changed files with 254 additions and 41 deletions
139
base/session_test.go
Normal file
139
base/session_test.go
Normal file
|
|
@ -0,0 +1,139 @@
|
|||
package base
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
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)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue