mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
feat: expose Everyone group through UI (#9117)
- Allows setting quota allowances on the 'Everyone' group.
This commit is contained in:
81
coderd/database/tx_test.go
Normal file
81
coderd/database/tx_test.go
Normal file
@ -0,0 +1,81 @@
|
||||
package database_test
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"testing"
|
||||
|
||||
"github.com/golang/mock/gomock"
|
||||
"github.com/lib/pq"
|
||||
"github.com/stretchr/testify/require"
|
||||
"golang.org/x/xerrors"
|
||||
|
||||
"github.com/coder/coder/coderd/database"
|
||||
"github.com/coder/coder/coderd/database/dbmock"
|
||||
)
|
||||
|
||||
func TestReadModifyUpdate_OK(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mDB := dbmock.NewMockStore(gomock.NewController(t))
|
||||
|
||||
mDB.EXPECT().
|
||||
InTx(gomock.Any(), &sql.TxOptions{Isolation: sql.LevelRepeatableRead}).
|
||||
Times(1).
|
||||
Return(nil)
|
||||
err := database.ReadModifyUpdate(mDB, func(tx database.Store) error {
|
||||
return nil
|
||||
})
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestReadModifyUpdate_RetryOK(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mDB := dbmock.NewMockStore(gomock.NewController(t))
|
||||
|
||||
firstUpdate := mDB.EXPECT().
|
||||
InTx(gomock.Any(), &sql.TxOptions{Isolation: sql.LevelRepeatableRead}).
|
||||
Times(1).
|
||||
Return(&pq.Error{Code: pq.ErrorCode("40001")})
|
||||
mDB.EXPECT().
|
||||
InTx(gomock.Any(), &sql.TxOptions{Isolation: sql.LevelRepeatableRead}).
|
||||
After(firstUpdate).
|
||||
Times(1).
|
||||
Return(nil)
|
||||
|
||||
err := database.ReadModifyUpdate(mDB, func(tx database.Store) error {
|
||||
return nil
|
||||
})
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestReadModifyUpdate_HardError(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mDB := dbmock.NewMockStore(gomock.NewController(t))
|
||||
|
||||
mDB.EXPECT().
|
||||
InTx(gomock.Any(), &sql.TxOptions{Isolation: sql.LevelRepeatableRead}).
|
||||
Times(1).
|
||||
Return(xerrors.New("a bad thing happened"))
|
||||
|
||||
err := database.ReadModifyUpdate(mDB, func(tx database.Store) error {
|
||||
return nil
|
||||
})
|
||||
require.ErrorContains(t, err, "a bad thing happened")
|
||||
}
|
||||
|
||||
func TestReadModifyUpdate_TooManyRetries(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mDB := dbmock.NewMockStore(gomock.NewController(t))
|
||||
|
||||
mDB.EXPECT().
|
||||
InTx(gomock.Any(), &sql.TxOptions{Isolation: sql.LevelRepeatableRead}).
|
||||
Times(5).
|
||||
Return(&pq.Error{Code: pq.ErrorCode("40001")})
|
||||
err := database.ReadModifyUpdate(mDB, func(tx database.Store) error {
|
||||
return nil
|
||||
})
|
||||
require.ErrorContains(t, err, "too many errors")
|
||||
}
|
Reference in New Issue
Block a user