//
// Copyright 2023 DXOS.org
//

import { afterEach, onTestFinished, beforeEach, describe, expect, test } from 'vitest';

import { Trigger } from '@dxos/async';
import { Context } from '@dxos/context';
import { type NetworkService, ConnectionState } from '@dxos/protocols/proto/dxos/client/services';

import { NetworkServiceImpl } from './network-service';
import { type ServiceContext } from '../services';
import { createServiceContext } from '../testing';

describe('NetworkService', () => {
  let serviceContext: ServiceContext;
  let networkService: NetworkService;

  beforeEach(async () => {
    serviceContext = await createServiceContext();
    await serviceContext.open(new Context());
    networkService = new NetworkServiceImpl(serviceContext.networkManager, serviceContext.signalManager);
  });

  afterEach(async () => {
    await serviceContext.close();
  });

  test('setNetworkOptions changes network status', async () => {
    await networkService.updateConfig({
      swarm: ConnectionState.OFFLINE,
    });

    expect(serviceContext.networkManager.connectionState).to.equal(ConnectionState.OFFLINE);
  });

  test('subscribeToNetworkStatus returns current network status', async () => {
    const query = networkService.queryStatus();
    let result = new Trigger<ConnectionState | undefined>();
    query.subscribe(({ swarm }) => {
      result.wake(swarm);
    });
    onTestFinished(() => query.close());
    expect(await result.wait()).to.equal(ConnectionState.ONLINE);

    result = new Trigger<ConnectionState | undefined>();
    await networkService.updateConfig({
      swarm: ConnectionState.OFFLINE,
    });
    expect(await result.wait()).to.equal(ConnectionState.OFFLINE);
  });
});