mirror of
https://github.com/0rangebananaspy/authelia.git
synced 2024-09-14 22:47:21 +07:00
83488d52a6
* refactor(suites): replace selenium with go-rod This change replaces [tebeka/selenium](https://github.com/tebeka/selenium) with [go-rod](https://github.com/go-rod/rod). We no longer have a chromedriver/external driver dependency to utilise Selenium as we instead utilise the Chrome Dev Protocol to communicate with the browser. Rod [documents](https://go-rod.github.io/#/why-rod) benefits of choosing the library as opposed to the available alternatives.
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package suites
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-rod/rod"
|
|
"github.com/pquerna/otp/totp"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func (rs *RodSession) doRegisterTOTP(t *testing.T, page *rod.Page) string {
|
|
err := rs.WaitElementLocatedByCSSSelector(t, page, "register-link").Click("left")
|
|
require.NoError(t, err)
|
|
rs.verifyMailNotificationDisplayed(t, page)
|
|
link := doGetLinkFromLastMail(t)
|
|
rs.doVisit(t, page, link)
|
|
secretURL, err := page.MustElement("#secret-url").Attribute("value")
|
|
assert.NoError(t, err)
|
|
|
|
secret := (*secretURL)[strings.LastIndex(*secretURL, "=")+1:]
|
|
assert.NotEqual(t, "", secret)
|
|
assert.NotNil(t, secret)
|
|
|
|
return secret
|
|
}
|
|
|
|
func (rs *RodSession) doEnterOTP(t *testing.T, page *rod.Page, code string) {
|
|
inputs := rs.WaitElementsLocatedByCSSSelector(t, page, "otp-input input")
|
|
|
|
for i := 0; i < 6; i++ {
|
|
_ = inputs[i].Input(string(code[i]))
|
|
}
|
|
}
|
|
|
|
func (rs *RodSession) doValidateTOTP(t *testing.T, page *rod.Page, secret string) {
|
|
code, err := totp.GenerateCode(secret, time.Now())
|
|
assert.NoError(t, err)
|
|
rs.doEnterOTP(t, page, code)
|
|
}
|