@build-astron-co/nimbus 0.4.1 → 0.4.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +268 -89
- package/README.md +26 -567
- package/dist/src/agent/compaction-agent.js +24 -12
- package/dist/src/agent/context-manager.js +2 -1
- package/dist/src/agent/expand-files.js +2 -1
- package/dist/src/agent/loop.js +71 -33
- package/dist/src/agent/permissions.js +4 -2
- package/dist/src/agent/system-prompt.js +34 -17
- package/dist/src/app.js +1 -1
- package/dist/src/auth/keychain.js +8 -4
- package/dist/src/auth/store.js +70 -107
- package/dist/src/cli/init.js +35 -19
- package/dist/src/cli/run.js +18 -10
- package/dist/src/cli/serve.js +4 -2
- package/dist/src/cli.js +52 -11
- package/dist/src/commands/alias.js +5 -3
- package/dist/src/commands/audit/index.js +2 -1
- package/dist/src/commands/aws-terraform.js +36 -18
- package/dist/src/commands/completions.js +1 -1
- package/dist/src/commands/config.js +3 -2
- package/dist/src/commands/connect-github.js +92 -0
- package/dist/src/commands/cost/index.js +3 -2
- package/dist/src/commands/deploy.js +15 -10
- package/dist/src/commands/doctor.js +9 -6
- package/dist/src/commands/drift/index.js +2 -1
- package/dist/src/commands/export.js +5 -3
- package/dist/src/commands/generate-terraform.js +110 -2
- package/dist/src/commands/import.js +3 -3
- package/dist/src/commands/incident.js +10 -5
- package/dist/src/commands/login.js +8 -93
- package/dist/src/commands/logs.js +16 -8
- package/dist/src/commands/onboarding.js +6 -4
- package/dist/src/commands/pipeline.js +6 -3
- package/dist/src/commands/plugin.js +3 -2
- package/dist/src/commands/profile.js +27 -14
- package/dist/src/commands/questionnaire.js +1 -1
- package/dist/src/commands/rollback.js +3 -2
- package/dist/src/commands/rollout.js +5 -3
- package/dist/src/commands/runbook.js +17 -10
- package/dist/src/commands/schedule.js +10 -5
- package/dist/src/commands/status.js +2 -1
- package/dist/src/commands/team-context.js +12 -7
- package/dist/src/commands/template.js +1 -1
- package/dist/src/commands/tf/index.js +6 -3
- package/dist/src/commands/upgrade.js +5 -3
- package/dist/src/commands/version.js +6 -3
- package/dist/src/commands/watch.js +6 -3
- package/dist/src/compat/sqlite.js +5 -3
- package/dist/src/config/mode-store.js +2 -1
- package/dist/src/config/profiles.js +4 -2
- package/dist/src/config/types.js +2 -1
- package/dist/src/engine/executor.js +8 -4
- package/dist/src/engine/planner.js +9 -5
- package/dist/src/llm/providers/anthropic.js +6 -3
- package/dist/src/llm/providers/ollama.js +1 -1
- package/dist/src/llm/router.js +22 -7
- package/dist/src/nimbus.js +1 -0
- package/dist/src/sessions/manager.js +6 -3
- package/dist/src/sharing/viewer.js +2 -1
- package/dist/src/tools/file-ops.js +1 -2
- package/dist/src/tools/schemas/devops.js +197 -108
- package/dist/src/tools/schemas/standard.js +1 -1
- package/dist/src/ui/App.js +25 -13
- package/dist/src/ui/FileDiffModal.js +22 -11
- package/dist/src/ui/HelpModal.js +2 -1
- package/dist/src/ui/InputBox.js +6 -3
- package/dist/src/ui/MessageList.js +40 -20
- package/dist/src/ui/TerminalPane.js +2 -1
- package/dist/src/ui/ToolCallDisplay.js +12 -6
- package/dist/src/ui/TreePane.js +2 -1
- package/dist/src/ui/ink/index.js +37 -21
- package/dist/src/version.js +1 -1
- package/dist/src/watcher/index.js +8 -4
- package/package.json +3 -5
- package/src/__tests__/alias.test.ts +0 -133
- package/src/__tests__/app.test.ts +0 -76
- package/src/__tests__/audit.test.ts +0 -877
- package/src/__tests__/circuit-breaker.test.ts +0 -116
- package/src/__tests__/cli-run.test.ts +0 -351
- package/src/__tests__/compat-sqlite.test.ts +0 -68
- package/src/__tests__/context-manager.test.ts +0 -632
- package/src/__tests__/context.test.ts +0 -242
- package/src/__tests__/devops-terminal-gaps.test.ts +0 -718
- package/src/__tests__/doctor.test.ts +0 -48
- package/src/__tests__/enterprise.test.ts +0 -401
- package/src/__tests__/export.test.ts +0 -236
- package/src/__tests__/gap-11-18-20.test.ts +0 -958
- package/src/__tests__/generator.test.ts +0 -433
- package/src/__tests__/helm-streaming.test.ts +0 -127
- package/src/__tests__/hooks.test.ts +0 -582
- package/src/__tests__/incident.test.ts +0 -179
- package/src/__tests__/init.test.ts +0 -487
- package/src/__tests__/intent-parser.test.ts +0 -229
- package/src/__tests__/llm-router.test.ts +0 -209
- package/src/__tests__/logs.test.ts +0 -107
- package/src/__tests__/loop-errors.test.ts +0 -244
- package/src/__tests__/lsp.test.ts +0 -293
- package/src/__tests__/modes.test.ts +0 -336
- package/src/__tests__/perf-optimizations.test.ts +0 -847
- package/src/__tests__/permissions.test.ts +0 -338
- package/src/__tests__/pipeline.test.ts +0 -50
- package/src/__tests__/polish-phase3.test.ts +0 -340
- package/src/__tests__/profile.test.ts +0 -237
- package/src/__tests__/rollback.test.ts +0 -83
- package/src/__tests__/runbook.test.ts +0 -219
- package/src/__tests__/schedule.test.ts +0 -206
- package/src/__tests__/serve.test.ts +0 -275
- package/src/__tests__/sessions.test.ts +0 -322
- package/src/__tests__/sharing.test.ts +0 -340
- package/src/__tests__/snapshots.test.ts +0 -581
- package/src/__tests__/standalone-migration.test.ts +0 -199
- package/src/__tests__/state-db.test.ts +0 -334
- package/src/__tests__/status.test.ts +0 -158
- package/src/__tests__/stream-with-tools.test.ts +0 -778
- package/src/__tests__/subagents.test.ts +0 -176
- package/src/__tests__/system-prompt.test.ts +0 -248
- package/src/__tests__/terminal-gap-v2.test.ts +0 -395
- package/src/__tests__/terminal-parity.test.ts +0 -393
- package/src/__tests__/tf-apply.test.ts +0 -187
- package/src/__tests__/tool-converter.test.ts +0 -256
- package/src/__tests__/tool-schemas.test.ts +0 -602
- package/src/__tests__/tools.test.ts +0 -144
- package/src/__tests__/version-json.test.ts +0 -184
- package/src/__tests__/version.test.ts +0 -49
- package/src/__tests__/watch.test.ts +0 -129
- package/src/agent/compaction-agent.ts +0 -266
- package/src/agent/context-manager.ts +0 -499
- package/src/agent/context.ts +0 -427
- package/src/agent/deploy-preview.ts +0 -487
- package/src/agent/expand-files.ts +0 -108
- package/src/agent/index.ts +0 -68
- package/src/agent/loop.ts +0 -1998
- package/src/agent/modes.ts +0 -429
- package/src/agent/permissions.ts +0 -513
- package/src/agent/subagents/base.ts +0 -116
- package/src/agent/subagents/cost.ts +0 -51
- package/src/agent/subagents/explore.ts +0 -42
- package/src/agent/subagents/general.ts +0 -54
- package/src/agent/subagents/index.ts +0 -102
- package/src/agent/subagents/infra.ts +0 -59
- package/src/agent/subagents/security.ts +0 -69
- package/src/agent/system-prompt.ts +0 -990
- package/src/app.ts +0 -180
- package/src/audit/activity-log.ts +0 -290
- package/src/audit/compliance-checker.ts +0 -540
- package/src/audit/cost-tracker.ts +0 -318
- package/src/audit/index.ts +0 -23
- package/src/audit/security-scanner.ts +0 -641
- package/src/auth/guard.ts +0 -75
- package/src/auth/index.ts +0 -56
- package/src/auth/keychain.ts +0 -82
- package/src/auth/oauth.ts +0 -465
- package/src/auth/providers.ts +0 -470
- package/src/auth/sso.ts +0 -113
- package/src/auth/store.ts +0 -505
- package/src/auth/types.ts +0 -187
- package/src/build.ts +0 -141
- package/src/cli/index.ts +0 -16
- package/src/cli/init.ts +0 -1227
- package/src/cli/openapi-spec.ts +0 -356
- package/src/cli/run.ts +0 -628
- package/src/cli/serve-auth.ts +0 -80
- package/src/cli/serve.ts +0 -539
- package/src/cli/web.ts +0 -71
- package/src/cli.ts +0 -1728
- package/src/clients/core-engine-client.ts +0 -227
- package/src/clients/enterprise-client.ts +0 -334
- package/src/clients/generator-client.ts +0 -351
- package/src/clients/git-client.ts +0 -627
- package/src/clients/github-client.ts +0 -410
- package/src/clients/helm-client.ts +0 -504
- package/src/clients/index.ts +0 -80
- package/src/clients/k8s-client.ts +0 -497
- package/src/clients/llm-client.ts +0 -161
- package/src/clients/rest-client.ts +0 -130
- package/src/clients/service-discovery.ts +0 -38
- package/src/clients/terraform-client.ts +0 -482
- package/src/clients/tools-client.ts +0 -1843
- package/src/clients/ws-client.ts +0 -115
- package/src/commands/alias.ts +0 -100
- package/src/commands/analyze/index.ts +0 -352
- package/src/commands/apply/helm.ts +0 -473
- package/src/commands/apply/index.ts +0 -213
- package/src/commands/apply/k8s.ts +0 -454
- package/src/commands/apply/terraform.ts +0 -582
- package/src/commands/ask.ts +0 -167
- package/src/commands/audit/index.ts +0 -357
- package/src/commands/auth-cloud.ts +0 -407
- package/src/commands/auth-list.ts +0 -134
- package/src/commands/auth-profile.ts +0 -121
- package/src/commands/auth-refresh.ts +0 -187
- package/src/commands/auth-status.ts +0 -141
- package/src/commands/aws/ec2.ts +0 -501
- package/src/commands/aws/iam.ts +0 -397
- package/src/commands/aws/index.ts +0 -133
- package/src/commands/aws/lambda.ts +0 -396
- package/src/commands/aws/rds.ts +0 -439
- package/src/commands/aws/s3.ts +0 -439
- package/src/commands/aws/vpc.ts +0 -393
- package/src/commands/aws-discover.ts +0 -542
- package/src/commands/aws-terraform.ts +0 -755
- package/src/commands/azure/aks.ts +0 -376
- package/src/commands/azure/functions.ts +0 -253
- package/src/commands/azure/index.ts +0 -116
- package/src/commands/azure/storage.ts +0 -478
- package/src/commands/azure/vm.ts +0 -355
- package/src/commands/billing/index.ts +0 -256
- package/src/commands/chat.ts +0 -320
- package/src/commands/completions.ts +0 -268
- package/src/commands/config.ts +0 -372
- package/src/commands/cost/cloud-cost-estimator.ts +0 -266
- package/src/commands/cost/estimator.ts +0 -79
- package/src/commands/cost/index.ts +0 -810
- package/src/commands/cost/parsers/terraform.ts +0 -273
- package/src/commands/cost/parsers/types.ts +0 -25
- package/src/commands/cost/pricing/aws.ts +0 -544
- package/src/commands/cost/pricing/azure.ts +0 -499
- package/src/commands/cost/pricing/gcp.ts +0 -396
- package/src/commands/cost/pricing/index.ts +0 -40
- package/src/commands/demo.ts +0 -250
- package/src/commands/deploy.ts +0 -260
- package/src/commands/doctor.ts +0 -1386
- package/src/commands/drift/index.ts +0 -787
- package/src/commands/explain.ts +0 -277
- package/src/commands/export.ts +0 -146
- package/src/commands/feedback.ts +0 -389
- package/src/commands/fix.ts +0 -324
- package/src/commands/fs/index.ts +0 -402
- package/src/commands/gcp/compute.ts +0 -325
- package/src/commands/gcp/functions.ts +0 -271
- package/src/commands/gcp/gke.ts +0 -438
- package/src/commands/gcp/iam.ts +0 -344
- package/src/commands/gcp/index.ts +0 -129
- package/src/commands/gcp/storage.ts +0 -284
- package/src/commands/generate-helm.ts +0 -1249
- package/src/commands/generate-k8s.ts +0 -1508
- package/src/commands/generate-terraform.ts +0 -1202
- package/src/commands/gh/index.ts +0 -863
- package/src/commands/git/index.ts +0 -1343
- package/src/commands/helm/index.ts +0 -1126
- package/src/commands/help.ts +0 -715
- package/src/commands/history.ts +0 -149
- package/src/commands/import.ts +0 -868
- package/src/commands/incident.ts +0 -166
- package/src/commands/index.ts +0 -367
- package/src/commands/init.ts +0 -1051
- package/src/commands/k8s/index.ts +0 -1137
- package/src/commands/login.ts +0 -716
- package/src/commands/logout.ts +0 -83
- package/src/commands/logs.ts +0 -167
- package/src/commands/onboarding.ts +0 -405
- package/src/commands/pipeline.ts +0 -186
- package/src/commands/plan/display.ts +0 -279
- package/src/commands/plan/index.ts +0 -599
- package/src/commands/plugin.ts +0 -398
- package/src/commands/preview.ts +0 -452
- package/src/commands/profile.ts +0 -342
- package/src/commands/questionnaire.ts +0 -1172
- package/src/commands/resume.ts +0 -47
- package/src/commands/rollback.ts +0 -315
- package/src/commands/rollout.ts +0 -88
- package/src/commands/runbook.ts +0 -346
- package/src/commands/schedule.ts +0 -236
- package/src/commands/status.ts +0 -252
- package/src/commands/team/index.ts +0 -346
- package/src/commands/team-context.ts +0 -220
- package/src/commands/template.ts +0 -233
- package/src/commands/tf/index.ts +0 -1093
- package/src/commands/upgrade.ts +0 -607
- package/src/commands/usage/index.ts +0 -134
- package/src/commands/version.ts +0 -174
- package/src/commands/watch.ts +0 -153
- package/src/compat/index.ts +0 -2
- package/src/compat/runtime.ts +0 -12
- package/src/compat/sqlite.ts +0 -177
- package/src/config/index.ts +0 -17
- package/src/config/manager.ts +0 -530
- package/src/config/mode-store.ts +0 -62
- package/src/config/profiles.ts +0 -84
- package/src/config/safety-policy.ts +0 -358
- package/src/config/schema.ts +0 -125
- package/src/config/types.ts +0 -609
- package/src/config/workspace-state.ts +0 -53
- package/src/context/context-db.ts +0 -199
- package/src/demo/index.ts +0 -349
- package/src/demo/scenarios/full-journey.ts +0 -229
- package/src/demo/scenarios/getting-started.ts +0 -127
- package/src/demo/scenarios/helm-release.ts +0 -341
- package/src/demo/scenarios/k8s-deployment.ts +0 -194
- package/src/demo/scenarios/terraform-vpc.ts +0 -170
- package/src/demo/types.ts +0 -92
- package/src/engine/cost-estimator.ts +0 -480
- package/src/engine/diagram-generator.ts +0 -256
- package/src/engine/drift-detector.ts +0 -902
- package/src/engine/executor.ts +0 -1066
- package/src/engine/index.ts +0 -76
- package/src/engine/orchestrator.ts +0 -636
- package/src/engine/planner.ts +0 -787
- package/src/engine/safety.ts +0 -743
- package/src/engine/verifier.ts +0 -770
- package/src/enterprise/audit.ts +0 -348
- package/src/enterprise/auth.ts +0 -270
- package/src/enterprise/billing.ts +0 -822
- package/src/enterprise/index.ts +0 -17
- package/src/enterprise/teams.ts +0 -443
- package/src/generator/best-practices.ts +0 -1608
- package/src/generator/helm.ts +0 -630
- package/src/generator/index.ts +0 -37
- package/src/generator/intent-parser.ts +0 -514
- package/src/generator/kubernetes.ts +0 -976
- package/src/generator/terraform.ts +0 -1875
- package/src/history/index.ts +0 -8
- package/src/history/manager.ts +0 -250
- package/src/history/types.ts +0 -34
- package/src/hooks/config.ts +0 -432
- package/src/hooks/engine.ts +0 -392
- package/src/hooks/index.ts +0 -4
- package/src/llm/auth-bridge.ts +0 -198
- package/src/llm/circuit-breaker.ts +0 -140
- package/src/llm/config-loader.ts +0 -201
- package/src/llm/cost-calculator.ts +0 -171
- package/src/llm/index.ts +0 -8
- package/src/llm/model-aliases.ts +0 -115
- package/src/llm/provider-registry.ts +0 -63
- package/src/llm/providers/anthropic.ts +0 -462
- package/src/llm/providers/bedrock.ts +0 -477
- package/src/llm/providers/google.ts +0 -405
- package/src/llm/providers/ollama.ts +0 -767
- package/src/llm/providers/openai-compatible.ts +0 -340
- package/src/llm/providers/openai.ts +0 -328
- package/src/llm/providers/openrouter.ts +0 -338
- package/src/llm/router.ts +0 -1104
- package/src/llm/types.ts +0 -232
- package/src/lsp/client.ts +0 -298
- package/src/lsp/languages.ts +0 -119
- package/src/lsp/manager.ts +0 -294
- package/src/mcp/client.ts +0 -402
- package/src/mcp/index.ts +0 -5
- package/src/mcp/manager.ts +0 -133
- package/src/nimbus.ts +0 -233
- package/src/plugins/index.ts +0 -27
- package/src/plugins/loader.ts +0 -334
- package/src/plugins/manager.ts +0 -376
- package/src/plugins/types.ts +0 -284
- package/src/scanners/cicd-scanner.ts +0 -258
- package/src/scanners/cloud-scanner.ts +0 -466
- package/src/scanners/framework-scanner.ts +0 -469
- package/src/scanners/iac-scanner.ts +0 -388
- package/src/scanners/index.ts +0 -539
- package/src/scanners/language-scanner.ts +0 -276
- package/src/scanners/package-manager-scanner.ts +0 -277
- package/src/scanners/types.ts +0 -172
- package/src/sessions/manager.ts +0 -472
- package/src/sessions/types.ts +0 -44
- package/src/sharing/sync.ts +0 -300
- package/src/sharing/viewer.ts +0 -163
- package/src/snapshots/index.ts +0 -2
- package/src/snapshots/manager.ts +0 -530
- package/src/state/artifacts.ts +0 -147
- package/src/state/audit.ts +0 -137
- package/src/state/billing.ts +0 -240
- package/src/state/checkpoints.ts +0 -117
- package/src/state/config.ts +0 -67
- package/src/state/conversations.ts +0 -14
- package/src/state/credentials.ts +0 -154
- package/src/state/db.ts +0 -58
- package/src/state/index.ts +0 -26
- package/src/state/messages.ts +0 -115
- package/src/state/projects.ts +0 -123
- package/src/state/schema.ts +0 -236
- package/src/state/sessions.ts +0 -147
- package/src/state/teams.ts +0 -200
- package/src/telemetry.ts +0 -108
- package/src/tools/aws-ops.ts +0 -952
- package/src/tools/azure-ops.ts +0 -579
- package/src/tools/file-ops.ts +0 -615
- package/src/tools/gcp-ops.ts +0 -625
- package/src/tools/git-ops.ts +0 -773
- package/src/tools/github-ops.ts +0 -799
- package/src/tools/helm-ops.ts +0 -943
- package/src/tools/index.ts +0 -17
- package/src/tools/k8s-ops.ts +0 -819
- package/src/tools/schemas/converter.ts +0 -184
- package/src/tools/schemas/devops.ts +0 -3502
- package/src/tools/schemas/index.ts +0 -73
- package/src/tools/schemas/standard.ts +0 -1148
- package/src/tools/schemas/types.ts +0 -735
- package/src/tools/spawn-exec.ts +0 -148
- package/src/tools/terraform-ops.ts +0 -862
- package/src/types/ambient.d.ts +0 -193
- package/src/types/config.ts +0 -83
- package/src/types/drift.ts +0 -116
- package/src/types/enterprise.ts +0 -335
- package/src/types/index.ts +0 -20
- package/src/types/plan.ts +0 -44
- package/src/types/request.ts +0 -65
- package/src/types/response.ts +0 -54
- package/src/types/service.ts +0 -51
- package/src/ui/App.tsx +0 -2114
- package/src/ui/DeployPreview.tsx +0 -174
- package/src/ui/FileDiffModal.tsx +0 -162
- package/src/ui/Header.tsx +0 -131
- package/src/ui/HelpModal.tsx +0 -57
- package/src/ui/InputBox.tsx +0 -503
- package/src/ui/MessageList.tsx +0 -1032
- package/src/ui/PermissionPrompt.tsx +0 -163
- package/src/ui/StatusBar.tsx +0 -277
- package/src/ui/TerminalPane.tsx +0 -84
- package/src/ui/ToolCallDisplay.tsx +0 -643
- package/src/ui/TreePane.tsx +0 -132
- package/src/ui/chat-ui.ts +0 -850
- package/src/ui/index.ts +0 -33
- package/src/ui/ink/index.ts +0 -1444
- package/src/ui/streaming.ts +0 -176
- package/src/ui/theme.ts +0 -104
- package/src/ui/types.ts +0 -75
- package/src/utils/analytics.ts +0 -72
- package/src/utils/cost-warning.ts +0 -27
- package/src/utils/env.ts +0 -46
- package/src/utils/errors.ts +0 -69
- package/src/utils/event-bus.ts +0 -38
- package/src/utils/index.ts +0 -24
- package/src/utils/logger.ts +0 -171
- package/src/utils/rate-limiter.ts +0 -121
- package/src/utils/service-auth.ts +0 -49
- package/src/utils/validation.ts +0 -53
- package/src/version.ts +0 -4
- package/src/watcher/index.ts +0 -214
- package/src/wizard/approval.ts +0 -383
- package/src/wizard/index.ts +0 -25
- package/src/wizard/prompts.ts +0 -338
- package/src/wizard/types.ts +0 -172
- package/src/wizard/ui.ts +0 -556
- package/src/wizard/wizard.ts +0 -304
- package/tsconfig.json +0 -24
|
@@ -1,48 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Doctor Command Tests — C1
|
|
3
|
-
*
|
|
4
|
-
* Verifies that process.exit(1) is called when any check fails.
|
|
5
|
-
*/
|
|
6
|
-
|
|
7
|
-
import { describe, test, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
8
|
-
|
|
9
|
-
// We need to mock process.exit before importing doctorCommand
|
|
10
|
-
const exitSpy = vi.spyOn(process, 'exit').mockImplementation((() => {}) as (code?: string | number | null) => never);
|
|
11
|
-
|
|
12
|
-
describe('doctorCommand exit codes (C1)', () => {
|
|
13
|
-
beforeEach(() => {
|
|
14
|
-
exitSpy.mockClear();
|
|
15
|
-
});
|
|
16
|
-
|
|
17
|
-
afterEach(() => {
|
|
18
|
-
vi.clearAllMocks();
|
|
19
|
-
});
|
|
20
|
-
|
|
21
|
-
test('exits 0 when all checks pass (spy stays uncalled)', async () => {
|
|
22
|
-
// A simple unit test: verify the exit spy pattern works
|
|
23
|
-
// We don't call process.exit here, so the spy should not be called
|
|
24
|
-
expect(exitSpy).not.toHaveBeenCalled();
|
|
25
|
-
});
|
|
26
|
-
|
|
27
|
-
test('process.exit(1) is called after JSON output when a check fails', async () => {
|
|
28
|
-
// Simulate what doctorCommand does on failure
|
|
29
|
-
const allPassed = false;
|
|
30
|
-
if (!allPassed) process.exit(1);
|
|
31
|
-
expect(exitSpy).toHaveBeenCalledWith(1);
|
|
32
|
-
});
|
|
33
|
-
|
|
34
|
-
test('process.exit(1) is called in summary section when a check fails', async () => {
|
|
35
|
-
exitSpy.mockClear();
|
|
36
|
-
// Simulate text-mode failure path
|
|
37
|
-
const allPassed = false;
|
|
38
|
-
if (!allPassed) process.exit(1);
|
|
39
|
-
expect(exitSpy).toHaveBeenCalledWith(1);
|
|
40
|
-
});
|
|
41
|
-
|
|
42
|
-
test('process.exit is not called when all checks pass', async () => {
|
|
43
|
-
exitSpy.mockClear();
|
|
44
|
-
const allPassed = true;
|
|
45
|
-
if (!allPassed) process.exit(1);
|
|
46
|
-
expect(exitSpy).not.toHaveBeenCalled();
|
|
47
|
-
});
|
|
48
|
-
});
|
|
@@ -1,401 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Tests for the enterprise modules:
|
|
3
|
-
* - src/enterprise/auth.ts – initiateDeviceFlow, pollDeviceCode, verifyDeviceCode, validateToken
|
|
4
|
-
* - src/enterprise/teams.ts – createTeam, getTeam, listUserTeams, inviteMember, listMembers
|
|
5
|
-
* - src/enterprise/billing.ts – subscribe, getBillingStatus, recordUsage, getUsage
|
|
6
|
-
* - src/enterprise/audit.ts – createLog, queryLogs, exportLogs
|
|
7
|
-
*
|
|
8
|
-
* Every test uses an isolated in-memory SQLite database via NIMBUS_DB_PATH=':memory:'
|
|
9
|
-
* so the real ~/.nimbus/nimbus.db is never touched.
|
|
10
|
-
*
|
|
11
|
-
* IMPORTANT: The enterprise modules use `getDb()` (singleton) internally when
|
|
12
|
-
* they reach for the database. We set NIMBUS_DB_PATH to ':memory:' before
|
|
13
|
-
* each test group and reset the db singleton afterwards via `closeDb()`.
|
|
14
|
-
*/
|
|
15
|
-
|
|
16
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
17
|
-
import { getTestDb, closeDb } from '../state/db';
|
|
18
|
-
|
|
19
|
-
// ---------------------------------------------------------------------------
|
|
20
|
-
// Setup: point every getDb() call to a fresh in-memory database.
|
|
21
|
-
// ---------------------------------------------------------------------------
|
|
22
|
-
|
|
23
|
-
beforeEach(() => {
|
|
24
|
-
// closeDb() resets the db singleton so the next getDb() call re-opens.
|
|
25
|
-
closeDb();
|
|
26
|
-
// Re-initialise using NIMBUS_DB_PATH=':memory:' by creating a test db
|
|
27
|
-
// and setting the env variable so subsequent getDb() calls hit :memory:.
|
|
28
|
-
process.env.NIMBUS_DB_PATH = ':memory:';
|
|
29
|
-
// Warm-up the singleton with migrations applied
|
|
30
|
-
getTestDb(); // just to confirm migration works; enterprise fns call getDb() on their own
|
|
31
|
-
});
|
|
32
|
-
|
|
33
|
-
afterEach(() => {
|
|
34
|
-
closeDb();
|
|
35
|
-
delete process.env.NIMBUS_DB_PATH;
|
|
36
|
-
});
|
|
37
|
-
|
|
38
|
-
// ---------------------------------------------------------------------------
|
|
39
|
-
// Auth: device-authorization flow
|
|
40
|
-
// ---------------------------------------------------------------------------
|
|
41
|
-
|
|
42
|
-
describe('enterprise auth – initiateDeviceFlow', () => {
|
|
43
|
-
it('returns a DeviceCodeResponse with all expected fields', async () => {
|
|
44
|
-
const { initiateDeviceFlow } = await import('../enterprise/auth');
|
|
45
|
-
const response = await initiateDeviceFlow();
|
|
46
|
-
|
|
47
|
-
expect(typeof response.deviceCode).toBe('string');
|
|
48
|
-
expect(response.deviceCode.length).toBeGreaterThan(0);
|
|
49
|
-
expect(typeof response.userCode).toBe('string');
|
|
50
|
-
// User code format: XXXX-NNNN
|
|
51
|
-
expect(/^[A-Z]{4}-\d{4}$/.test(response.userCode)).toBe(true);
|
|
52
|
-
expect(typeof response.verificationUri).toBe('string');
|
|
53
|
-
expect(response.expiresIn).toBe(900);
|
|
54
|
-
expect(response.interval).toBe(5);
|
|
55
|
-
});
|
|
56
|
-
|
|
57
|
-
it('generates a unique deviceCode on each call', async () => {
|
|
58
|
-
const { initiateDeviceFlow } = await import('../enterprise/auth');
|
|
59
|
-
const r1 = await initiateDeviceFlow();
|
|
60
|
-
const r2 = await initiateDeviceFlow();
|
|
61
|
-
expect(r1.deviceCode).not.toBe(r2.deviceCode);
|
|
62
|
-
expect(r1.userCode).not.toBe(r2.userCode);
|
|
63
|
-
});
|
|
64
|
-
|
|
65
|
-
it('userCode consists of uppercase letters and digits only', async () => {
|
|
66
|
-
const { initiateDeviceFlow } = await import('../enterprise/auth');
|
|
67
|
-
const { userCode } = await initiateDeviceFlow();
|
|
68
|
-
// Valid chars: uppercase ASCII letters (no I or O) + digits + hyphen
|
|
69
|
-
expect(/^[A-HJ-NP-Z0-9]{4}-[0-9]{4}$/.test(userCode)).toBe(true);
|
|
70
|
-
});
|
|
71
|
-
});
|
|
72
|
-
|
|
73
|
-
describe('enterprise auth – pollDeviceCode', () => {
|
|
74
|
-
it('returns authorization_pending for a newly created device code', async () => {
|
|
75
|
-
const { initiateDeviceFlow, pollDeviceCode } = await import('../enterprise/auth');
|
|
76
|
-
const { deviceCode } = await initiateDeviceFlow();
|
|
77
|
-
const response = await pollDeviceCode(deviceCode);
|
|
78
|
-
|
|
79
|
-
expect(response.error).toBe('authorization_pending');
|
|
80
|
-
expect(response.accessToken).toBeUndefined();
|
|
81
|
-
});
|
|
82
|
-
|
|
83
|
-
it('returns expired_token for an unknown device code', async () => {
|
|
84
|
-
const { pollDeviceCode } = await import('../enterprise/auth');
|
|
85
|
-
const response = await pollDeviceCode('nonexistent-device-code');
|
|
86
|
-
expect(response.error).toBe('expired_token');
|
|
87
|
-
});
|
|
88
|
-
});
|
|
89
|
-
|
|
90
|
-
describe('enterprise auth – validateToken', () => {
|
|
91
|
-
it('returns valid: false for a non-existent token', async () => {
|
|
92
|
-
const { validateToken } = await import('../enterprise/auth');
|
|
93
|
-
const result = await validateToken({ accessToken: 'not-a-real-token' });
|
|
94
|
-
expect(result.valid).toBe(false);
|
|
95
|
-
});
|
|
96
|
-
|
|
97
|
-
it('returns valid: false for an empty token string', async () => {
|
|
98
|
-
const { validateToken } = await import('../enterprise/auth');
|
|
99
|
-
const result = await validateToken({ accessToken: '' });
|
|
100
|
-
expect(result.valid).toBe(false);
|
|
101
|
-
});
|
|
102
|
-
});
|
|
103
|
-
|
|
104
|
-
// ---------------------------------------------------------------------------
|
|
105
|
-
// Teams
|
|
106
|
-
// ---------------------------------------------------------------------------
|
|
107
|
-
|
|
108
|
-
describe('enterprise teams – createTeam', () => {
|
|
109
|
-
it('creates a team and returns a Team object', async () => {
|
|
110
|
-
const { createTeam } = await import('../enterprise/teams');
|
|
111
|
-
const team = await createTeam({ name: 'Nimbus Core', ownerId: 'user-001' });
|
|
112
|
-
|
|
113
|
-
expect(typeof team.id).toBe('string');
|
|
114
|
-
expect(team.id.length).toBeGreaterThan(0);
|
|
115
|
-
expect(team.name).toBe('Nimbus Core');
|
|
116
|
-
expect(team.ownerId).toBe('user-001');
|
|
117
|
-
expect(team.plan).toBe('free');
|
|
118
|
-
});
|
|
119
|
-
|
|
120
|
-
it('throws when name is missing', async () => {
|
|
121
|
-
const { createTeam } = await import('../enterprise/teams');
|
|
122
|
-
await expect(createTeam({ name: '', ownerId: 'user-001' })).rejects.toThrow();
|
|
123
|
-
});
|
|
124
|
-
|
|
125
|
-
it('throws when ownerId is missing', async () => {
|
|
126
|
-
const { createTeam } = await import('../enterprise/teams');
|
|
127
|
-
await expect(createTeam({ name: 'Team', ownerId: '' })).rejects.toThrow();
|
|
128
|
-
});
|
|
129
|
-
});
|
|
130
|
-
|
|
131
|
-
describe('enterprise teams – getTeam', () => {
|
|
132
|
-
it('retrieves a team that was just created', async () => {
|
|
133
|
-
const { createTeam, getTeam } = await import('../enterprise/teams');
|
|
134
|
-
const created = await createTeam({ name: 'Alpha Team', ownerId: 'u-alpha' });
|
|
135
|
-
const fetched = await getTeam(created.id);
|
|
136
|
-
|
|
137
|
-
expect(fetched).not.toBeNull();
|
|
138
|
-
expect(fetched!.id).toBe(created.id);
|
|
139
|
-
expect(fetched!.name).toBe('Alpha Team');
|
|
140
|
-
});
|
|
141
|
-
|
|
142
|
-
it('returns null for a non-existent team id', async () => {
|
|
143
|
-
const { getTeam } = await import('../enterprise/teams');
|
|
144
|
-
const result = await getTeam('non-existent-id');
|
|
145
|
-
expect(result).toBeNull();
|
|
146
|
-
});
|
|
147
|
-
});
|
|
148
|
-
|
|
149
|
-
describe('enterprise teams – listUserTeams', () => {
|
|
150
|
-
it('returns teams where user is a member', async () => {
|
|
151
|
-
const { createTeam, listUserTeams } = await import('../enterprise/teams');
|
|
152
|
-
await createTeam({ name: 'My Team', ownerId: 'user-list' });
|
|
153
|
-
const teams = await listUserTeams('user-list');
|
|
154
|
-
|
|
155
|
-
expect(Array.isArray(teams)).toBe(true);
|
|
156
|
-
expect(teams.length).toBeGreaterThanOrEqual(1);
|
|
157
|
-
expect(teams.some(t => t.name === 'My Team')).toBe(true);
|
|
158
|
-
});
|
|
159
|
-
|
|
160
|
-
it('returns empty array for user with no team memberships', async () => {
|
|
161
|
-
const { listUserTeams } = await import('../enterprise/teams');
|
|
162
|
-
const teams = await listUserTeams('user-with-no-teams-xyz');
|
|
163
|
-
expect(Array.isArray(teams)).toBe(true);
|
|
164
|
-
expect(teams.length).toBe(0);
|
|
165
|
-
});
|
|
166
|
-
});
|
|
167
|
-
|
|
168
|
-
describe('enterprise teams – inviteMember', () => {
|
|
169
|
-
it('invites a new member to a team by email', async () => {
|
|
170
|
-
const { createTeam, inviteMember } = await import('../enterprise/teams');
|
|
171
|
-
const team = await createTeam({ name: 'Invite Team', ownerId: 'owner-u' });
|
|
172
|
-
const member = await inviteMember(team.id, { email: 'new@example.com', role: 'member' });
|
|
173
|
-
|
|
174
|
-
expect(member.teamId).toBe(team.id);
|
|
175
|
-
expect(member.role).toBe('member');
|
|
176
|
-
expect(member.user?.email).toBe('new@example.com');
|
|
177
|
-
});
|
|
178
|
-
|
|
179
|
-
it('throws for an invalid role', async () => {
|
|
180
|
-
const { createTeam, inviteMember } = await import('../enterprise/teams');
|
|
181
|
-
const team = await createTeam({ name: 'Role Team', ownerId: 'owner-r' });
|
|
182
|
-
await expect(
|
|
183
|
-
inviteMember(team.id, { email: 'x@example.com', role: 'superadmin' })
|
|
184
|
-
).rejects.toThrow();
|
|
185
|
-
});
|
|
186
|
-
});
|
|
187
|
-
|
|
188
|
-
describe('enterprise teams – listMembers', () => {
|
|
189
|
-
it('lists members including the team owner', async () => {
|
|
190
|
-
const { createTeam, listMembers } = await import('../enterprise/teams');
|
|
191
|
-
const team = await createTeam({ name: 'List Members Team', ownerId: 'owner-lm' });
|
|
192
|
-
const members = await listMembers(team.id);
|
|
193
|
-
|
|
194
|
-
expect(Array.isArray(members)).toBe(true);
|
|
195
|
-
expect(members.length).toBeGreaterThanOrEqual(1);
|
|
196
|
-
// Owner is automatically added as a member with role 'owner'
|
|
197
|
-
const ownerEntry = members.find(m => m.role === 'owner');
|
|
198
|
-
expect(ownerEntry).toBeDefined();
|
|
199
|
-
});
|
|
200
|
-
});
|
|
201
|
-
|
|
202
|
-
// ---------------------------------------------------------------------------
|
|
203
|
-
// Billing
|
|
204
|
-
// ---------------------------------------------------------------------------
|
|
205
|
-
|
|
206
|
-
describe('enterprise billing – subscribe', () => {
|
|
207
|
-
it('creates a subscription and returns billing status', async () => {
|
|
208
|
-
const { createTeam } = await import('../enterprise/teams');
|
|
209
|
-
const { subscribe } = await import('../enterprise/billing');
|
|
210
|
-
// A subscription has a FK on team_id; create the team first.
|
|
211
|
-
const team = await createTeam({ name: 'Sub Team Pro', ownerId: 'owner-sub-001' });
|
|
212
|
-
const result = await subscribe({ teamId: team.id, plan: 'pro' });
|
|
213
|
-
|
|
214
|
-
expect(result).toBeDefined();
|
|
215
|
-
expect(result.billing.plan).toBe('pro');
|
|
216
|
-
expect(result.billing.status).toBe('active');
|
|
217
|
-
expect(result.stripe).toBeDefined();
|
|
218
|
-
expect(result.stripe.object).toBe('subscription');
|
|
219
|
-
});
|
|
220
|
-
|
|
221
|
-
it('throws for an invalid plan name', async () => {
|
|
222
|
-
const { subscribe } = await import('../enterprise/billing');
|
|
223
|
-
await expect(subscribe({ teamId: 'team-bad', plan: 'diamond' })).rejects.toThrow();
|
|
224
|
-
});
|
|
225
|
-
|
|
226
|
-
it('throws when teamId is empty', async () => {
|
|
227
|
-
const { subscribe } = await import('../enterprise/billing');
|
|
228
|
-
await expect(subscribe({ teamId: '', plan: 'pro' })).rejects.toThrow();
|
|
229
|
-
});
|
|
230
|
-
});
|
|
231
|
-
|
|
232
|
-
describe('enterprise billing – getBillingStatus', () => {
|
|
233
|
-
it('returns free plan defaults for a team with no subscription', async () => {
|
|
234
|
-
const { getBillingStatus } = await import('../enterprise/billing');
|
|
235
|
-
const status = await getBillingStatus('team-no-sub');
|
|
236
|
-
|
|
237
|
-
expect(status.plan).toBe('free');
|
|
238
|
-
expect(status.status).toBe('active');
|
|
239
|
-
expect(status.cancelAtPeriodEnd).toBe(false);
|
|
240
|
-
});
|
|
241
|
-
|
|
242
|
-
it('returns active status after subscribing', async () => {
|
|
243
|
-
const { createTeam } = await import('../enterprise/teams');
|
|
244
|
-
const { subscribe, getBillingStatus } = await import('../enterprise/billing');
|
|
245
|
-
// A subscription has a FK on team_id; create the team first.
|
|
246
|
-
const team = await createTeam({ name: 'Status Team', ownerId: 'owner-status' });
|
|
247
|
-
await subscribe({ teamId: team.id, plan: 'enterprise' });
|
|
248
|
-
const status = await getBillingStatus(team.id);
|
|
249
|
-
|
|
250
|
-
expect(status.plan).toBe('enterprise');
|
|
251
|
-
expect(status.status).toBe('active');
|
|
252
|
-
});
|
|
253
|
-
});
|
|
254
|
-
|
|
255
|
-
describe('enterprise billing – recordUsage', () => {
|
|
256
|
-
it('records a usage event and returns a confirmation receipt', async () => {
|
|
257
|
-
const { recordUsage } = await import('../enterprise/billing');
|
|
258
|
-
const result = await recordUsage({
|
|
259
|
-
teamId: 'team-usage',
|
|
260
|
-
operationType: 'generate',
|
|
261
|
-
tokensUsed: 1000,
|
|
262
|
-
costUsd: 0.003,
|
|
263
|
-
});
|
|
264
|
-
|
|
265
|
-
expect(result.recorded).toBe(true);
|
|
266
|
-
expect(typeof result.id).toBe('string');
|
|
267
|
-
expect(result.teamId).toBe('team-usage');
|
|
268
|
-
expect(result.operationType).toBe('generate');
|
|
269
|
-
expect(result.tokensUsed).toBe(1000);
|
|
270
|
-
expect(result.costUsd).toBe(0.003);
|
|
271
|
-
expect(typeof result.timestamp).toBe('string');
|
|
272
|
-
});
|
|
273
|
-
|
|
274
|
-
it('throws for negative tokensUsed', async () => {
|
|
275
|
-
const { recordUsage } = await import('../enterprise/billing');
|
|
276
|
-
await expect(
|
|
277
|
-
recordUsage({ teamId: 'team-bad', operationType: 'gen', tokensUsed: -1, costUsd: 0 })
|
|
278
|
-
).rejects.toThrow();
|
|
279
|
-
});
|
|
280
|
-
|
|
281
|
-
it('throws for negative costUsd', async () => {
|
|
282
|
-
const { recordUsage } = await import('../enterprise/billing');
|
|
283
|
-
await expect(
|
|
284
|
-
recordUsage({ teamId: 'team-bad', operationType: 'gen', tokensUsed: 0, costUsd: -0.5 })
|
|
285
|
-
).rejects.toThrow();
|
|
286
|
-
});
|
|
287
|
-
});
|
|
288
|
-
|
|
289
|
-
describe('enterprise billing – getUsage', () => {
|
|
290
|
-
it('returns an EnhancedUsageSummary with expected shape', async () => {
|
|
291
|
-
const { getUsage } = await import('../enterprise/billing');
|
|
292
|
-
const summary = await getUsage('team-get-usage', 'month');
|
|
293
|
-
|
|
294
|
-
expect(summary).toBeDefined();
|
|
295
|
-
expect(typeof summary.totals.operations).toBe('number');
|
|
296
|
-
expect(typeof summary.totals.tokensUsed).toBe('number');
|
|
297
|
-
expect(typeof summary.totals.costUsd).toBe('number');
|
|
298
|
-
expect(summary.quota).toBeDefined();
|
|
299
|
-
expect(summary.quota.plan).toBe('free'); // No subscription → free plan
|
|
300
|
-
expect(typeof summary.rateLimit.requestsPerMinute).toBe('number');
|
|
301
|
-
});
|
|
302
|
-
});
|
|
303
|
-
|
|
304
|
-
// ---------------------------------------------------------------------------
|
|
305
|
-
// Audit
|
|
306
|
-
// ---------------------------------------------------------------------------
|
|
307
|
-
|
|
308
|
-
describe('enterprise audit – createLog', () => {
|
|
309
|
-
it('creates an audit log entry and returns an AuditLog', async () => {
|
|
310
|
-
const { createLog } = await import('../enterprise/audit');
|
|
311
|
-
const log = await createLog({
|
|
312
|
-
action: 'generate.terraform',
|
|
313
|
-
status: 'success',
|
|
314
|
-
userId: 'user-audit-001',
|
|
315
|
-
resourceType: 'terraform',
|
|
316
|
-
resourceId: 'vpc-module',
|
|
317
|
-
});
|
|
318
|
-
|
|
319
|
-
expect(typeof log.id).toBe('string');
|
|
320
|
-
expect(log.id.length).toBeGreaterThan(0);
|
|
321
|
-
expect(log.action).toBe('generate.terraform');
|
|
322
|
-
expect(log.status).toBe('success');
|
|
323
|
-
expect(log.userId).toBe('user-audit-001');
|
|
324
|
-
expect(log.resourceType).toBe('terraform');
|
|
325
|
-
expect(typeof log.timestamp).toBe('string');
|
|
326
|
-
});
|
|
327
|
-
|
|
328
|
-
it('throws when action is missing', async () => {
|
|
329
|
-
const { createLog } = await import('../enterprise/audit');
|
|
330
|
-
await expect(createLog({ action: '', status: 'success' })).rejects.toThrow();
|
|
331
|
-
});
|
|
332
|
-
|
|
333
|
-
it('throws when status is missing', async () => {
|
|
334
|
-
const { createLog } = await import('../enterprise/audit');
|
|
335
|
-
await expect(createLog({ action: 'do.thing', status: '' })).rejects.toThrow();
|
|
336
|
-
});
|
|
337
|
-
|
|
338
|
-
it('stores teamId and ipAddress in the returned log', async () => {
|
|
339
|
-
const { createLog } = await import('../enterprise/audit');
|
|
340
|
-
const log = await createLog({
|
|
341
|
-
action: 'login',
|
|
342
|
-
status: 'success',
|
|
343
|
-
teamId: 'team-audit',
|
|
344
|
-
ipAddress: '127.0.0.1',
|
|
345
|
-
});
|
|
346
|
-
|
|
347
|
-
expect(log.teamId).toBe('team-audit');
|
|
348
|
-
expect(log.ipAddress).toBe('127.0.0.1');
|
|
349
|
-
});
|
|
350
|
-
});
|
|
351
|
-
|
|
352
|
-
describe('enterprise audit – queryLogs', () => {
|
|
353
|
-
it('returns paginated results with total count', async () => {
|
|
354
|
-
const { createLog, queryLogs } = await import('../enterprise/audit');
|
|
355
|
-
|
|
356
|
-
await createLog({ action: 'action.a', status: 'success' });
|
|
357
|
-
await createLog({ action: 'action.b', status: 'failure' });
|
|
358
|
-
|
|
359
|
-
const result = await queryLogs({ limit: 10, offset: 0 });
|
|
360
|
-
|
|
361
|
-
expect(Array.isArray(result.logs)).toBe(true);
|
|
362
|
-
expect(typeof result.total).toBe('number');
|
|
363
|
-
expect(typeof result.limit).toBe('number');
|
|
364
|
-
expect(typeof result.offset).toBe('number');
|
|
365
|
-
expect(result.total).toBeGreaterThanOrEqual(2);
|
|
366
|
-
});
|
|
367
|
-
|
|
368
|
-
it('filters by action when action param is provided', async () => {
|
|
369
|
-
const { createLog, queryLogs } = await import('../enterprise/audit');
|
|
370
|
-
|
|
371
|
-
await createLog({ action: 'filtered.action', status: 'success' });
|
|
372
|
-
await createLog({ action: 'other.action', status: 'success' });
|
|
373
|
-
|
|
374
|
-
const result = await queryLogs({ action: 'filtered.action' });
|
|
375
|
-
expect(result.logs.every(l => l.action === 'filtered.action')).toBe(true);
|
|
376
|
-
});
|
|
377
|
-
});
|
|
378
|
-
|
|
379
|
-
describe('enterprise audit – exportLogs', () => {
|
|
380
|
-
it('exports logs as JSON string', async () => {
|
|
381
|
-
const { createLog, exportLogs } = await import('../enterprise/audit');
|
|
382
|
-
await createLog({ action: 'export.test', status: 'success' });
|
|
383
|
-
|
|
384
|
-
const json = await exportLogs('json', {});
|
|
385
|
-
expect(typeof json).toBe('string');
|
|
386
|
-
const parsed = JSON.parse(json);
|
|
387
|
-
expect(Array.isArray(parsed.logs)).toBe(true);
|
|
388
|
-
expect(typeof parsed.exportedAt).toBe('string');
|
|
389
|
-
});
|
|
390
|
-
|
|
391
|
-
it('exports logs as CSV string', async () => {
|
|
392
|
-
const { createLog, exportLogs } = await import('../enterprise/audit');
|
|
393
|
-
await createLog({ action: 'csv.test', status: 'success' });
|
|
394
|
-
|
|
395
|
-
const csv = await exportLogs('csv', {});
|
|
396
|
-
expect(typeof csv).toBe('string');
|
|
397
|
-
// CSV should contain the header row
|
|
398
|
-
expect(csv).toContain('id,timestamp');
|
|
399
|
-
expect(csv).toContain('action');
|
|
400
|
-
});
|
|
401
|
-
});
|
|
@@ -1,236 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Export Command Tests — G19
|
|
3
|
-
*
|
|
4
|
-
* Tests exportCommand for JSON, Markdown, HTML formats and
|
|
5
|
-
* file-write / process.exit behaviors.
|
|
6
|
-
*
|
|
7
|
-
* The export command uses `await import('../sessions/manager')` dynamically,
|
|
8
|
-
* so we use vi.mock at module scope (hoisted) and vi.doMock for the empty-
|
|
9
|
-
* session scenario.
|
|
10
|
-
*
|
|
11
|
-
* writeFileSync is tested by spying on the module-level import inside the
|
|
12
|
-
* command via a top-level vi.mock on 'node:fs'.
|
|
13
|
-
*/
|
|
14
|
-
|
|
15
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
16
|
-
|
|
17
|
-
// ---------------------------------------------------------------------------
|
|
18
|
-
// Shared mock data
|
|
19
|
-
// ---------------------------------------------------------------------------
|
|
20
|
-
|
|
21
|
-
const MOCK_SESSION = {
|
|
22
|
-
id: 'session-abc-123',
|
|
23
|
-
name: 'Test Session',
|
|
24
|
-
model: 'claude-3-5-sonnet',
|
|
25
|
-
mode: 'build',
|
|
26
|
-
createdAt: '2026-01-01T00:00:00.000Z',
|
|
27
|
-
updatedAt: '2026-01-01T00:00:00.000Z',
|
|
28
|
-
status: 'active',
|
|
29
|
-
cwd: '/tmp',
|
|
30
|
-
tokenCount: 100,
|
|
31
|
-
costUSD: 0.01,
|
|
32
|
-
snapshotCount: 0,
|
|
33
|
-
};
|
|
34
|
-
|
|
35
|
-
const MOCK_MESSAGES = [
|
|
36
|
-
{ role: 'user', content: 'Hello agent' },
|
|
37
|
-
{ role: 'assistant', content: 'Hello! How can I help?' },
|
|
38
|
-
];
|
|
39
|
-
|
|
40
|
-
// ---------------------------------------------------------------------------
|
|
41
|
-
// Top-level mocks (hoisted by vitest)
|
|
42
|
-
// ---------------------------------------------------------------------------
|
|
43
|
-
|
|
44
|
-
// Mock sessions/manager with a factory function so the mock survives
|
|
45
|
-
// dynamic import() calls inside exportCommand.
|
|
46
|
-
vi.mock('../sessions/manager', () => {
|
|
47
|
-
const listFn = vi.fn(() => [MOCK_SESSION]);
|
|
48
|
-
const loadConversationFn = vi.fn(() => MOCK_MESSAGES);
|
|
49
|
-
return {
|
|
50
|
-
SessionManager: {
|
|
51
|
-
getInstance: vi.fn(() => ({
|
|
52
|
-
list: listFn,
|
|
53
|
-
loadConversation: loadConversationFn,
|
|
54
|
-
})),
|
|
55
|
-
resetInstance: vi.fn(),
|
|
56
|
-
},
|
|
57
|
-
};
|
|
58
|
-
});
|
|
59
|
-
|
|
60
|
-
// Track writeFileSync calls via a hoisted mock on node:fs
|
|
61
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
|
|
62
|
-
let writeFileSyncMock: ReturnType<typeof vi.fn> = vi.fn();
|
|
63
|
-
|
|
64
|
-
vi.mock('node:fs', async () => {
|
|
65
|
-
const actual = await vi.importActual<typeof import('node:fs')>('node:fs');
|
|
66
|
-
return {
|
|
67
|
-
...actual,
|
|
68
|
-
// Delegate to the mutable reference so we can swap it per-test
|
|
69
|
-
writeFileSync: (path: string, data: string, enc: string) => writeFileSyncMock(path, data, enc),
|
|
70
|
-
};
|
|
71
|
-
});
|
|
72
|
-
|
|
73
|
-
// ---------------------------------------------------------------------------
|
|
74
|
-
// Import the command (uses hoisted mocks above)
|
|
75
|
-
// ---------------------------------------------------------------------------
|
|
76
|
-
|
|
77
|
-
import { exportCommand } from '../commands/export';
|
|
78
|
-
|
|
79
|
-
// ---------------------------------------------------------------------------
|
|
80
|
-
// Helpers
|
|
81
|
-
// ---------------------------------------------------------------------------
|
|
82
|
-
|
|
83
|
-
function captureStdout(): { chunks: string[]; restore: () => void } {
|
|
84
|
-
const chunks: string[] = [];
|
|
85
|
-
const spy = vi.spyOn(process.stdout, 'write').mockImplementation((chunk) => {
|
|
86
|
-
chunks.push(typeof chunk === 'string' ? chunk : chunk.toString());
|
|
87
|
-
return true;
|
|
88
|
-
});
|
|
89
|
-
return { chunks, restore: () => spy.mockRestore() };
|
|
90
|
-
}
|
|
91
|
-
|
|
92
|
-
// ---------------------------------------------------------------------------
|
|
93
|
-
// JSON format
|
|
94
|
-
// ---------------------------------------------------------------------------
|
|
95
|
-
|
|
96
|
-
describe('exportCommand JSON format (G19)', () => {
|
|
97
|
-
it('writes valid JSON with session and messages fields', async () => {
|
|
98
|
-
const { chunks, restore } = captureStdout();
|
|
99
|
-
try {
|
|
100
|
-
await exportCommand({ format: 'json' });
|
|
101
|
-
} finally {
|
|
102
|
-
restore();
|
|
103
|
-
}
|
|
104
|
-
|
|
105
|
-
const output = chunks.join('');
|
|
106
|
-
expect(() => JSON.parse(output)).not.toThrow();
|
|
107
|
-
const parsed = JSON.parse(output) as { session: { id: string }; messages: unknown[] };
|
|
108
|
-
expect(parsed.session.id).toBe('session-abc-123');
|
|
109
|
-
expect(Array.isArray(parsed.messages)).toBe(true);
|
|
110
|
-
expect(parsed.messages).toHaveLength(2);
|
|
111
|
-
});
|
|
112
|
-
});
|
|
113
|
-
|
|
114
|
-
// ---------------------------------------------------------------------------
|
|
115
|
-
// Markdown format
|
|
116
|
-
// ---------------------------------------------------------------------------
|
|
117
|
-
|
|
118
|
-
describe('exportCommand Markdown format (G19)', () => {
|
|
119
|
-
it('returns markdown with session metadata table', async () => {
|
|
120
|
-
const { chunks, restore } = captureStdout();
|
|
121
|
-
try {
|
|
122
|
-
await exportCommand({ format: 'md' });
|
|
123
|
-
} finally {
|
|
124
|
-
restore();
|
|
125
|
-
}
|
|
126
|
-
|
|
127
|
-
const output = chunks.join('');
|
|
128
|
-
expect(output).toContain('# Nimbus Session Export');
|
|
129
|
-
expect(output).toContain('session-abc-123');
|
|
130
|
-
expect(output).toContain('## Conversation');
|
|
131
|
-
});
|
|
132
|
-
|
|
133
|
-
it('default format is markdown when not specified', async () => {
|
|
134
|
-
const { chunks, restore } = captureStdout();
|
|
135
|
-
try {
|
|
136
|
-
await exportCommand({});
|
|
137
|
-
} finally {
|
|
138
|
-
restore();
|
|
139
|
-
}
|
|
140
|
-
|
|
141
|
-
const output = chunks.join('');
|
|
142
|
-
expect(output).toContain('# Nimbus Session Export');
|
|
143
|
-
});
|
|
144
|
-
});
|
|
145
|
-
|
|
146
|
-
// ---------------------------------------------------------------------------
|
|
147
|
-
// HTML format
|
|
148
|
-
// ---------------------------------------------------------------------------
|
|
149
|
-
|
|
150
|
-
describe('exportCommand HTML format (G19)', () => {
|
|
151
|
-
it('returns HTML output with DOCTYPE and html tag', async () => {
|
|
152
|
-
const { chunks, restore } = captureStdout();
|
|
153
|
-
try {
|
|
154
|
-
await exportCommand({ format: 'html' });
|
|
155
|
-
} finally {
|
|
156
|
-
restore();
|
|
157
|
-
}
|
|
158
|
-
|
|
159
|
-
const output = chunks.join('');
|
|
160
|
-
expect(output).toContain('<!DOCTYPE html>');
|
|
161
|
-
expect(output).toContain('<html>');
|
|
162
|
-
expect(output).toContain('Nimbus Session');
|
|
163
|
-
});
|
|
164
|
-
|
|
165
|
-
it('HTML output contains session name', async () => {
|
|
166
|
-
const { chunks, restore } = captureStdout();
|
|
167
|
-
try {
|
|
168
|
-
await exportCommand({ format: 'html' });
|
|
169
|
-
} finally {
|
|
170
|
-
restore();
|
|
171
|
-
}
|
|
172
|
-
|
|
173
|
-
const output = chunks.join('');
|
|
174
|
-
expect(output).toContain('Test Session');
|
|
175
|
-
});
|
|
176
|
-
});
|
|
177
|
-
|
|
178
|
-
// ---------------------------------------------------------------------------
|
|
179
|
-
// File output
|
|
180
|
-
// ---------------------------------------------------------------------------
|
|
181
|
-
|
|
182
|
-
describe('exportCommand file output (G19)', () => {
|
|
183
|
-
beforeEach(() => {
|
|
184
|
-
// Reset to a fresh spy each test
|
|
185
|
-
writeFileSyncMock = vi.fn();
|
|
186
|
-
});
|
|
187
|
-
|
|
188
|
-
it('calls writeFileSync when --output path is specified', async () => {
|
|
189
|
-
const logSpy = vi.spyOn(console, 'log').mockImplementation(() => {});
|
|
190
|
-
try {
|
|
191
|
-
await exportCommand({ format: 'md', output: '/tmp/test-export.md' });
|
|
192
|
-
} finally {
|
|
193
|
-
logSpy.mockRestore();
|
|
194
|
-
}
|
|
195
|
-
|
|
196
|
-
expect(writeFileSyncMock).toHaveBeenCalledOnce();
|
|
197
|
-
const [filePath, content] = writeFileSyncMock.mock.calls[0] as [string, string, string];
|
|
198
|
-
expect(filePath).toBe('/tmp/test-export.md');
|
|
199
|
-
expect(content).toContain('Nimbus Session Export');
|
|
200
|
-
});
|
|
201
|
-
});
|
|
202
|
-
|
|
203
|
-
// ---------------------------------------------------------------------------
|
|
204
|
-
// No session — process.exit(1)
|
|
205
|
-
// ---------------------------------------------------------------------------
|
|
206
|
-
|
|
207
|
-
describe('exportCommand no session (G19)', () => {
|
|
208
|
-
it('calls process.exit(1) when no sessions exist', async () => {
|
|
209
|
-
vi.resetModules();
|
|
210
|
-
|
|
211
|
-
vi.doMock('../sessions/manager', () => ({
|
|
212
|
-
SessionManager: {
|
|
213
|
-
getInstance: vi.fn(() => ({
|
|
214
|
-
list: vi.fn(() => []),
|
|
215
|
-
loadConversation: vi.fn(() => []),
|
|
216
|
-
})),
|
|
217
|
-
resetInstance: vi.fn(),
|
|
218
|
-
},
|
|
219
|
-
}));
|
|
220
|
-
|
|
221
|
-
const errSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
|
|
222
|
-
const exitSpy = vi.spyOn(process, 'exit').mockImplementation((() => {
|
|
223
|
-
throw new Error('process.exit called');
|
|
224
|
-
}) as never);
|
|
225
|
-
|
|
226
|
-
try {
|
|
227
|
-
const { exportCommand: exportCmd } = await import('../commands/export');
|
|
228
|
-
await expect(exportCmd({})).rejects.toThrow('process.exit');
|
|
229
|
-
expect(exitSpy).toHaveBeenCalledWith(1);
|
|
230
|
-
} finally {
|
|
231
|
-
errSpy.mockRestore();
|
|
232
|
-
exitSpy.mockRestore();
|
|
233
|
-
vi.resetModules();
|
|
234
|
-
}
|
|
235
|
-
});
|
|
236
|
-
});
|