d8d-vite-starter/src/server/utils/__tests__/backup.test.ts

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
import { DatabaseBackup } from '../backup'
import { promises as fs } from 'fs'
import path from 'path'

// Mock pg-dump-restore
vi.mock('pg-dump-restore', () => ({
  pgDump: vi.fn().mockResolvedValue(undefined),
  pgRestore: vi.fn().mockResolvedValue(undefined),
}))

// Mock node-cron
vi.mock('node-cron', () => ({
  default: {
    schedule: vi.fn().mockReturnValue({
      stop: vi.fn(),
      nextDate: vi.fn().mockReturnValue(new Date()),
    }),
  },
}))

// Mock fs with importOriginal for partial mocking
vi.mock('fs', async (importOriginal) => {
  const actual = await importOriginal()
  return {
    ...actual,
    promises: {
      ...actual.promises,
      mkdir: vi.fn().mockResolvedValue(undefined),
      chmod: vi.fn().mockResolvedValue(undefined),
      readdir: vi.fn().mockResolvedValue([]),
      stat: vi.fn().mockResolvedValue({ size: 1024, mtimeMs: Date.now() }),
      access: vi.fn().mockResolvedValue(undefined),
      unlink: vi.fn().mockResolvedValue(undefined),
    },
  }
})

// Mock logger
vi.mock('../logger', () => ({
  logger: {
    db: vi.fn(),
    error: vi.fn(),
  },
}))

describe('DatabaseBackup', () => {
  let backup: DatabaseBackup

  beforeEach(() => {
    vi.clearAllMocks()
    backup = DatabaseBackup.getInstance()
  })

  afterEach(() => {
    vi.restoreAllMocks()
  })

  describe('getInstance', () => {
    it('应该返回单例实例', () => {
      const instance1 = DatabaseBackup.getInstance()
      const instance2 = DatabaseBackup.getInstance()
      expect(instance1).toBe(instance2)
    })
  })

  describe('ensureBackupDir', () => {
    it('应该创建备份目录并设置权限', async () => {
      const { mkdir, chmod } = await import('fs')

      await backup.ensureBackupDir()

      expect(mkdir).toHaveBeenCalledWith('./backups', { recursive: true })
      expect(chmod).toHaveBeenCalledWith('./backups', 0o700)
    })

    it('应该在创建目录失败时抛出错误', async () => {
      const { mkdir } = await import('fs')
      const { logger } = await import('../logger')

      vi.mocked(mkdir).mockRejectedValueOnce(new Error('创建目录失败'))

      await expect(backup.ensureBackupDir()).rejects.toThrow('创建目录失败')
      expect(logger.error).toHaveBeenCalled()
    })
  })

  describe('getDbConfig', () => {
    it('应该返回正确的数据库配置', () => {
      process.env.DB_HOST = 'test-host'
      process.env.DB_PORT = '5433'
      process.env.DB_DATABASE = 'test-db'
      process.env.DB_USERNAME = 'test-user'
      process.env.DB_PASSWORD = 'test-password'

      const config = (backup as any).getDbConfig()

      expect(config).toEqual({
        host: 'test-host',
        port: 5433,
        database: 'test-db',
        username: 'test-user',
        password: 'test-password',
      })
    })

    it('应该使用默认值当环境变量未设置时', () => {
      delete process.env.DB_HOST
      delete process.env.DB_PORT
      delete process.env.DB_DATABASE
      delete process.env.DB_USERNAME
      delete process.env.DB_PASSWORD

      const config = (backup as any).getDbConfig()

      expect(config).toEqual({
        host: 'localhost',
        port: 5432,
        database: 'postgres',
        username: 'postgres',
        password: '',
      })
    })
  })

  describe('formatFileSize', () => {
    it('应该正确格式化文件大小', () => {
      const formatFileSize = (backup as any).formatFileSize

      expect(formatFileSize(0)).toBe('0 B')
      expect(formatFileSize(1024)).toBe('1 KB')
      expect(formatFileSize(1048576)).toBe('1 MB')
      expect(formatFileSize(1073741824)).toBe('1 GB')
    })
  })

  describe('backupExists', () => {
    it('应该返回true当备份文件存在时', async () => {
      const { access } = await import('fs')

      const exists = await backup.backupExists('/path/to/backup.dump')

      expect(exists).toBe(true)
      expect(access).toHaveBeenCalledWith('/path/to/backup.dump')
    })

    it('应该返回false当备份文件不存在时', async () => {
      const { access } = await import('fs')
      vi.mocked(access).mockRejectedValueOnce(new Error('文件不存在'))

      const exists = await backup.backupExists('/path/to/backup.dump')

      expect(exists).toBe(false)
    })
  })

  describe('cleanupOldBackups', () => {
    it('应该清理7天前的旧备份', async () => {
      const { readdir, stat, unlink } = await import('fs')
      const { logger } = await import('../logger')

      const now = Date.now()
      const oldFileTime = now - (8 * 24 * 60 * 60 * 1000) // 8天前
      const newFileTime = now - (6 * 24 * 60 * 60 * 1000) // 6天前

      vi.mocked(readdir).mockResolvedValue(['backup-old.dump', 'backup-new.dump'])
      vi.mocked(stat)
        .mockResolvedValueOnce({ mtimeMs: oldFileTime } as any)
        .mockResolvedValueOnce({ mtimeMs: newFileTime } as any)

      await backup.cleanupOldBackups()

      expect(unlink).toHaveBeenCalledTimes(1)
      expect(unlink).toHaveBeenCalledWith(path.join('./backups', 'backup-old.dump'))
      expect(logger.db).toHaveBeenCalledWith('删除旧备份文件: backup-old.dump')
    })

    it('应该在清理失败时记录错误但不抛出', async () => {
      const { readdir, stat } = await import('fs')
      const { logger } = await import('../logger')

      vi.mocked(readdir).mockRejectedValueOnce(new Error('读取目录失败'))

      await expect(backup.cleanupOldBackups()).resolves.not.toThrow()
      expect(logger.error).toHaveBeenCalled()
    })
  })

  describe('startScheduledBackups', () => {
    it('应该启动定时备份任务', () => {
      const cron = require('node-cron')
      const { logger } = require('../logger')

      backup.startScheduledBackups()

      expect(cron.schedule).toHaveBeenCalledWith('0 2 * * *', expect.any(Function))
      expect(logger.db).toHaveBeenCalledWith('备份调度已启动: 0 2 * * *')
    })
  })

  describe('stopScheduledBackups', () => {
    it('应该停止定时备份任务', () => {
      const { logger } = require('../logger')

      // 先启动再停止
      backup.startScheduledBackups()
      backup.stopScheduledBackups()

      expect(logger.db).toHaveBeenCalledWith('备份调度已停止')
    })
  })
})