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