2FAS/2fas-server
1
0
Fork 0
mirror of https://github.com/twofas/2fas-server.git synced 2024-12-04 16:20:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: configurable tests

Browse Source
This commit is contained in:
Tobiasz Heller 2024-02-07 16:58:52 +01:00
parent 7eeedb39cb
commit 5e4f84af10
1 changed files with 34 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
tests/pass/pass_test.go
View File

@ -25,7 +25,23 @@ var (
wsDialer = websocket.DefaultDialer wsDialer = websocket.DefaultDialer
) )
func getAPIURL() string { func getApiURL() string {
apiURL := os.Getenv("API_URL")
if apiURL != "" {
return apiURL
}
return "http://" + getPassAddr()
}
func getWsURL() string {
wsURL := os.Getenv("WS_URL")
if wsURL != "" {
return wsURL
}
return "ws://" + getPassAddr()
}
func getPassAddr() string {
addr := os.Getenv("PASS_ADDR") addr := os.Getenv("PASS_ADDR")
if addr != "" { if addr != "" {
return addr return addr
@ -52,7 +68,7 @@ func TestPassHappyFlow(t *testing.T) {
} }
err = proxyWebSocket( err = proxyWebSocket(
"ws://"+getAPIURL()+"/browser_extension/proxy_to_mobile", getWsURL()+"/browser_extension/proxy_to_mobile",
extProxyToken, extProxyToken,
"sent from browser extension", "sent from browser extension",
"sent from mobile") "sent from mobile")
@ -72,7 +88,7 @@ func TestPassHappyFlow(t *testing.T) {
} }
err = proxyWebSocket( err = proxyWebSocket(
"ws://"+getAPIURL()+"/mobile/proxy_to_browser_extension", getWsURL()+"/mobile/proxy_to_browser_extension",
mobileProxyToken, mobileProxyToken,
"sent from mobile", "sent from mobile",
"sent from browser extension", "sent from browser extension",
@ -87,7 +103,7 @@ func TestPassHappyFlow(t *testing.T) {
} }
func browserExtensionWaitForConfirm(token string) (string, error) { func browserExtensionWaitForConfirm(token string) (string, error) {
url := "ws://" + getAPIURL() + "/browser_extension/wait_for_connection" url := getWsURL() + "/browser_extension/wait_for_connection"
var resp struct { var resp struct {
BrowserExtensionProxyToken string `json:"browser_extension_proxy_token"` BrowserExtensionProxyToken string `json:"browser_extension_proxy_token"`
@ -117,9 +133,14 @@ func browserExtensionWaitForConfirm(token string) (string, error) {
} }
func configureBrowserExtension() (ConfigureBrowserExtensionResponse, error) { func configureBrowserExtension() (ConfigureBrowserExtensionResponse, error) {
url := "http://" + getAPIURL() + "/browser_extension/configure" url := getApiURL() + "/browser_extension/configure"
req, err := http.NewRequest("POST", url, bytesPrintf(`{"extension_id":"%s"}`, uuid.New().String())) extensionID := uuid.NewString()
if extensionIDFromEnv := os.Getenv("TEST_EXTENSION_ID"); extensionIDFromEnv != "" {
extensionID = extensionIDFromEnv
}
req, err := http.NewRequest("POST", url, bytesPrintf(`{"extension_id":"%s"}`, extensionID))
if err != nil { if err != nil {
return ConfigureBrowserExtensionResponse{}, fmt.Errorf("failed to create http request: %w", err) return ConfigureBrowserExtensionResponse{}, fmt.Errorf("failed to create http request: %w", err)
} }
@ -148,9 +169,14 @@ func configureBrowserExtension() (ConfigureBrowserExtensionResponse, error) {
// confirmMobile confirms pairing and returns mobile proxy token. // confirmMobile confirms pairing and returns mobile proxy token.
func confirmMobile(connectionToken string) (string, error) { func confirmMobile(connectionToken string) (string, error) {
url := "http://" + getAPIURL() + "/mobile/confirm" url := getApiURL() + "/mobile/confirm"
req, err := http.NewRequest("POST", url, bytesPrintf(`{"device_id":"%s"}`, uuid.New().String())) deviceID := uuid.NewString()
if deviceIDFromEnv := os.Getenv("TEST_DEVICE_ID"); deviceIDFromEnv != "" {
deviceID = deviceIDFromEnv
}
req, err := http.NewRequest("POST", url, bytesPrintf(`{"device_id":"%s"}`, deviceID))
if err != nil { if err != nil {
return "", fmt.Errorf("failed to prepare the reqest: %w", err) return "", fmt.Errorf("failed to prepare the reqest: %w", err)
} }