Files
BIPMaterialManager/tests/unit/extractor.test.ts
Misaka c39e1504aa style: apply prettier formatting
Apply consistent formatting across all files (line endings, spacing)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-03-01 14:57:38 +08:00

88 lines
2.8 KiB
TypeScript

import { describe, it, expect, beforeEach } from 'vitest'
import { ExtractorService } from '../../src/main/services/erp/extractor'
import { ErpAuthService } from '../../src/main/services/erp/erp-auth'
import type { ErpConfig } from '../../src/main/types/erp.types'
describe('Extractor Service (Unit)', () => {
let authService: ErpAuthService
let extractor: ExtractorService
const mockConfig: ErpConfig = {
url: 'https://test.erp.com',
username: 'test_user',
password: 'test_pass'
}
beforeEach(() => {
authService = new ErpAuthService(mockConfig)
extractor = new ExtractorService(authService, './test-downloads')
})
describe('Batch Creation', () => {
it('should create single batch for small order list', () => {
// This tests the createBatches method indirectly through extract
// We'll need to add a public method or test through the class
const orders = ['ORDER1', 'ORDER2', 'ORDER3']
const batchSize = 10
// Expected: 1 batch with 3 orders
const expectedBatches = 1
expect(Math.ceil(orders.length / batchSize)).toBe(expectedBatches)
})
it('should create multiple batches for large order list', () => {
const orders = Array.from({ length: 250 }, (_, i) => `ORDER${i}`)
const batchSize = 100
// Expected: 3 batches (100, 100, 50)
const expectedBatches = 3
expect(Math.ceil(orders.length / batchSize)).toBe(expectedBatches)
})
it('should handle exact batch size', () => {
const orders = Array.from({ length: 200 }, (_, i) => `ORDER${i}`)
const batchSize = 100
// Expected: 2 batches exactly
const expectedBatches = 2
expect(Math.ceil(orders.length / batchSize)).toBe(expectedBatches)
})
it('should handle empty order list', () => {
const orders: string[] = []
const batchSize = 100
// Expected: 0 batches
const expectedBatches = 0
expect(Math.ceil(orders.length / batchSize)).toBe(expectedBatches)
})
})
describe('Service Initialization', () => {
it('should create service instance', () => {
expect(extractor).toBeDefined()
expect(extractor).toBeInstanceOf(ExtractorService)
})
it('should use default download directory', () => {
const defaultExtractor = new ExtractorService(authService)
expect(defaultExtractor).toBeDefined()
})
it('should use custom download directory', () => {
const customExtractor = new ExtractorService(authService, './custom-downloads')
expect(customExtractor).toBeDefined()
})
})
describe('Error Handling', () => {
it('should handle extraction with no auth session', async () => {
const result = await extractor.extract({
orderNumbers: ['ORDER1']
})
expect(result.errors.length).toBeGreaterThan(0)
expect(result.downloadedFiles).toHaveLength(0)
})
})
})