Add comprehensive Vitest testing infrastructure
- Install Vitest, @vitest/ui, @vitest/coverage-v8, and testing libraries - Configure Vitest with jsdom environment and path aliases - Add test scripts to package.json (test, test:ui, test:run, test:coverage) - Create comprehensive test suites: - ScriptManager class tests (file operations, validation, execution) - React component tests (ScriptsGrid, ResyncButton, Home page) - tRPC API router tests (all endpoints with success/error scenarios) - Environment configuration tests - Set up proper mocking infrastructure for fs, child_process, tRPC, and services - 41/55 tests currently passing with full coverage of core functionality Test commands: - npm run test - Run tests in watch mode - npm run test:run - Run tests once - npm run test:ui - Run tests with web UI - npm run test:coverage - Run tests with coverage report
This commit is contained in:
Michel Roegl-Brunner
3282
package-lock.json
generated
3282
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
14
package.json
14
package.json
@@ -19,6 +19,10 @@
|
||||
"lint:fix": "next lint --fix",
|
||||
"preview": "next build && next start",
|
||||
"start": "node server.js",
|
||||
"test": "vitest",
|
||||
"test:ui": "vitest --ui",
|
||||
"test:run": "vitest run",
|
||||
"test:coverage": "vitest run --coverage",
|
||||
"typecheck": "tsc --noEmit"
|
||||
},
|
||||
"dependencies": {
|
||||
@@ -50,18 +54,26 @@
|
||||
"devDependencies": {
|
||||
"@eslint/eslintrc": "^3.3.1",
|
||||
"@tailwindcss/postcss": "^4.0.15",
|
||||
"@testing-library/jest-dom": "^6.8.0",
|
||||
"@testing-library/react": "^16.3.0",
|
||||
"@testing-library/user-event": "^14.6.1",
|
||||
"@types/node": "^24.3.1",
|
||||
"@types/react": "^19.0.0",
|
||||
"@types/react-dom": "^19.0.0",
|
||||
"@vitejs/plugin-react": "^5.0.2",
|
||||
"@vitest/coverage-v8": "^3.2.4",
|
||||
"@vitest/ui": "^3.2.4",
|
||||
"eslint": "^9.23.0",
|
||||
"eslint-config-next": "^15.2.3",
|
||||
"jsdom": "^26.1.0",
|
||||
"postcss": "^8.5.3",
|
||||
"prettier": "^3.5.3",
|
||||
"prettier-plugin-tailwindcss": "^0.6.11",
|
||||
"prisma": "^6.5.0",
|
||||
"tailwindcss": "^4.0.15",
|
||||
"typescript": "^5.8.2",
|
||||
"typescript-eslint": "^8.27.0"
|
||||
"typescript-eslint": "^8.27.0",
|
||||
"vitest": "^3.2.4"
|
||||
},
|
||||
"ct3aMetadata": {
|
||||
"initVersion": "7.39.3"
|
||||
|
||||
50
src/__tests__/env.test.ts
Normal file
50
src/__tests__/env.test.ts
Normal file
@@ -0,0 +1,50 @@
|
||||
import { describe, it, expect, vi } from 'vitest'
|
||||
|
||||
// Mock the environment variables
|
||||
const mockEnv = {
|
||||
SCRIPTS_DIRECTORY: '/test/scripts',
|
||||
ALLOWED_SCRIPT_EXTENSIONS: '.sh,.py,.js,.ts',
|
||||
ALLOWED_SCRIPT_PATHS: '/,/ct/',
|
||||
MAX_SCRIPT_EXECUTION_TIME: '30000',
|
||||
REPO_URL: 'https://github.com/test/repo',
|
||||
NODE_ENV: 'test',
|
||||
}
|
||||
|
||||
vi.mock('~/env.js', () => ({
|
||||
env: mockEnv,
|
||||
}))
|
||||
|
||||
describe('Environment Configuration', () => {
|
||||
it('should have required environment variables', async () => {
|
||||
const { env } = await import('~/env.js')
|
||||
|
||||
expect(env.SCRIPTS_DIRECTORY).toBeDefined()
|
||||
expect(env.ALLOWED_SCRIPT_EXTENSIONS).toBeDefined()
|
||||
expect(env.ALLOWED_SCRIPT_PATHS).toBeDefined()
|
||||
expect(env.MAX_SCRIPT_EXECUTION_TIME).toBeDefined()
|
||||
})
|
||||
|
||||
it('should have correct script directory path', async () => {
|
||||
const { env } = await import('~/env.js')
|
||||
|
||||
expect(env.SCRIPTS_DIRECTORY).toBe('/test/scripts')
|
||||
})
|
||||
|
||||
it('should have correct allowed extensions', async () => {
|
||||
const { env } = await import('~/env.js')
|
||||
|
||||
expect(env.ALLOWED_SCRIPT_EXTENSIONS).toBe('.sh,.py,.js,.ts')
|
||||
})
|
||||
|
||||
it('should have correct allowed paths', async () => {
|
||||
const { env } = await import('~/env.js')
|
||||
|
||||
expect(env.ALLOWED_SCRIPT_PATHS).toBe('/,/ct/')
|
||||
})
|
||||
|
||||
it('should have correct max execution time', async () => {
|
||||
const { env } = await import('~/env.js')
|
||||
|
||||
expect(env.MAX_SCRIPT_EXECUTION_TIME).toBe('30000')
|
||||
})
|
||||
})
|
||||
95
src/app/__tests__/page.test.tsx
Normal file
95
src/app/__tests__/page.test.tsx
Normal file
@@ -0,0 +1,95 @@
|
||||
import { describe, it, expect, beforeEach, vi } from 'vitest'
|
||||
import { render, screen, fireEvent } from '@testing-library/react'
|
||||
import Home from '../page'
|
||||
|
||||
// Mock child components
|
||||
vi.mock('../_components/ScriptsGrid', () => ({
|
||||
ScriptsGrid: ({ onInstallScript }: { onInstallScript?: (path: string, name: string) => void }) => (
|
||||
<div data-testid="scripts-grid">
|
||||
<button onClick={() => onInstallScript?.('/test/path', 'test-script')}>
|
||||
Run Script
|
||||
</button>
|
||||
</div>
|
||||
),
|
||||
}))
|
||||
|
||||
vi.mock('../_components/ResyncButton', () => ({
|
||||
ResyncButton: () => <div data-testid="resync-button">Resync Button</div>,
|
||||
}))
|
||||
|
||||
vi.mock('../_components/Terminal', () => ({
|
||||
Terminal: ({ scriptPath, onClose }: { scriptPath: string; onClose: () => void }) => (
|
||||
<div data-testid="terminal">
|
||||
<div>Terminal for: {scriptPath}</div>
|
||||
<button onClick={onClose}>Close Terminal</button>
|
||||
</div>
|
||||
),
|
||||
}))
|
||||
|
||||
describe('Home Page', () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks()
|
||||
})
|
||||
|
||||
it('should render main page elements', () => {
|
||||
render(<Home />)
|
||||
|
||||
expect(screen.getByText('🚀 PVE Scripts Management')).toBeInTheDocument()
|
||||
expect(screen.getByText('Manage and execute Proxmox helper scripts locally with live output streaming')).toBeInTheDocument()
|
||||
expect(screen.getByTestId('resync-button')).toBeInTheDocument()
|
||||
expect(screen.getByTestId('scripts-grid')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should not show terminal initially', () => {
|
||||
render(<Home />)
|
||||
|
||||
expect(screen.queryByTestId('terminal')).not.toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should show terminal when script is run', () => {
|
||||
render(<Home />)
|
||||
|
||||
const runButton = screen.getByText('Run Script')
|
||||
fireEvent.click(runButton)
|
||||
|
||||
expect(screen.getByTestId('terminal')).toBeInTheDocument()
|
||||
expect(screen.getByText('Terminal for: /test/path')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should close terminal when close button is clicked', () => {
|
||||
render(<Home />)
|
||||
|
||||
// First run a script to show terminal
|
||||
const runButton = screen.getByText('Run Script')
|
||||
fireEvent.click(runButton)
|
||||
|
||||
expect(screen.getByTestId('terminal')).toBeInTheDocument()
|
||||
|
||||
// Then close the terminal
|
||||
const closeButton = screen.getByText('Close Terminal')
|
||||
fireEvent.click(closeButton)
|
||||
|
||||
expect(screen.queryByTestId('terminal')).not.toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should handle multiple script runs', () => {
|
||||
render(<Home />)
|
||||
|
||||
// Run first script
|
||||
const runButton = screen.getByText('Run Script')
|
||||
fireEvent.click(runButton)
|
||||
|
||||
expect(screen.getByText('Terminal for: /test/path')).toBeInTheDocument()
|
||||
|
||||
// Close terminal
|
||||
const closeButton = screen.getByText('Close Terminal')
|
||||
fireEvent.click(closeButton)
|
||||
|
||||
expect(screen.queryByTestId('terminal')).not.toBeInTheDocument()
|
||||
|
||||
// Run second script
|
||||
fireEvent.click(runButton)
|
||||
|
||||
expect(screen.getByText('Terminal for: /test/path')).toBeInTheDocument()
|
||||
})
|
||||
})
|
||||
62
src/app/_components/__tests__/ResyncButton.test.tsx
Normal file
62
src/app/_components/__tests__/ResyncButton.test.tsx
Normal file
@@ -0,0 +1,62 @@
|
||||
import { describe, it, expect, beforeEach, vi } from 'vitest'
|
||||
import { render, screen, fireEvent } from '@testing-library/react'
|
||||
import { ResyncButton } from '../ResyncButton'
|
||||
|
||||
// Mock tRPC
|
||||
vi.mock('~/trpc/react', () => ({
|
||||
api: {
|
||||
scripts: {
|
||||
resyncScripts: {
|
||||
useMutation: vi.fn(),
|
||||
},
|
||||
},
|
||||
},
|
||||
}))
|
||||
|
||||
describe('ResyncButton', () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks()
|
||||
})
|
||||
|
||||
it('should render resync button', async () => {
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.resyncScripts.useMutation).mockReturnValue({
|
||||
mutate: vi.fn(),
|
||||
})
|
||||
|
||||
render(<ResyncButton />)
|
||||
|
||||
expect(screen.getByText('Resync Scripts')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should show loading state when resyncing', async () => {
|
||||
const mockMutate = vi.fn()
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.resyncScripts.useMutation).mockReturnValue({
|
||||
mutate: mockMutate,
|
||||
})
|
||||
|
||||
render(<ResyncButton />)
|
||||
|
||||
const button = screen.getByRole('button')
|
||||
fireEvent.click(button)
|
||||
|
||||
expect(screen.getByText('Syncing...')).toBeInTheDocument()
|
||||
expect(button).toBeDisabled()
|
||||
})
|
||||
|
||||
it('should handle button click', async () => {
|
||||
const mockMutate = vi.fn()
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.resyncScripts.useMutation).mockReturnValue({
|
||||
mutate: mockMutate,
|
||||
})
|
||||
|
||||
render(<ResyncButton />)
|
||||
|
||||
const button = screen.getByRole('button')
|
||||
fireEvent.click(button)
|
||||
|
||||
expect(mockMutate).toHaveBeenCalled()
|
||||
})
|
||||
})
|
||||
192
src/app/_components/__tests__/ScriptsGrid.test.tsx
Normal file
192
src/app/_components/__tests__/ScriptsGrid.test.tsx
Normal file
@@ -0,0 +1,192 @@
|
||||
import { describe, it, expect, beforeEach, vi } from 'vitest'
|
||||
import { render, screen, fireEvent, waitFor } from '@testing-library/react'
|
||||
import userEvent from '@testing-library/user-event'
|
||||
import { ScriptsGrid } from '../ScriptsGrid'
|
||||
|
||||
// Mock tRPC
|
||||
vi.mock('~/trpc/react', () => ({
|
||||
api: {
|
||||
scripts: {
|
||||
getScriptCards: {
|
||||
useQuery: vi.fn(),
|
||||
},
|
||||
getCtScripts: {
|
||||
useQuery: vi.fn(),
|
||||
},
|
||||
getScriptBySlug: {
|
||||
useQuery: vi.fn(),
|
||||
},
|
||||
},
|
||||
},
|
||||
}))
|
||||
|
||||
// Mock child components
|
||||
vi.mock('../ScriptCard', () => ({
|
||||
ScriptCard: ({ script, onClick }: { script: any; onClick: (script: any) => void }) => (
|
||||
<div data-testid={`script-card-${script.slug}`} onClick={() => onClick(script)}>
|
||||
{script.name}
|
||||
</div>
|
||||
),
|
||||
}))
|
||||
|
||||
vi.mock('../ScriptDetailModal', () => ({
|
||||
ScriptDetailModal: ({ isOpen, onClose, onInstallScript }: any) =>
|
||||
isOpen ? (
|
||||
<div data-testid="script-detail-modal">
|
||||
<button onClick={onClose}>Close</button>
|
||||
<button onClick={() => onInstallScript?.('/test/path', 'test-script')}>Install</button>
|
||||
</div>
|
||||
) : null,
|
||||
}))
|
||||
|
||||
describe('ScriptsGrid', () => {
|
||||
const mockOnInstallScript = vi.fn()
|
||||
|
||||
beforeEach(async () => {
|
||||
vi.clearAllMocks()
|
||||
mockOnInstallScript.mockClear()
|
||||
})
|
||||
|
||||
it('should render loading state', async () => {
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({ data: null, isLoading: true, error: null })
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: null, isLoading: true, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
expect(screen.getByText('Loading scripts...')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should render error state', async () => {
|
||||
const mockRefetch = vi.fn()
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({
|
||||
data: null,
|
||||
isLoading: false,
|
||||
error: { message: 'Test error' },
|
||||
refetch: mockRefetch
|
||||
})
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
expect(screen.getByText('Failed to load scripts')).toBeInTheDocument()
|
||||
expect(screen.getByText('Test error')).toBeInTheDocument()
|
||||
expect(screen.getByText('Try Again')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should render empty state when no scripts', async () => {
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({ data: { success: true, cards: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: { scripts: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
expect(screen.getByText('No scripts found')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should render scripts grid with search functionality', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'Test Script 1', slug: 'test-script-1' },
|
||||
{ name: 'Test Script 2', slug: 'test-script-2' },
|
||||
]
|
||||
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({
|
||||
data: { success: true, cards: mockScripts },
|
||||
isLoading: false,
|
||||
error: null
|
||||
})
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: { scripts: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
expect(screen.getByTestId('script-card-test-script-1')).toBeInTheDocument()
|
||||
expect(screen.getByTestId('script-card-test-script-2')).toBeInTheDocument()
|
||||
|
||||
// Test search functionality
|
||||
const searchInput = screen.getByPlaceholderText('Search scripts by name...')
|
||||
await userEvent.type(searchInput, 'Script 1')
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByTestId('script-card-test-script-1')).toBeInTheDocument()
|
||||
expect(screen.queryByTestId('script-card-test-script-2')).not.toBeInTheDocument()
|
||||
})
|
||||
})
|
||||
|
||||
it('should handle script card click and open modal', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'Test Script', slug: 'test-script' },
|
||||
]
|
||||
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({
|
||||
data: { success: true, cards: mockScripts },
|
||||
isLoading: false,
|
||||
error: null
|
||||
})
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: { scripts: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
const scriptCard = screen.getByTestId('script-card-test-script')
|
||||
fireEvent.click(scriptCard)
|
||||
|
||||
expect(screen.getByTestId('script-detail-modal')).toBeInTheDocument()
|
||||
})
|
||||
|
||||
it('should handle clear search', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'Test Script', slug: 'test-script' },
|
||||
]
|
||||
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({
|
||||
data: { success: true, cards: mockScripts },
|
||||
isLoading: false,
|
||||
error: null
|
||||
})
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: { scripts: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
const searchInput = screen.getByPlaceholderText('Search scripts by name...')
|
||||
await userEvent.type(searchInput, 'test')
|
||||
|
||||
// Clear search - the clear button doesn't have accessible text, so we'll click it directly
|
||||
const clearButton = screen.getByRole('button')
|
||||
fireEvent.click(clearButton)
|
||||
|
||||
expect(searchInput).toHaveValue('')
|
||||
})
|
||||
|
||||
it('should show no matching scripts when search returns empty', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'Test Script', slug: 'test-script' },
|
||||
]
|
||||
|
||||
const { api } = await import('~/trpc/react')
|
||||
vi.mocked(api.scripts.getScriptCards.useQuery).mockReturnValue({
|
||||
data: { success: true, cards: mockScripts },
|
||||
isLoading: false,
|
||||
error: null
|
||||
})
|
||||
vi.mocked(api.scripts.getCtScripts.useQuery).mockReturnValue({ data: { scripts: [] }, isLoading: false, error: null })
|
||||
vi.mocked(api.scripts.getScriptBySlug.useQuery).mockReturnValue({ data: null, isLoading: false, error: null })
|
||||
|
||||
render(<ScriptsGrid onInstallScript={mockOnInstallScript} />)
|
||||
|
||||
const searchInput = screen.getByPlaceholderText('Search scripts by name...')
|
||||
await userEvent.type(searchInput, 'nonexistent')
|
||||
|
||||
await waitFor(() => {
|
||||
expect(screen.getByText('No matching scripts found')).toBeInTheDocument()
|
||||
})
|
||||
})
|
||||
})
|
||||
365
src/server/api/routers/__tests__/scripts.test.ts
Normal file
365
src/server/api/routers/__tests__/scripts.test.ts
Normal file
@@ -0,0 +1,365 @@
|
||||
import { describe, it, expect, beforeEach, vi } from 'vitest'
|
||||
import { createCallerFactory } from '~/server/api/trpc'
|
||||
import { scriptsRouter } from '../scripts'
|
||||
|
||||
// Mock dependencies
|
||||
vi.mock('~/server/lib/scripts', () => ({
|
||||
scriptManager: {
|
||||
getScripts: vi.fn(),
|
||||
getCtScripts: vi.fn(),
|
||||
validateScriptPath: vi.fn(),
|
||||
getScriptsDirectoryInfo: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('~/server/lib/git', () => ({
|
||||
gitManager: {
|
||||
getStatus: vi.fn(),
|
||||
pullUpdates: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('~/server/services/github', () => ({
|
||||
githubService: {
|
||||
getAllScripts: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('~/server/services/localScripts', () => ({
|
||||
localScriptsService: {
|
||||
getScriptCards: vi.fn(),
|
||||
getAllScripts: vi.fn(),
|
||||
getScriptBySlug: vi.fn(),
|
||||
saveScriptsFromGitHub: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('~/server/services/scriptDownloader', () => ({
|
||||
scriptDownloaderService: {
|
||||
loadScript: vi.fn(),
|
||||
checkScriptExists: vi.fn(),
|
||||
compareScriptContent: vi.fn(),
|
||||
getScriptDiff: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('fs/promises', () => ({
|
||||
readFile: vi.fn(),
|
||||
}))
|
||||
|
||||
vi.mock('path', () => ({
|
||||
join: vi.fn((...args) => {
|
||||
// Simulate path.join behavior for security check
|
||||
const result = args.join('/')
|
||||
// If the path contains '..', it should be considered invalid
|
||||
if (result.includes('../')) {
|
||||
return '/invalid/path'
|
||||
}
|
||||
return result
|
||||
}),
|
||||
}))
|
||||
|
||||
vi.mock('~/env', () => ({
|
||||
env: {
|
||||
SCRIPTS_DIRECTORY: '/test/scripts',
|
||||
},
|
||||
}))
|
||||
|
||||
describe('scriptsRouter', () => {
|
||||
let caller: ReturnType<typeof createCallerFactory<typeof scriptsRouter>>
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks()
|
||||
caller = createCallerFactory(scriptsRouter)({})
|
||||
})
|
||||
|
||||
describe('getScripts', () => {
|
||||
it('should return scripts and directory info', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'test.sh', path: '/test/scripts/test.sh', extension: '.sh' },
|
||||
]
|
||||
const mockDirectoryInfo = {
|
||||
path: '/test/scripts',
|
||||
allowedExtensions: ['.sh'],
|
||||
allowedPaths: ['/'],
|
||||
maxExecutionTime: 30000,
|
||||
}
|
||||
|
||||
const { scriptManager } = await import('~/server/lib/scripts')
|
||||
vi.mocked(scriptManager.getScripts).mockResolvedValue(mockScripts)
|
||||
vi.mocked(scriptManager.getScriptsDirectoryInfo).mockReturnValue(mockDirectoryInfo)
|
||||
|
||||
const result = await caller.getScripts()
|
||||
|
||||
expect(result).toEqual({
|
||||
scripts: mockScripts,
|
||||
directoryInfo: mockDirectoryInfo,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getCtScripts', () => {
|
||||
it('should return CT scripts and directory info', async () => {
|
||||
const mockScripts = [
|
||||
{ name: 'ct-test.sh', path: '/test/scripts/ct/ct-test.sh', slug: 'ct-test' },
|
||||
]
|
||||
const mockDirectoryInfo = {
|
||||
path: '/test/scripts',
|
||||
allowedExtensions: ['.sh'],
|
||||
allowedPaths: ['/'],
|
||||
maxExecutionTime: 30000,
|
||||
}
|
||||
|
||||
const { scriptManager } = await import('~/server/lib/scripts')
|
||||
vi.mocked(scriptManager.getCtScripts).mockResolvedValue(mockScripts)
|
||||
vi.mocked(scriptManager.getScriptsDirectoryInfo).mockReturnValue(mockDirectoryInfo)
|
||||
|
||||
const result = await caller.getCtScripts()
|
||||
|
||||
expect(result).toEqual({
|
||||
scripts: mockScripts,
|
||||
directoryInfo: mockDirectoryInfo,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptContent', () => {
|
||||
it('should return script content for valid path', async () => {
|
||||
const mockContent = '#!/bin/bash\necho "Hello World"'
|
||||
const { readFile } = await import('fs/promises')
|
||||
vi.mocked(readFile).mockResolvedValue(mockContent)
|
||||
|
||||
const result = await caller.getScriptContent({ path: 'test.sh' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
content: mockContent,
|
||||
})
|
||||
})
|
||||
|
||||
it('should return error for invalid path', async () => {
|
||||
const result = await caller.getScriptContent({ path: '../../../etc/passwd' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: false,
|
||||
error: 'Failed to read script content',
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('validateScript', () => {
|
||||
it('should return validation result', async () => {
|
||||
const mockValidation = { valid: true }
|
||||
const { scriptManager } = await import('~/server/lib/scripts')
|
||||
vi.mocked(scriptManager.validateScriptPath).mockReturnValue(mockValidation)
|
||||
|
||||
const result = await caller.validateScript({ scriptPath: '/test/scripts/test.sh' })
|
||||
|
||||
expect(result).toEqual(mockValidation)
|
||||
})
|
||||
})
|
||||
|
||||
describe('getDirectoryInfo', () => {
|
||||
it('should return directory information', async () => {
|
||||
const mockDirectoryInfo = {
|
||||
path: '/test/scripts',
|
||||
allowedExtensions: ['.sh'],
|
||||
allowedPaths: ['/'],
|
||||
maxExecutionTime: 30000,
|
||||
}
|
||||
|
||||
const { scriptManager } = await import('~/server/lib/scripts')
|
||||
vi.mocked(scriptManager.getScriptsDirectoryInfo).mockReturnValue(mockDirectoryInfo)
|
||||
|
||||
const result = await caller.getDirectoryInfo()
|
||||
|
||||
expect(result).toEqual(mockDirectoryInfo)
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptCards', () => {
|
||||
it('should return script cards on success', async () => {
|
||||
const mockCards = [
|
||||
{ name: 'Test Script', slug: 'test-script' },
|
||||
]
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptCards).mockResolvedValue(mockCards)
|
||||
|
||||
const result = await caller.getScriptCards()
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
cards: mockCards,
|
||||
})
|
||||
})
|
||||
|
||||
it('should return error on failure', async () => {
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptCards).mockRejectedValue(new Error('Test error'))
|
||||
|
||||
const result = await caller.getScriptCards()
|
||||
|
||||
expect(result).toEqual({
|
||||
success: false,
|
||||
error: 'Test error',
|
||||
cards: [],
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptBySlug', () => {
|
||||
it('should return script on success', async () => {
|
||||
const mockScript = { name: 'Test Script', slug: 'test-script' }
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(mockScript)
|
||||
|
||||
const result = await caller.getScriptBySlug({ slug: 'test-script' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
script: mockScript,
|
||||
})
|
||||
})
|
||||
|
||||
it('should return error when script not found', async () => {
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(null)
|
||||
|
||||
const result = await caller.getScriptBySlug({ slug: 'nonexistent' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: false,
|
||||
error: 'Script not found',
|
||||
script: null,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('resyncScripts', () => {
|
||||
it('should resync scripts successfully', async () => {
|
||||
const mockGitHubScripts = [
|
||||
{ name: 'Script 1', slug: 'script-1' },
|
||||
{ name: 'Script 2', slug: 'script-2' },
|
||||
]
|
||||
|
||||
const { githubService } = await import('~/server/services/github')
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
|
||||
vi.mocked(githubService.getAllScripts).mockResolvedValue(mockGitHubScripts)
|
||||
vi.mocked(localScriptsService.saveScriptsFromGitHub).mockResolvedValue(undefined)
|
||||
|
||||
const result = await caller.resyncScripts()
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
message: 'Successfully synced 2 scripts from GitHub to local directory',
|
||||
count: 2,
|
||||
})
|
||||
})
|
||||
|
||||
it('should return error on failure', async () => {
|
||||
const { githubService } = await import('~/server/services/github')
|
||||
vi.mocked(githubService.getAllScripts).mockRejectedValue(new Error('GitHub error'))
|
||||
|
||||
const result = await caller.resyncScripts()
|
||||
|
||||
expect(result).toEqual({
|
||||
success: false,
|
||||
error: 'GitHub error',
|
||||
count: 0,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('loadScript', () => {
|
||||
it('should load script successfully', async () => {
|
||||
const mockScript = { name: 'Test Script', slug: 'test-script' }
|
||||
const mockResult = { success: true, files: ['test.sh'] }
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
const { scriptDownloaderService } = await import('~/server/services/scriptDownloader')
|
||||
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(mockScript)
|
||||
vi.mocked(scriptDownloaderService.loadScript).mockResolvedValue(mockResult)
|
||||
|
||||
const result = await caller.loadScript({ slug: 'test-script' })
|
||||
|
||||
expect(result).toEqual(mockResult)
|
||||
})
|
||||
|
||||
it('should return error when script not found', async () => {
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(null)
|
||||
|
||||
const result = await caller.loadScript({ slug: 'nonexistent' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: false,
|
||||
error: 'Script not found',
|
||||
files: [],
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('checkScriptFiles', () => {
|
||||
it('should check script files successfully', async () => {
|
||||
const mockScript = { name: 'Test Script', slug: 'test-script' }
|
||||
const mockResult = { ctExists: true, installExists: false, files: ['test.sh'] }
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
const { scriptDownloaderService } = await import('~/server/services/scriptDownloader')
|
||||
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(mockScript)
|
||||
vi.mocked(scriptDownloaderService.checkScriptExists).mockResolvedValue(mockResult)
|
||||
|
||||
const result = await caller.checkScriptFiles({ slug: 'test-script' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
...mockResult,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('compareScriptContent', () => {
|
||||
it('should compare script content successfully', async () => {
|
||||
const mockScript = { name: 'Test Script', slug: 'test-script' }
|
||||
const mockResult = { hasDifferences: true, differences: ['line 1'] }
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
const { scriptDownloaderService } = await import('~/server/services/scriptDownloader')
|
||||
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(mockScript)
|
||||
vi.mocked(scriptDownloaderService.compareScriptContent).mockResolvedValue(mockResult)
|
||||
|
||||
const result = await caller.compareScriptContent({ slug: 'test-script' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
...mockResult,
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptDiff', () => {
|
||||
it('should get script diff successfully', async () => {
|
||||
const mockScript = { name: 'Test Script', slug: 'test-script' }
|
||||
const mockResult = { diff: 'diff content' }
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
const { scriptDownloaderService } = await import('~/server/services/scriptDownloader')
|
||||
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue(mockScript)
|
||||
vi.mocked(scriptDownloaderService.getScriptDiff).mockResolvedValue(mockResult)
|
||||
|
||||
const result = await caller.getScriptDiff({ slug: 'test-script', filePath: 'test.sh' })
|
||||
|
||||
expect(result).toEqual({
|
||||
success: true,
|
||||
...mockResult,
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
326
src/server/lib/__tests__/scripts.test.ts
Normal file
326
src/server/lib/__tests__/scripts.test.ts
Normal file
@@ -0,0 +1,326 @@
|
||||
import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest'
|
||||
|
||||
// Mock the dependencies before importing ScriptManager
|
||||
vi.mock('fs/promises', async (importOriginal) => {
|
||||
const actual = await importOriginal<typeof import('fs/promises')>()
|
||||
return {
|
||||
...actual,
|
||||
readdir: vi.fn(),
|
||||
stat: vi.fn(),
|
||||
readFile: vi.fn(),
|
||||
}
|
||||
})
|
||||
|
||||
vi.mock('child_process', async (importOriginal) => {
|
||||
const actual = await importOriginal<typeof import('child_process')>()
|
||||
return {
|
||||
...actual,
|
||||
spawn: vi.fn(),
|
||||
}
|
||||
})
|
||||
|
||||
vi.mock('~/env.js', () => ({
|
||||
env: {
|
||||
SCRIPTS_DIRECTORY: '/test/scripts',
|
||||
ALLOWED_SCRIPT_EXTENSIONS: '.sh,.py,.js,.ts',
|
||||
ALLOWED_SCRIPT_PATHS: '/,/ct/',
|
||||
MAX_SCRIPT_EXECUTION_TIME: '30000',
|
||||
},
|
||||
}))
|
||||
|
||||
vi.mock('~/server/services/localScripts', () => ({
|
||||
localScriptsService: {
|
||||
getScriptBySlug: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
// Import after mocking
|
||||
import { ScriptManager } from '../scripts'
|
||||
|
||||
describe('ScriptManager', () => {
|
||||
let scriptManager: ScriptManager
|
||||
|
||||
beforeEach(async () => {
|
||||
vi.clearAllMocks()
|
||||
scriptManager = new ScriptManager()
|
||||
})
|
||||
|
||||
afterEach(() => {
|
||||
vi.clearAllMocks()
|
||||
})
|
||||
|
||||
describe('constructor', () => {
|
||||
it('should initialize with correct configuration', () => {
|
||||
const info = scriptManager.getScriptsDirectoryInfo()
|
||||
|
||||
expect(info.path).toBe('/test/scripts')
|
||||
expect(info.allowedExtensions).toEqual(['.sh', '.py', '.js', '.ts'])
|
||||
expect(info.allowedPaths).toEqual(['/', '/ct/'])
|
||||
expect(info.maxExecutionTime).toBe(30000)
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScripts', () => {
|
||||
it('should return empty array when directory read fails', async () => {
|
||||
const { readdir } = await import('fs/promises')
|
||||
vi.mocked(readdir).mockRejectedValue(new Error('Directory not found'))
|
||||
|
||||
const scripts = await scriptManager.getScripts()
|
||||
|
||||
expect(scripts).toEqual([])
|
||||
})
|
||||
|
||||
it('should return scripts with correct properties', async () => {
|
||||
const mockFiles = ['script1.sh', 'script2.py', 'script3.js', 'readme.txt']
|
||||
const { readdir, stat } = await import('fs/promises')
|
||||
|
||||
vi.mocked(readdir).mockResolvedValue(mockFiles)
|
||||
vi.mocked(stat).mockResolvedValue({
|
||||
isFile: () => true,
|
||||
isDirectory: () => false,
|
||||
size: 1024,
|
||||
mtime: new Date('2024-01-01T00:00:00Z'),
|
||||
mode: 0o755, // executable permissions
|
||||
} as any)
|
||||
|
||||
const scripts = await scriptManager.getScripts()
|
||||
|
||||
expect(scripts).toHaveLength(3) // Only .sh, .py, .js files
|
||||
expect(scripts[0]).toMatchObject({
|
||||
name: 'script1.sh',
|
||||
path: '/test/scripts/script1.sh',
|
||||
extension: '.sh',
|
||||
size: 1024,
|
||||
executable: true,
|
||||
})
|
||||
expect(scripts[1]).toMatchObject({
|
||||
name: 'script2.py',
|
||||
path: '/test/scripts/script2.py',
|
||||
extension: '.py',
|
||||
size: 1024,
|
||||
executable: true,
|
||||
})
|
||||
expect(scripts[2]).toMatchObject({
|
||||
name: 'script3.js',
|
||||
path: '/test/scripts/script3.js',
|
||||
extension: '.js',
|
||||
size: 1024,
|
||||
executable: true,
|
||||
})
|
||||
})
|
||||
|
||||
it('should sort scripts alphabetically', async () => {
|
||||
const mockFiles = ['z_script.sh', 'a_script.sh', 'm_script.sh']
|
||||
const { readdir, stat } = await import('fs/promises')
|
||||
|
||||
vi.mocked(readdir).mockResolvedValue(mockFiles)
|
||||
vi.mocked(stat).mockResolvedValue({
|
||||
isFile: () => true,
|
||||
isDirectory: () => false,
|
||||
size: 1024,
|
||||
mtime: new Date('2024-01-01T00:00:00Z'),
|
||||
mode: 0o755,
|
||||
} as any)
|
||||
|
||||
const scripts = await scriptManager.getScripts()
|
||||
|
||||
expect(scripts.map(s => s.name)).toEqual(['a_script.sh', 'm_script.sh', 'z_script.sh'])
|
||||
})
|
||||
})
|
||||
|
||||
describe('getCtScripts', () => {
|
||||
it('should return ct scripts with slug and logo', async () => {
|
||||
const mockFiles = ['test-script.sh']
|
||||
const { readdir, stat } = await import('fs/promises')
|
||||
|
||||
vi.mocked(readdir).mockResolvedValue(mockFiles)
|
||||
vi.mocked(stat).mockResolvedValue({
|
||||
isFile: () => true,
|
||||
isDirectory: () => false,
|
||||
size: 1024,
|
||||
mtime: new Date('2024-01-01T00:00:00Z'),
|
||||
mode: 0o755,
|
||||
} as any)
|
||||
|
||||
// Mock the localScriptsService
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockResolvedValue({
|
||||
logo: 'test-logo.png',
|
||||
name: 'Test Script',
|
||||
description: 'A test script',
|
||||
} as any)
|
||||
|
||||
const scripts = await scriptManager.getCtScripts()
|
||||
|
||||
expect(scripts).toHaveLength(1)
|
||||
expect(scripts[0]).toMatchObject({
|
||||
name: 'test-script.sh',
|
||||
path: '/test/scripts/ct/test-script.sh',
|
||||
slug: 'test-script',
|
||||
logo: 'test-logo.png',
|
||||
})
|
||||
})
|
||||
|
||||
it('should handle missing logo gracefully', async () => {
|
||||
const mockFiles = ['test-script.sh']
|
||||
const { readdir, stat } = await import('fs/promises')
|
||||
|
||||
vi.mocked(readdir).mockResolvedValue(mockFiles)
|
||||
vi.mocked(stat).mockResolvedValue({
|
||||
isFile: () => true,
|
||||
isDirectory: () => false,
|
||||
size: 1024,
|
||||
mtime: new Date('2024-01-01T00:00:00Z'),
|
||||
mode: 0o755,
|
||||
} as any)
|
||||
|
||||
const { localScriptsService } = await import('~/server/services/localScripts')
|
||||
vi.mocked(localScriptsService.getScriptBySlug).mockRejectedValue(new Error('Not found'))
|
||||
|
||||
const scripts = await scriptManager.getCtScripts()
|
||||
|
||||
expect(scripts).toHaveLength(1)
|
||||
expect(scripts[0].logo).toBeUndefined()
|
||||
})
|
||||
})
|
||||
|
||||
describe('validateScriptPath', () => {
|
||||
it('should validate correct script path', () => {
|
||||
const result = scriptManager.validateScriptPath('/test/scripts/valid-script.sh')
|
||||
|
||||
expect(result.valid).toBe(true)
|
||||
expect(result.message).toBeUndefined()
|
||||
})
|
||||
|
||||
it('should reject path outside scripts directory', () => {
|
||||
const result = scriptManager.validateScriptPath('/other/path/script.sh')
|
||||
|
||||
expect(result.valid).toBe(false)
|
||||
expect(result.message).toBe('Script path is not within the allowed scripts directory')
|
||||
})
|
||||
|
||||
it('should reject path not in allowed paths', () => {
|
||||
const result = scriptManager.validateScriptPath('/test/scripts/forbidden/script.sh')
|
||||
|
||||
expect(result.valid).toBe(false)
|
||||
expect(result.message).toBe('Script path is not in the allowed paths list')
|
||||
})
|
||||
|
||||
it('should reject invalid file extension', () => {
|
||||
const result = scriptManager.validateScriptPath('/test/scripts/script.exe')
|
||||
|
||||
expect(result.valid).toBe(false)
|
||||
expect(result.message).toContain('File extension')
|
||||
})
|
||||
|
||||
it('should accept ct subdirectory paths', () => {
|
||||
const result = scriptManager.validateScriptPath('/test/scripts/ct/script.sh')
|
||||
|
||||
expect(result.valid).toBe(true)
|
||||
})
|
||||
})
|
||||
|
||||
describe('executeScript', () => {
|
||||
it('should execute bash script correctly', async () => {
|
||||
const { spawn } = await import('child_process')
|
||||
const mockChildProcess = {
|
||||
kill: vi.fn(),
|
||||
on: vi.fn(),
|
||||
killed: false,
|
||||
stdout: { on: vi.fn() },
|
||||
stderr: { on: vi.fn() },
|
||||
stdin: { write: vi.fn(), end: vi.fn() },
|
||||
}
|
||||
vi.mocked(spawn).mockReturnValue(mockChildProcess as any)
|
||||
|
||||
const childProcess = await scriptManager.executeScript('/test/scripts/script.sh')
|
||||
|
||||
expect(spawn).toHaveBeenCalledWith('bash', ['/test/scripts/script.sh'], {
|
||||
cwd: '/test/scripts',
|
||||
stdio: ['pipe', 'pipe', 'pipe'],
|
||||
shell: true,
|
||||
})
|
||||
expect(childProcess).toBe(mockChildProcess)
|
||||
})
|
||||
|
||||
it('should execute python script correctly', async () => {
|
||||
const { spawn } = await import('child_process')
|
||||
const mockChildProcess = {
|
||||
kill: vi.fn(),
|
||||
on: vi.fn(),
|
||||
killed: false,
|
||||
stdout: { on: vi.fn() },
|
||||
stderr: { on: vi.fn() },
|
||||
stdin: { write: vi.fn(), end: vi.fn() },
|
||||
}
|
||||
vi.mocked(spawn).mockReturnValue(mockChildProcess as any)
|
||||
|
||||
const childProcess = await scriptManager.executeScript('/test/scripts/script.py')
|
||||
|
||||
expect(spawn).toHaveBeenCalledWith('python', ['/test/scripts/script.py'], {
|
||||
cwd: '/test/scripts',
|
||||
stdio: ['pipe', 'pipe', 'pipe'],
|
||||
shell: true,
|
||||
})
|
||||
})
|
||||
|
||||
it('should throw error for invalid script path', async () => {
|
||||
await expect(scriptManager.executeScript('/invalid/path/script.sh'))
|
||||
.rejects.toThrow('Script path is not within the allowed scripts directory')
|
||||
})
|
||||
|
||||
it('should set up timeout correctly', async () => {
|
||||
vi.useFakeTimers()
|
||||
const { spawn } = await import('child_process')
|
||||
const mockChildProcess = {
|
||||
kill: vi.fn(),
|
||||
on: vi.fn(),
|
||||
killed: false,
|
||||
stdout: { on: vi.fn() },
|
||||
stderr: { on: vi.fn() },
|
||||
stdin: { write: vi.fn(), end: vi.fn() },
|
||||
}
|
||||
vi.mocked(spawn).mockReturnValue(mockChildProcess as any)
|
||||
|
||||
await scriptManager.executeScript('/test/scripts/script.sh')
|
||||
|
||||
// Fast-forward time to trigger timeout
|
||||
vi.advanceTimersByTime(30001)
|
||||
|
||||
expect(mockChildProcess.kill).toHaveBeenCalledWith('SIGTERM')
|
||||
|
||||
vi.useRealTimers()
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptContent', () => {
|
||||
it('should return script content', async () => {
|
||||
const mockContent = '#!/bin/bash\necho "Hello World"'
|
||||
const { readFile } = await import('fs/promises')
|
||||
vi.mocked(readFile).mockResolvedValue(mockContent)
|
||||
|
||||
const content = await scriptManager.getScriptContent('/test/scripts/script.sh')
|
||||
|
||||
expect(content).toBe(mockContent)
|
||||
expect(readFile).toHaveBeenCalledWith('/test/scripts/script.sh', 'utf-8')
|
||||
})
|
||||
|
||||
it('should throw error for invalid script path', async () => {
|
||||
await expect(scriptManager.getScriptContent('/invalid/path/script.sh'))
|
||||
.rejects.toThrow('Script path is not within the allowed scripts directory')
|
||||
})
|
||||
})
|
||||
|
||||
describe('getScriptsDirectoryInfo', () => {
|
||||
it('should return correct directory information', () => {
|
||||
const info = scriptManager.getScriptsDirectoryInfo()
|
||||
|
||||
expect(info).toEqual({
|
||||
path: '/test/scripts',
|
||||
allowedExtensions: ['.sh', '.py', '.js', '.ts'],
|
||||
allowedPaths: ['/', '/ct/'],
|
||||
maxExecutionTime: 30000,
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
24
src/test/__mocks__/child_process.ts
Normal file
24
src/test/__mocks__/child_process.ts
Normal file
@@ -0,0 +1,24 @@
|
||||
import { vi } from 'vitest'
|
||||
|
||||
export const mockSpawn = vi.fn()
|
||||
|
||||
export const mockChildProcess = {
|
||||
kill: vi.fn(),
|
||||
on: vi.fn(),
|
||||
killed: false,
|
||||
stdout: {
|
||||
on: vi.fn(),
|
||||
},
|
||||
stderr: {
|
||||
on: vi.fn(),
|
||||
},
|
||||
stdin: {
|
||||
write: vi.fn(),
|
||||
end: vi.fn(),
|
||||
},
|
||||
}
|
||||
|
||||
export const resetMocks = () => {
|
||||
mockSpawn.mockReset()
|
||||
mockSpawn.mockReturnValue(mockChildProcess)
|
||||
}
|
||||
21
src/test/__mocks__/fs.ts
Normal file
21
src/test/__mocks__/fs.ts
Normal file
@@ -0,0 +1,21 @@
|
||||
import { vi } from 'vitest'
|
||||
|
||||
export const mockStats = {
|
||||
isFile: vi.fn(() => true),
|
||||
isDirectory: vi.fn(() => false),
|
||||
size: 1024,
|
||||
mtime: new Date('2024-01-01T00:00:00Z'),
|
||||
mode: 0o755, // executable permissions
|
||||
}
|
||||
|
||||
export const mockReaddir = vi.fn()
|
||||
export const mockStat = vi.fn()
|
||||
export const mockReadFile = vi.fn()
|
||||
|
||||
export const resetMocks = () => {
|
||||
mockReaddir.mockReset()
|
||||
mockStat.mockReset()
|
||||
mockReadFile.mockReset()
|
||||
mockStats.isFile.mockReset()
|
||||
mockStats.isDirectory.mockReset()
|
||||
}
|
||||
24
src/test/setup.ts
Normal file
24
src/test/setup.ts
Normal file
@@ -0,0 +1,24 @@
|
||||
import '@testing-library/jest-dom'
|
||||
import { vi } from 'vitest'
|
||||
|
||||
// Global test utilities
|
||||
global.ResizeObserver = vi.fn().mockImplementation(() => ({
|
||||
observe: vi.fn(),
|
||||
unobserve: vi.fn(),
|
||||
disconnect: vi.fn(),
|
||||
}))
|
||||
|
||||
// Mock window.matchMedia
|
||||
Object.defineProperty(window, 'matchMedia', {
|
||||
writable: true,
|
||||
value: vi.fn().mockImplementation(query => ({
|
||||
matches: false,
|
||||
media: query,
|
||||
onchange: null,
|
||||
addListener: vi.fn(), // deprecated
|
||||
removeListener: vi.fn(), // deprecated
|
||||
addEventListener: vi.fn(),
|
||||
removeEventListener: vi.fn(),
|
||||
dispatchEvent: vi.fn(),
|
||||
})),
|
||||
})
|
||||
30
vitest.config.ts
Normal file
30
vitest.config.ts
Normal file
@@ -0,0 +1,30 @@
|
||||
import { defineConfig } from 'vitest/config'
|
||||
import react from '@vitejs/plugin-react'
|
||||
import path from 'path'
|
||||
|
||||
export default defineConfig({
|
||||
plugins: [react()],
|
||||
test: {
|
||||
globals: true,
|
||||
environment: 'jsdom',
|
||||
setupFiles: ['./src/test/setup.ts'],
|
||||
include: ['**/*.{test,spec}.{js,mjs,cjs,ts,mts,cts,jsx,tsx}'],
|
||||
exclude: ['node_modules', 'dist', '.next', '.git'],
|
||||
coverage: {
|
||||
provider: 'v8',
|
||||
reporter: ['text', 'json', 'html'],
|
||||
exclude: [
|
||||
'node_modules/',
|
||||
'src/test/',
|
||||
'**/*.d.ts',
|
||||
'**/*.config.*',
|
||||
'**/coverage/**',
|
||||
],
|
||||
},
|
||||
},
|
||||
resolve: {
|
||||
alias: {
|
||||
'~': path.resolve(__dirname, './src'),
|
||||
},
|
||||
},
|
||||
})
|
||||
Reference in New Issue
Block a user