Skip to content

Commit 0ce9a91

Browse files
committed
refactor maps to pull names from the test
1 parent 204160b commit 0ce9a91

1 file changed

Lines changed: 49 additions & 40 deletions

File tree

src/scenarios/index.ts

Lines changed: 49 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -47,59 +47,68 @@ import {
4747
PromptsGetWithImageScenario
4848
} from './server/prompts.js';
4949

50-
export const scenarios = new Map<string, Scenario>([
51-
['initialize', new InitializeScenario()],
52-
['tools-call', new ToolsCallScenario()],
53-
['auth/basic-dcr', new AuthBasicDCRScenario()],
54-
['auth/basic-metadata-var1', new AuthBasicMetadataVar1Scenario()],
55-
[
56-
'elicitation-sep1034-client-defaults',
57-
new ElicitationClientDefaultsScenario()
58-
]
59-
]);
60-
61-
export const clientScenarios = new Map<string, ClientScenario>([
50+
// Client scenarios
51+
const clientScenariosList: ClientScenario[] = [
6252
// Lifecycle scenarios
63-
['server-initialize', new ServerInitializeScenario()],
53+
new ServerInitializeScenario(),
6454

6555
// Utilities scenarios
66-
['logging-set-level', new LoggingSetLevelScenario()],
67-
['completion-complete', new CompletionCompleteScenario()],
56+
new LoggingSetLevelScenario(),
57+
new CompletionCompleteScenario(),
6858

6959
// Tools scenarios
70-
['tools-list', new ToolsListScenario()],
71-
['tools-call-simple-text', new ToolsCallSimpleTextScenario()],
72-
['tools-call-image', new ToolsCallImageScenario()],
73-
['tools-call-audio', new ToolsCallAudioScenario()],
74-
['tools-call-embedded-resource', new ToolsCallEmbeddedResourceScenario()],
75-
['tools-call-mixed-content', new ToolsCallMultipleContentTypesScenario()],
76-
['tools-call-with-logging', new ToolsCallWithLoggingScenario()],
77-
['tools-call-error', new ToolsCallErrorScenario()],
78-
['tools-call-with-progress', new ToolsCallWithProgressScenario()],
79-
['tools-call-sampling', new ToolsCallSamplingScenario()],
80-
['tools-call-elicitation', new ToolsCallElicitationScenario()],
60+
new ToolsListScenario(),
61+
new ToolsCallSimpleTextScenario(),
62+
new ToolsCallImageScenario(),
63+
new ToolsCallAudioScenario(),
64+
new ToolsCallEmbeddedResourceScenario(),
65+
new ToolsCallMultipleContentTypesScenario(),
66+
new ToolsCallWithLoggingScenario(),
67+
new ToolsCallErrorScenario(),
68+
new ToolsCallWithProgressScenario(),
69+
new ToolsCallSamplingScenario(),
70+
new ToolsCallElicitationScenario(),
8171

8272
// Elicitation scenarios (SEP-1034)
83-
['elicitation-sep1034-defaults', new ElicitationDefaultsScenario()],
73+
new ElicitationDefaultsScenario(),
8474

8575
// Elicitation scenarios (SEP-1330)
86-
['elicitation-sep1330-enums', new ElicitationEnumsScenario()],
76+
new ElicitationEnumsScenario(),
8777

8878
// Resources scenarios
89-
['resources-list', new ResourcesListScenario()],
90-
['resources-read-text', new ResourcesReadTextScenario()],
91-
['resources-read-binary', new ResourcesReadBinaryScenario()],
92-
['resources-templates-read', new ResourcesTemplateReadScenario()],
93-
['resources-subscribe', new ResourcesSubscribeScenario()],
94-
['resources-unsubscribe', new ResourcesUnsubscribeScenario()],
79+
new ResourcesListScenario(),
80+
new ResourcesReadTextScenario(),
81+
new ResourcesReadBinaryScenario(),
82+
new ResourcesTemplateReadScenario(),
83+
new ResourcesSubscribeScenario(),
84+
new ResourcesUnsubscribeScenario(),
9585

9686
// Prompts scenarios
97-
['prompts-list', new PromptsListScenario()],
98-
['prompts-get-simple', new PromptsGetSimpleScenario()],
99-
['prompts-get-with-args', new PromptsGetWithArgsScenario()],
100-
['prompts-get-embedded-resource', new PromptsGetEmbeddedResourceScenario()],
101-
['prompts-get-with-image', new PromptsGetWithImageScenario()]
102-
]);
87+
new PromptsListScenario(),
88+
new PromptsGetSimpleScenario(),
89+
new PromptsGetWithArgsScenario(),
90+
new PromptsGetEmbeddedResourceScenario(),
91+
new PromptsGetWithImageScenario()
92+
];
93+
94+
// Client scenarios map - built from list
95+
export const clientScenarios = new Map<string, ClientScenario>(
96+
clientScenariosList.map((scenario) => [scenario.name, scenario])
97+
);
98+
99+
// Scenario scenarios
100+
const scenariosList: Scenario[] = [
101+
new InitializeScenario(),
102+
new ToolsCallScenario(),
103+
new AuthBasicDCRScenario(),
104+
new AuthBasicMetadataVar1Scenario(),
105+
new ElicitationClientDefaultsScenario()
106+
];
107+
108+
// Scenarios map - built from list
109+
export const scenarios = new Map<string, Scenario>(
110+
scenariosList.map((scenario) => [scenario.name, scenario])
111+
);
103112

104113
export function registerScenario(name: string, scenario: Scenario): void {
105114
scenarios.set(name, scenario);

0 commit comments

Comments
 (0)