super-productivity/packages/super-sync-server/tests/device.service.spec.ts

import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
import { DeviceService } from '../src/sync/services/device.service';
import { ONLINE_DEVICE_THRESHOLD_MS } from '../src/sync/sync.types';

// Mock prisma
vi.mock('../src/db', () => ({
  prisma: {
    syncDevice: {
      count: vi.fn(),
    },
    userSyncState: {
      findMany: vi.fn(),
    },
  },
}));

import { prisma } from '../src/db';

describe('DeviceService', () => {
  let service: DeviceService;

  beforeEach(() => {
    vi.clearAllMocks();
    service = new DeviceService();
  });

  afterEach(() => {
    vi.useRealTimers();
  });

  describe('isDeviceOwner', () => {
    it('should return true when device exists for user', async () => {
      vi.mocked(prisma.syncDevice.count).mockResolvedValue(1);

      const result = await service.isDeviceOwner(1, 'client-123');

      expect(result).toBe(true);
      expect(prisma.syncDevice.count).toHaveBeenCalledWith({
        where: { userId: 1, clientId: 'client-123' },
      });
    });

    it('should return false when device does not exist', async () => {
      vi.mocked(prisma.syncDevice.count).mockResolvedValue(0);

      const result = await service.isDeviceOwner(1, 'unknown-client');

      expect(result).toBe(false);
    });

    it('should return true when multiple devices match (edge case)', async () => {
      vi.mocked(prisma.syncDevice.count).mockResolvedValue(2);

      const result = await service.isDeviceOwner(1, 'client-123');

      expect(result).toBe(true);
    });
  });

  describe('getAllUserIds', () => {
    it('should return empty array when no users exist', async () => {
      vi.mocked(prisma.userSyncState.findMany).mockResolvedValue([]);

      const result = await service.getAllUserIds();

      expect(result).toEqual([]);
      expect(prisma.userSyncState.findMany).toHaveBeenCalledWith({
        select: { userId: true },
        distinct: ['userId'],
      });
    });

    it('should return user IDs from sync state', async () => {
      vi.mocked(prisma.userSyncState.findMany).mockResolvedValue([
        { userId: 1 },
        { userId: 2 },
        { userId: 3 },
      ] as any);

      const result = await service.getAllUserIds();

      expect(result).toEqual([1, 2, 3]);
    });

    it('should handle single user', async () => {
      vi.mocked(prisma.userSyncState.findMany).mockResolvedValue([{ userId: 42 }] as any);

      const result = await service.getAllUserIds();

      expect(result).toEqual([42]);
    });
  });

  describe('getOnlineDeviceCount', () => {
    it('should return count of online devices', async () => {
      vi.useFakeTimers();
      const now = Date.now();
      vi.setSystemTime(now);

      vi.mocked(prisma.syncDevice.count).mockResolvedValue(3);

      const result = await service.getOnlineDeviceCount(1);

      expect(result).toBe(3);
      expect(prisma.syncDevice.count).toHaveBeenCalledWith({
        where: {
          userId: 1,
          lastSeenAt: { gt: BigInt(now - ONLINE_DEVICE_THRESHOLD_MS) },
        },
      });
    });

    it('should return zero when no online devices', async () => {
      vi.mocked(prisma.syncDevice.count).mockResolvedValue(0);

      const result = await service.getOnlineDeviceCount(1);

      expect(result).toBe(0);
    });

    it('should calculate threshold correctly', async () => {
      vi.useFakeTimers();
      const now = 1700000000000; // Fixed timestamp
      vi.setSystemTime(now);

      vi.mocked(prisma.syncDevice.count).mockResolvedValue(1);

      await service.getOnlineDeviceCount(1);

      const expectedThreshold = BigInt(now - ONLINE_DEVICE_THRESHOLD_MS);
      expect(prisma.syncDevice.count).toHaveBeenCalledWith({
        where: {
          userId: 1,
          lastSeenAt: { gt: expectedThreshold },
        },
      });
    });
  });
});