react-native-ble-nitro 1.3.1 → 1.4.1

package/src/__tests__/index.test.ts

@@ -2,6 +2,7 @@
 jest.mock('../specs/NativeBleNitro', () => ({
   __esModule: true,
   default: {
+    setRestoreStateCallback: jest.fn(),
     startScan: jest.fn(),
     stopScan: jest.fn(),
     isScanning: jest.fn(),
@@ -39,11 +40,14 @@ jest.mock('../specs/NativeBleNitro', () => ({
   },
 }));
 
-import { ble as BleNitro } from '../index';
+import { BleNitro } from '../index';
 
 // Get reference to the mocked module
 const mockNative = require('../specs/NativeBleNitro').default; // eslint-disable-line @typescript-eslint/no-var-requires
 
+// Get BLE instance
+const BleManager = BleNitro.instance();
+
 describe('BleNitro', () => {
   beforeEach(() => {
     jest.clearAllMocks();
@@ -55,7 +59,7 @@ describe('BleNitro', () => {
     });
 
     const scanCallback = jest.fn();
-    await BleNitro.startScan({ serviceUUIDs: ['test'] }, scanCallback);
+    BleManager.startScan({ serviceUUIDs: ['test'] }, scanCallback);
 
     expect(mockNative.startScan).toHaveBeenCalledWith(
       {
@@ -75,12 +79,12 @@ describe('BleNitro', () => {
     });
     
     const scanCallback = jest.fn();
-    await BleNitro.startScan({ serviceUUIDs: ['test'] }, scanCallback);
+    BleManager.startScan({ serviceUUIDs: ['test'] }, scanCallback);
 
     // Now stop the scan
     mockNative.stopScan.mockImplementation(() => true);
 
-    BleNitro.stopScan();
+    BleManager.stopScan();
     
     expect(mockNative.stopScan).toHaveBeenCalled();
   });
@@ -91,7 +95,7 @@ describe('BleNitro', () => {
       callback(true, id, '');
     });
 
-    const result = await BleNitro.connect(deviceId);
+    const result = await BleManager.connect(deviceId);
     
     expect(mockNative.connect).toHaveBeenCalledWith(deviceId, expect.any(Function), undefined);
     expect(result).toBe(deviceId);
@@ -102,13 +106,13 @@ describe('BleNitro', () => {
       callback(false, '', 'Connection failed');
     });
 
-    await expect(BleNitro.connect('test')).rejects.toThrow('Connection failed');
+    await expect(BleManager.connect('test')).rejects.toThrow('Connection failed');
   });
 
   test('isBluetoothEnabled calls native', () => {
     mockNative.state.mockReturnValue(5); // PoweredOn
 
-    const result = BleNitro.isBluetoothEnabled();
+    const result = BleManager.isBluetoothEnabled();
     
     expect(mockNative.state).toHaveBeenCalled();
     expect(result).toBe(true);
@@ -117,16 +121,72 @@ describe('BleNitro', () => {
   test('writeCharacteristic requires connected device', async () => {
     const data = [1, 2, 3];
     await expect(
-      BleNitro.writeCharacteristic('device', 'service', 'char',data)
+      BleManager.writeCharacteristic('device', 'service', 'char', data)
     ).rejects.toThrow('Device not connected');
   });
 
+  test('writeCharacteristic without response returns empty array', async () => {
+    // First connect
+    mockNative.connect.mockImplementation((id: string, callback: (success: boolean, deviceId: string, error: string) => void, _disconnectCallback?: (deviceId: string, interrupted: boolean, error: string) => void) => {
+      callback(true, id, '');
+    });
+    await BleManager.connect('device-write');
+
+    // Mock writeCharacteristic with new signature (success, responseData, error)
+    mockNative.writeCharacteristic.mockImplementation((_deviceId: string, _serviceId: string, _charId: string, _data: ArrayBuffer, withResponse: boolean, callback: (success: boolean, responseData: ArrayBuffer, error: string) => void) => {
+      // For withResponse=false, return empty ArrayBuffer
+      const emptyBuffer = new ArrayBuffer(0);
+      callback(true, emptyBuffer, '');
+    });
+
+    const data = [1, 2, 3];
+    const result = await BleManager.writeCharacteristic('device-write', 'service', 'char', data, false);
+    
+    expect(mockNative.writeCharacteristic).toHaveBeenCalledWith(
+      'device-write',
+      '0service-0000-1000-8000-00805f9b34fb',
+      '0000char-0000-1000-8000-00805f9b34fb',
+      expect.any(ArrayBuffer),
+      false,
+      expect.any(Function)
+    );
+    expect(result).toEqual([]); // Empty ByteArray for no response
+  });
+
+  test('writeCharacteristic with response returns response data', async () => {
+    // First connect
+    mockNative.connect.mockImplementation((id: string, callback: (success: boolean, deviceId: string, error: string) => void, _disconnectCallback?: (deviceId: string, interrupted: boolean, error: string) => void) => {
+      callback(true, id, '');
+    });
+    await BleManager.connect('device-write-resp');
+
+    // Mock writeCharacteristic to return response data
+    mockNative.writeCharacteristic.mockImplementation((_deviceId: string, _serviceId: string, _charId: string, _data: ArrayBuffer, withResponse: boolean, callback: (success: boolean, responseData: ArrayBuffer, error: string) => void) => {
+      // For withResponse=true, return some response data
+      const responseData = new Uint8Array([0xAA, 0xBB, 0xCC]).buffer;
+      callback(true, responseData, '');
+    });
+
+    const data = [1, 2, 3];
+    const result = await BleManager.writeCharacteristic('device-write-resp', 'service', 'char', data, true);
+    
+    expect(mockNative.writeCharacteristic).toHaveBeenCalledWith(
+      'device-write-resp',
+      '0service-0000-1000-8000-00805f9b34fb',
+      '0000char-0000-1000-8000-00805f9b34fb',
+      expect.any(ArrayBuffer),
+      true,
+      expect.any(Function)
+    );
+    expect(result).toEqual([0xAA, 0xBB, 0xCC]); // Response data as ByteArray
+  });
+
   test('readCharacteristic works after connection', async () => {
     // First connect
     mockNative.connect.mockImplementation((id: string, callback: (success: boolean, deviceId: string, error: string) => void, _disconnectCallback?: (deviceId: string, interrupted: boolean, error: string) => void) => {
       callback(true, id, '');
     });
-    await BleNitro.connect('device');
+    await BleManager.connect('device');
 
     // Then read
     mockNative.readCharacteristic.mockImplementation((_device: string, _service: string, _char: string, callback: (success: boolean, data: ArrayBuffer, error: string) => void) => {
@@ -134,7 +194,7 @@ describe('BleNitro', () => {
       callback(true, testData.buffer, ''); // Battery level 85%
     });
 
-    const result = await BleNitro.readCharacteristic('device', 'service', 'char');
+    const result = await BleManager.readCharacteristic('device', 'service', 'char');
     
     // UUIDs should be normalized in the call
     expect(mockNative.readCharacteristic).toHaveBeenCalledWith(
@@ -154,14 +214,14 @@ describe('BleNitro', () => {
     mockNative.connect.mockImplementation((id: string, callback: (success: boolean, deviceId: string, error: string) => void, _disconnectCallback?: (deviceId: string, interrupted: boolean, error: string) => void) => {
       callback(true, id, '');
     });
-    await BleNitro.connect('device');
+    await BleManager.connect('device');
 
     // Then disconnect
     mockNative.disconnect.mockImplementation((_id: string, callback: (success: boolean, error: string) => void) => {
       callback(true, '');
     });
 
-    const result = await BleNitro.disconnect('device');
+    const result = await BleManager.disconnect('device');
     
     expect(mockNative.disconnect).toHaveBeenCalledWith('device', expect.any(Function));
     expect(result).toBe(undefined);
@@ -172,7 +232,7 @@ describe('BleNitro', () => {
     mockNative.connect.mockImplementation((id: string, callback: (success: boolean, deviceId: string, error: string) => void, _disconnectCallback?: (deviceId: string, interrupted: boolean, error: string) => void) => {
       callback(true, id, '');
     });
-    await BleNitro.connect('device');
+    await BleManager.connect('device');
 
     // Mock subscription
     mockNative.subscribeToCharacteristic.mockImplementation((_device: string, _service: string, _char: string, updateCallback: (charId: string, data: ArrayBuffer) => void, resultCallback: (success: boolean, error: string) => void) => {
@@ -183,7 +243,7 @@ describe('BleNitro', () => {
     });
 
     const notificationCallback = jest.fn();
-    const subscription = BleNitro.subscribeToCharacteristic('device', 'service', 'char', notificationCallback);
+    const subscription = BleManager.subscribeToCharacteristic('device', 'service', 'char', notificationCallback);
     
     expect(mockNative.subscribeToCharacteristic).toHaveBeenCalled();
     expect(notificationCallback).toHaveBeenCalledWith('char-id', [1, 2, 3]);
@@ -207,7 +267,7 @@ describe('BleNitro', () => {
       }
     });
 
-    const result = await BleNitro.connect(deviceId, onDisconnect);
+    const result = await BleManager.connect(deviceId, onDisconnect);
     
     expect(mockNative.connect).toHaveBeenCalledWith(deviceId, expect.any(Function), expect.any(Function));
     expect(result).toBe(deviceId);