Add tests

packages/net/__tests__/Connection.test.ts

@@ -0,0 +1,40 @@
+import {Connection, ConnectionStatus} from "../src/Connection"
+import {ConnectionEvent} from "../src/ConnectionEvent"
+import {vi, describe, it, expect, beforeEach, afterEach} from "vitest"
+
+describe("Connection", () => {
+  let connection: Connection
+
+  beforeEach(() => {
+    connection = new Connection("wss://test.relay/")
+  })
+
+  afterEach(() => {
+    connection.cleanup()
+  })
+
+  it("should initialize with correct state", () => {
+    expect(connection.status).toBe(ConnectionStatus.Open)
+    expect(connection.url).toBe("wss://test.relay/")
+  })
+
+  it("should emit events with connection instance", () => {
+    const spy = vi.fn()
+    connection.on(ConnectionEvent.Open, spy)
+    connection.emit(ConnectionEvent.Open)
+    expect(spy).toHaveBeenCalledWith(connection)
+  })
+
+  it("should throw when sending message on closed connection", async () => {
+    connection.close()
+    await expect(connection.send(["EVENT", {}])).rejects.toThrow()
+  })
+
+  it("should cleanup properly", () => {
+    const spy = vi.fn()
+    connection.on("test", spy)
+    connection.cleanup()
+    connection.emit("test" as any)
+    expect(spy).not.toHaveBeenCalled()
+  })
+})