import { WebSocketServer } from '@/lib/websocket/server'

describe('WebSocketServer', () => {
  let server: WebSocketServer

  beforeEach(() => {
    server = new WebSocketServer(3011)
  })

  afterEach(() => {
    server.close()
  })

  it('should initialize WebSocket server', () => {
    expect(server).toBeDefined()
    expect(server.getPort()).toBe(3011)
  })

  it('should have empty connections on start', () => {
    expect(server.getConnectionCount()).toBe(0)
  })

  it('should emit ready event when started', (done) => {
    server.on('ready', () => {
      expect(server.isRunning()).toBe(true)
      done()
    })
    server.start()
  })

  it('should handle client connection', (done) => {
    server.on('client-connect', (clientId) => {
      expect(clientId).toBeDefined()
      expect(server.getConnectionCount()).toBe(1)
      done()
    })
    server.start()
    server.handleClientConnect('test-client-1', 'user-1')
  })
})