2022-02-16 12:30:49 +00:00
|
|
|
package initialise
|
|
|
|
|
|
|
|
import (
|
|
|
|
"database/sql"
|
|
|
|
"errors"
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
|
|
|
func Test_verifyUser(t *testing.T) {
|
2022-08-31 07:52:43 +00:00
|
|
|
err := ReadStmts("cockroach") //TODO: check all dialects
|
|
|
|
if err != nil {
|
|
|
|
t.Errorf("unable to read stmts: %v", err)
|
|
|
|
t.FailNow()
|
|
|
|
}
|
|
|
|
|
2022-02-16 12:30:49 +00:00
|
|
|
type args struct {
|
2022-03-15 06:19:02 +00:00
|
|
|
db db
|
|
|
|
username string
|
|
|
|
password string
|
2022-02-16 12:30:49 +00:00
|
|
|
}
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
targetErr error
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
name: "doesn't exists, create fails",
|
|
|
|
args: args{
|
|
|
|
db: prepareDB(t,
|
2022-08-31 07:52:43 +00:00
|
|
|
expectExec("-- replace zitadel-user with the name of the user\nCREATE USER IF NOT EXISTS zitadel-user", sql.ErrTxDone),
|
2022-02-16 12:30:49 +00:00
|
|
|
),
|
2022-03-15 06:19:02 +00:00
|
|
|
username: "zitadel-user",
|
|
|
|
password: "",
|
2022-02-16 12:30:49 +00:00
|
|
|
},
|
|
|
|
targetErr: sql.ErrTxDone,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "correct without password",
|
|
|
|
args: args{
|
|
|
|
db: prepareDB(t,
|
2022-08-31 07:52:43 +00:00
|
|
|
expectExec("-- replace zitadel-user with the name of the user\nCREATE USER IF NOT EXISTS zitadel-user", nil),
|
2022-02-16 12:30:49 +00:00
|
|
|
),
|
2022-03-15 06:19:02 +00:00
|
|
|
username: "zitadel-user",
|
|
|
|
password: "",
|
2022-02-16 12:30:49 +00:00
|
|
|
},
|
|
|
|
targetErr: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "correct with password",
|
|
|
|
args: args{
|
|
|
|
db: prepareDB(t,
|
2022-08-31 07:52:43 +00:00
|
|
|
expectExec("-- replace zitadel-user with the name of the user\nCREATE USER IF NOT EXISTS zitadel-user WITH PASSWORD 'password'", nil),
|
2022-02-16 12:30:49 +00:00
|
|
|
),
|
2022-03-15 06:19:02 +00:00
|
|
|
username: "zitadel-user",
|
|
|
|
password: "password",
|
2022-02-16 12:30:49 +00:00
|
|
|
},
|
|
|
|
targetErr: nil,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "already exists",
|
|
|
|
args: args{
|
|
|
|
db: prepareDB(t,
|
2022-08-31 07:52:43 +00:00
|
|
|
expectExec("-- replace zitadel-user with the name of the user\nCREATE USER IF NOT EXISTS zitadel-user WITH PASSWORD 'password'", nil),
|
2022-02-16 12:30:49 +00:00
|
|
|
),
|
2022-03-15 06:19:02 +00:00
|
|
|
username: "zitadel-user",
|
|
|
|
password: "",
|
2022-02-16 12:30:49 +00:00
|
|
|
},
|
|
|
|
targetErr: nil,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
2022-03-15 06:19:02 +00:00
|
|
|
if err := VerifyUser(tt.args.username, tt.args.password)(tt.args.db.db); !errors.Is(err, tt.targetErr) {
|
|
|
|
t.Errorf("VerifyGrant() error = %v, want: %v", err, tt.targetErr)
|
2022-02-16 12:30:49 +00:00
|
|
|
}
|
|
|
|
if err := tt.args.db.mock.ExpectationsWereMet(); err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|