-
Notifications
You must be signed in to change notification settings - Fork 9
/
eventsdbmodule_test.go
63 lines (50 loc) · 1.88 KB
/
eventsdbmodule_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package keycloakb
import (
"context"
"database/sql"
"testing"
errorhandler "github.com/cloudtrust/common-service/errors"
api "github.com/cloudtrust/keycloak-bridge/api/events"
"github.com/cloudtrust/keycloak-bridge/pkg/events/mock"
"github.com/golang/mock/gomock"
"github.com/stretchr/testify/assert"
)
func TestModuleGetEvents(t *testing.T) {
var mockCtrl = gomock.NewController(t)
defer mockCtrl.Finish()
dbEvents := mock.NewDBEvents(mockCtrl)
module := NewEventsDBModule(dbEvents)
params := map[string]string{"origin": "origin-1", "max": "5"}
var empty [0]api.AuditRepresentation
var expectedResult = empty[:]
var expectedError error = errorhandler.CreateMissingParameterError("")
var rows sql.Rows
dbEvents.EXPECT().Query(gomock.Any(), params["origin"], nil, nil, nil, nil, nil, 0, params["max"]).Return(&rows, expectedError).Times(1)
res, err := module.GetEvents(context.Background(), params)
assert.Equal(t, expectedResult, res)
assert.Equal(t, expectedError, err)
}
func TestModuleGetEventsSummary(t *testing.T) {
var mockCtrl = gomock.NewController(t)
defer mockCtrl.Finish()
dbEvents := mock.NewDBEvents(mockCtrl)
module := NewEventsDBModule(dbEvents)
var expectedResult api.EventSummaryRepresentation
var expectedError error = errorhandler.CreateMissingParameterError("")
var rows sql.Rows
dbEvents.EXPECT().Query(gomock.Any()).Return(&rows, expectedError).Times(1)
res, err := module.GetEventsSummary(context.Background())
assert.Equal(t, expectedResult, res)
assert.Equal(t, expectedError, err)
}
func TestModuleGetTotalConnectionsCount(t *testing.T) {
var mockCtrl = gomock.NewController(t)
defer mockCtrl.Finish()
dbEvents := mock.NewDBEvents(mockCtrl)
module := NewEventsDBModule(dbEvents)
// Check SQL injection
{
_, err := module.GetTotalConnectionsCount(context.TODO(), "realm", "1 DAY'; TRUNCATE TABLE PASSWORD; select '")
assert.NotNil(t, err)
}
}