WellNuo/__tests__/services/ble-logout.test.ts

/**
 * Tests for BLE cleanup on logout functionality
 *
 * This test suite verifies that BLE connections are properly
 * disconnected when a user logs out of the application.
 */

import { RealBLEManager } from '@/services/ble/BLEManager';
import { MockBLEManager } from '@/services/ble/MockBLEManager';
import type { IBLEManager } from '@/services/ble/types';

// Mock the setOnLogoutBLECleanupCallback to avoid importing api module
const mockSetCallback = jest.fn();
jest.mock('@/services/api', () => ({
  setOnLogoutBLECleanupCallback: mockSetCallback,
  api: {
    logout: jest.fn().mockResolvedValue(undefined),
  },
}));

describe('BLE Cleanup on Logout', () => {
  describe('setOnLogoutBLECleanupCallback', () => {
    it('should accept a cleanup callback function', () => {
      const mockCallback = jest.fn().mockResolvedValue(undefined);
      const { setOnLogoutBLECleanupCallback } = require('@/services/api');

      expect(() => {
        setOnLogoutBLECleanupCallback(mockCallback);
      }).not.toThrow();

      expect(mockSetCallback).toHaveBeenCalledWith(mockCallback);
    });

    it('should accept null to clear the callback', () => {
      const { setOnLogoutBLECleanupCallback } = require('@/services/api');
      const mockCallback = jest.fn().mockResolvedValue(undefined);
      setOnLogoutBLECleanupCallback(mockCallback);

      expect(() => {
        setOnLogoutBLECleanupCallback(null);
      }).not.toThrow();

      expect(mockSetCallback).toHaveBeenCalledWith(null);
    });
  });

  describe('BLE Manager cleanup', () => {
    let bleManager: IBLEManager;

    beforeEach(() => {
      // Use MockBLEManager for tests (works in test environment)
      bleManager = new MockBLEManager();
    });

    it('should disconnect all connected devices on cleanup', async () => {
      // Mock scan to get devices
      const devices = await bleManager.scanDevices();
      expect(devices.length).toBeGreaterThan(0);

      // Connect to first device
      const deviceId = devices[0].id;
      const connected = await bleManager.connectDevice(deviceId);
      expect(connected).toBe(true);

      // Cleanup should disconnect
      await bleManager.cleanup();

      // After cleanup, connectedDevices should be empty
      // (We can't directly test the private map, but we can verify cleanup completes)
      expect(async () => await bleManager.cleanup()).not.toThrow();
    });

    it('should handle cleanup with no connected devices', async () => {
      // Cleanup with no connections should not throw
      await expect(bleManager.cleanup()).resolves.not.toThrow();
    });

    it('should handle cleanup errors gracefully', async () => {
      // Mock a device connection
      const devices = await bleManager.scanDevices();
      await bleManager.connectDevice(devices[0].id);

      // Cleanup should not throw even if disconnect fails
      await expect(bleManager.cleanup()).resolves.not.toThrow();
    });

    it('should stop scanning before cleanup', async () => {
      // Start a scan
      const scanPromise = bleManager.scanDevices();

      // Stop scan explicitly (simulate what cleanup does)
      bleManager.stopScan();

      // Wait for scan to complete
      await scanPromise;

      // Cleanup should handle stopped scan
      await expect(bleManager.cleanup()).resolves.not.toThrow();
    });
  });

  describe('Integration: Logout with BLE cleanup', () => {
    it('should register BLE cleanup callback', () => {
      const { setOnLogoutBLECleanupCallback } = require('@/services/api');
      const mockCleanup = jest.fn().mockResolvedValue(undefined);

      // Set the callback
      setOnLogoutBLECleanupCallback(mockCleanup);

      // Verify the mock was called
      expect(mockSetCallback).toHaveBeenCalledWith(mockCleanup);
    });

    it('should allow clearing the callback', () => {
      const { setOnLogoutBLECleanupCallback } = require('@/services/api');
      const mockCleanup = jest.fn().mockResolvedValue(undefined);

      // Set then clear
      setOnLogoutBLECleanupCallback(mockCleanup);
      setOnLogoutBLECleanupCallback(null);

      // Verify null was passed
      expect(mockSetCallback).toHaveBeenCalledWith(null);
    });
  });

  describe('RealBLEManager cleanup', () => {
    // These tests verify RealBLEManager has cleanup implemented
    it('should have cleanup method', () => {
      const manager = new RealBLEManager();
      expect(manager.cleanup).toBeDefined();
      expect(typeof manager.cleanup).toBe('function');
    });

    it('cleanup should return a Promise', () => {
      const manager = new RealBLEManager();
      const result = manager.cleanup();
      expect(result).toBeInstanceOf(Promise);
    });
  });

  describe('MockBLEManager cleanup', () => {
    it('should have cleanup method', () => {
      const manager = new MockBLEManager();
      expect(manager.cleanup).toBeDefined();
      expect(typeof manager.cleanup).toBe('function');
    });

    it('cleanup should disconnect mock devices', async () => {
      const manager = new MockBLEManager();

      // Get mock devices
      const devices = await manager.scanDevices();
      expect(devices.length).toBeGreaterThan(0);

      // Connect to first device
      const connected = await manager.connectDevice(devices[0].id);
      expect(connected).toBe(true);

      // Cleanup
      await manager.cleanup();

      // Verify cleanup logs (console.log is called)
      expect(true).toBe(true); // Placeholder - actual check would need console spy
    });
  });
});