mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 20:08:02 +00:00
127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package locker
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
)
|
|
|
|
type dbMock struct {
|
|
db *sql.DB
|
|
mock sqlmock.Sqlmock
|
|
}
|
|
|
|
func mockDB(t *testing.T) *dbMock {
|
|
mockDB := dbMock{}
|
|
var err error
|
|
mockDB.db, mockDB.mock, err = sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("error occured while creating stub db %v", err)
|
|
}
|
|
|
|
mockDB.mock.MatchExpectationsInOrder(true)
|
|
|
|
return &mockDB
|
|
}
|
|
|
|
func (db *dbMock) expectCommit() *dbMock {
|
|
db.mock.ExpectCommit()
|
|
|
|
return db
|
|
}
|
|
|
|
func (db *dbMock) expectRollback() *dbMock {
|
|
db.mock.ExpectRollback()
|
|
|
|
return db
|
|
}
|
|
|
|
func (db *dbMock) expectBegin() *dbMock {
|
|
db.mock.ExpectBegin()
|
|
|
|
return db
|
|
}
|
|
|
|
func (db *dbMock) expectSavepoint() *dbMock {
|
|
db.mock.ExpectExec("SAVEPOINT").WillReturnResult(sqlmock.NewResult(1, 1))
|
|
return db
|
|
}
|
|
|
|
func (db *dbMock) expectReleaseSavepoint() *dbMock {
|
|
db.mock.ExpectExec("RELEASE SAVEPOINT").WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
return db
|
|
}
|
|
|
|
func (db *dbMock) expectRenew(lockerID, view, instanceID string, affectedRows int64) *dbMock {
|
|
query := db.mock.
|
|
ExpectExec(`INSERT INTO table\.locks \(locker_id, locked_until, view_name, instance_id\) VALUES \(\$1, now\(\)\+\$2::INTERVAL, \$3\, \$4\) ON CONFLICT \(view_name, instance_id\) DO UPDATE SET locker_id = \$1, locked_until = now\(\)\+\$2::INTERVAL WHERE locks\.view_name = \$3 AND locks\.instance_id = \$4 AND \(locks\.locker_id = \$1 OR locks\.locked_until < now\(\)\)`).
|
|
WithArgs(lockerID, sqlmock.AnyArg(), view, instanceID).
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
if affectedRows == 0 {
|
|
query.WillReturnResult(sqlmock.NewResult(0, 0))
|
|
} else {
|
|
query.WillReturnResult(sqlmock.NewResult(1, affectedRows))
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func Test_locker_Renew(t *testing.T) {
|
|
type fields struct {
|
|
db *dbMock
|
|
}
|
|
type args struct {
|
|
tableName string
|
|
lockerID string
|
|
viewModel string
|
|
instanceID string
|
|
waitTime time.Duration
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "renew succeeded",
|
|
fields: fields{
|
|
db: mockDB(t).
|
|
expectBegin().
|
|
expectSavepoint().
|
|
expectRenew("locker", "view", "instanceID", 1).
|
|
expectReleaseSavepoint().
|
|
expectCommit(),
|
|
},
|
|
args: args{tableName: "table.locks", lockerID: "locker", viewModel: "view", instanceID: "instanceID", waitTime: 1 * time.Second},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "renew now rows updated",
|
|
fields: fields{
|
|
db: mockDB(t).
|
|
expectBegin().
|
|
expectSavepoint().
|
|
expectRenew("locker", "view", "instanceID", 0).
|
|
expectRollback(),
|
|
},
|
|
args: args{tableName: "table.locks", lockerID: "locker", viewModel: "view", instanceID: "instanceID", waitTime: 1 * time.Second},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if err := Renew(tt.fields.db.db, tt.args.tableName, tt.args.lockerID, tt.args.viewModel, tt.args.instanceID, tt.args.waitTime); (err != nil) != tt.wantErr {
|
|
t.Errorf("locker.Renew() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
if err := tt.fields.db.mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("not all database expectations met: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|