2fas-server/tests/pass/sync_test.go

package pass

import (
	"testing"

	"github.com/google/uuid"
)

func TestSyncHappyFlow(t *testing.T) {
	resp, err := configureBrowserExtension()
	if err != nil {
		t.Fatalf("Failed to configure browser extension: %v", err)
	}

	browserExtensionDone := make(chan struct{})
	mobileParingDone := make(chan struct{})

	fcm := uuid.NewString()

	go func() {
		defer close(browserExtensionDone)
		_, syncToken, err := browserExtensionWaitForConfirm(resp.BrowserExtensionPairingToken)
		if err != nil {
			t.Errorf("Error when Browser Extension waited for pairing confirm: %v", err)
			return
		}

		proxyToken, err := browserExtensionWaitForSyncConfirm(syncToken)
		if err != nil {
			t.Errorf("Error when Browser Extension waited for sync confirm: %v", err)
			return
		}

		err = proxyWebSocket(
			getWsURL()+"/browser_extension/sync/proxy",
			proxyToken,
			"sent from browser extension",
			"sent from mobile")
		if err != nil {
			t.Errorf("Browser Extension: proxy failed: %v", err)
			return
		}
	}()
	go func() {
		defer close(mobileParingDone)

		_, err := confirmMobile(resp.ConnectionToken, fcm)
		if err != nil {
			t.Errorf("Mobile: confirm failed: %v", err)
			return
		}

		confirmToken, err := getMobileToken(fcm)
		if err != nil {
			t.Errorf("Failed to fetch mobile token: %v", err)
			return
		}

		proxyToken, err := confirmSyncMobile(confirmToken)
		if err != nil {
			t.Errorf("Failed to confirm mobile: %v", err)
			return
		}
		if proxyToken == "" {
			t.Errorf("Mobile: proxy token is empty")
			return
		}

		err = proxyWebSocket(
			getWsURL()+"/mobile/sync/proxy",
			proxyToken,
			"sent from mobile",
			"sent from browser extension",
		)
		if err != nil {
			t.Errorf("Mobile: proxy failed: %v", err)
			return
		}
	}()

	<-browserExtensionDone
	<-mobileParingDone
}