Function waitForConnectionState

internal/api/syncapi/syncapi_test.go:638–679

(t *testing.T, ctx context.Context, peer *peerUnderTest, hostPeer *v1.Multihost_Peer, wantState v1sync.ConnectionState)

}

func waitForConnectionState(t *testing.T, ctx context.Context, peer *peerUnderTest, hostPeer *v1.Multihost_Peer, wantState v1sync.ConnectionState) {
	ctx, cancel := testutil.WithDeadlineFromTest(t, ctx)
	defer cancel()

	// Important that we subscribe to the state change before we try the initial check to avoid races.
	onStateChanged := peer.manager.peerStateManager.OnStateChanged().Subscribe()
	defer peer.manager.peerStateManager.OnStateChanged().Unsubscribe(onStateChanged)

	// First check if the peer is already connected.
	state := peer.manager.peerStateManager.GetPeerState(hostPeer.Keyid)
	if state != nil && state.ConnectionState == v1sync.ConnectionState_CONNECTION_STATE_CONNECTED {
		return // Already connected, nothing to do
	}

	// If not connected, wait for a connection event
	var lastState *PeerState
	stop := false
	for !stop {
		select {
		case state, ok := <-onStateChanged:
			if !ok {
				stop = true
				continue
			}
			if state.KeyID == hostPeer.Keyid && state.InstanceID == hostPeer.InstanceId {
				lastState = state
				if state.ConnectionState == v1sync.ConnectionState_CONNECTION_STATE_CONNECTED {
					stop = true
					continue
				}
			}
		case <-ctx.Done():
			stop = true
			continue
		}
	}
	if lastState == nil {
		t.Fatalf("timeout waiting for connection to host peer %s", hostPeer.InstanceId)
	} else if lastState.ConnectionState != v1sync.ConnectionState_CONNECTION_STATE_CONNECTED {
		t.Fatalf("expected connection state to be CONNECTED, got %v (reason: %q)", lastState.ConnectionState, lastState.ConnectionStateMessage)
	}
}

func tryConnect(t *testing.T, ctx context.Context, peer *peerUnderTest, hostPeer *v1.Multihost_Peer) {
	waitForConnectionState(t, ctx, peer, hostPeer, v1sync.ConnectionState_CONNECTION_STATE_CONNECTED)