Merge pull request #706 from retejs/add-tests

Add tests

test/index.test.ts

@@ -1,4 +1,3 @@
-// import jest globals
 import { describe, expect, it } from '@jest/globals'
 
 import { NodeEditor } from '../src/editor'
@@ -52,5 +51,43 @@ describe('NodeEditor', () => {
 
     await expect(() => editor.addConnection(connectionData)).rejects.toThrowError()
   })
+
+  it('removeNode should remove a node', async () => {
+    const editor = new NodeEditor()
+    const nodeData = { id: '1', label: 'Node 1' }
+
+    await editor.addNode(nodeData)
+    await editor.removeNode('1')
+    const nodes = editor.getNodes()
+
+    expect(nodes).toHaveLength(0)
+  })
+
+  it('removeConnection should remove a connection', async () => {
+    const editor = new NodeEditor()
+    const connectionData = { id: '1', source: '1', target: '2' }
+
+    await editor.addNode({ id: '1' })
+    await editor.addNode({ id: '2' })
+    await editor.addConnection(connectionData)
+    await editor.removeConnection('1')
+    const connections = editor.getConnections()
+
+    expect(connections).toHaveLength(0)
+  })
+
+  it('should clear all nodes and connections', async () => {
+    const editor = new NodeEditor()
+
+    await editor.addNode({ id: '1' })
+    await editor.addNode({ id: '2' })
+    await editor.addConnection({ id: '1', source: '1', target: '2' })
+    await editor.clear()
+    const nodes = editor.getNodes()
+    const connections = editor.getConnections()
+
+    expect(nodes).toHaveLength(0)
+    expect(connections).toHaveLength(0)
+  })
 })
 

test/mocks/crypto.ts

@@ -0,0 +1,24 @@
+import { jest } from '@jest/globals'
+import { Buffer } from 'buffer'
+
+export function mockCrypto(object: Record<string, unknown>) {
+  // eslint-disable-next-line no-undef
+  globalThis.crypto = object as unknown as Crypto
+}
+
+export function mockCryptoFromArray(array: Uint8Array) {
+  mockCrypto({
+    getRandomValues: jest.fn().mockReturnValue(array)
+  })
+}
+
+export function mockCryptoFromBuffer(buffer: Buffer) {
+  mockCrypto({
+    randomBytes: jest.fn().mockReturnValue(buffer)
+  })
+}
+
+export function resetCrypto() {
+  // eslint-disable-next-line no-undef, no-undefined
+  globalThis.crypto = undefined as unknown as Crypto
+}