mirror of
https://github.com/twofas/2fas-server.git
synced 2024-12-12 04:00:15 +01:00
55 lines
2.4 KiB
Go
55 lines
2.4 KiB
Go
|
package tests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
"github.com/twofas/2fas-server/e2e-tests"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMobileDeviceExtensionIntegrationTestSuite(t *testing.T) {
|
||
|
|
suite.Run(t, new(MobileDeviceExtensionIntegrationTestSuite))
|
||
|
|
}
|
||
|
|
|
||
|
|
type MobileDeviceExtensionIntegrationTestSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MobileDeviceExtensionIntegrationTestSuite) SetupTest() {
|
||
|
|
e2e_tests.RemoveAllMobileDevices(s.T())
|
||
|
|
e2e_tests.RemoveAllBrowserExtensions(s.T())
|
||
|
|
e2e_tests.RemoveAllBrowserExtensionsDevices(s.T())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MobileDeviceExtensionIntegrationTestSuite) TestGetPending2FaRequests() {
|
||
|
|
device, devicePubKey := e2e_tests.CreateDevice(s.T(), "SM-955F", "fcm-token")
|
||
|
|
browserExtension := e2e_tests.CreateBrowserExtension(s.T(), "go-ext")
|
||
|
|
e2e_tests.PairDeviceWithBrowserExtension(s.T(), devicePubKey, browserExtension, device)
|
||
|
|
|
||
|
|
var tokenRequest *e2e_tests.AuthTokenRequestResponse
|
||
|
|
request2FaTokenPayload := []byte(`{"domain":"domain.com"}`)
|
||
|
|
e2e_tests.DoAPISuccessPost(s.T(), "browser_extensions/"+browserExtension.Id+"/commands/request_2fa_token", request2FaTokenPayload, &tokenRequest)
|
||
|
|
|
||
|
|
var tokenRequestsCollection []*e2e_tests.AuthTokenRequestResponse
|
||
|
|
e2e_tests.DoAPISuccessGet(s.T(), "mobile/devices/"+device.Id+"/browser_extensions/2fa_requests", &tokenRequestsCollection)
|
||
|
|
assert.Len(s.T(), tokenRequestsCollection, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MobileDeviceExtensionIntegrationTestSuite) TestDoNotReturnCompleted2FaRequests() {
|
||
|
|
device, devicePubKey := e2e_tests.CreateDevice(s.T(), "SM-955F", "fcm-token")
|
||
|
|
browserExtension := e2e_tests.CreateBrowserExtension(s.T(), "go-ext")
|
||
|
|
e2e_tests.PairDeviceWithBrowserExtension(s.T(), devicePubKey, browserExtension, device)
|
||
|
|
|
||
|
|
var tokenRequest *e2e_tests.AuthTokenRequestResponse
|
||
|
|
request2FaTokenPayload := []byte(`{"domain":"domain.com"}`)
|
||
|
|
e2e_tests.DoAPISuccessPost(s.T(), "browser_extensions/"+browserExtension.Id+"/commands/request_2fa_token", request2FaTokenPayload, &tokenRequest)
|
||
|
|
|
||
|
|
closeTokenRequestPayload := []byte(`{"status":"completed"}`)
|
||
|
|
e2e_tests.DoAPISuccessPost(s.T(), "browser_extensions/"+browserExtension.Id+"/2fa_requests/"+tokenRequest.Id+"/commands/close_2fa_request", closeTokenRequestPayload, nil)
|
||
|
|
|
||
|
|
var tokenRequestsCollection []*e2e_tests.AuthTokenRequestResponse
|
||
|
|
e2e_tests.DoAPISuccessGet(s.T(), "mobile/devices/"+device.Id+"/browser_extensions/2fa_requests", &tokenRequestsCollection)
|
||
|
|
assert.Len(s.T(), tokenRequestsCollection, 0)
|
||
|
|
}
|