@@ -22,20 +22,6 @@ function createMockServer() {
2222
2323describe ( 'Prompts' , ( ) => {
2424 describe ( 'simple-prompt' , ( ) => {
25- it ( 'should register with correct name and config' , ( ) => {
26- const { mockServer } = createMockServer ( ) ;
27- registerSimplePrompt ( mockServer ) ;
28-
29- expect ( mockServer . registerPrompt ) . toHaveBeenCalledWith (
30- 'simple-prompt' ,
31- expect . objectContaining ( {
32- title : 'Simple Prompt' ,
33- description : 'A prompt with no arguments' ,
34- } ) ,
35- expect . any ( Function )
36- ) ;
37- } ) ;
38-
3925 it ( 'should return fixed message with no arguments' , ( ) => {
4026 const { mockServer, handlers } = createMockServer ( ) ;
4127 registerSimplePrompt ( mockServer ) ;
@@ -55,34 +41,9 @@ describe('Prompts', () => {
5541 ] ,
5642 } ) ;
5743 } ) ;
58-
59- it ( 'should return message with user role' , ( ) => {
60- const { mockServer, handlers } = createMockServer ( ) ;
61- registerSimplePrompt ( mockServer ) ;
62-
63- const handler = handlers . get ( 'simple-prompt' ) ! ;
64- const result = handler ( ) ;
65-
66- expect ( result . messages ) . toHaveLength ( 1 ) ;
67- expect ( result . messages [ 0 ] . role ) . toBe ( 'user' ) ;
68- } ) ;
6944 } ) ;
7045
7146 describe ( 'args-prompt' , ( ) => {
72- it ( 'should register with correct name and config' , ( ) => {
73- const { mockServer } = createMockServer ( ) ;
74- registerArgumentsPrompt ( mockServer ) ;
75-
76- expect ( mockServer . registerPrompt ) . toHaveBeenCalledWith (
77- 'args-prompt' ,
78- expect . objectContaining ( {
79- title : 'Arguments Prompt' ,
80- description : 'A prompt with two arguments, one required and one optional' ,
81- } ) ,
82- expect . any ( Function )
83- ) ;
84- } ) ;
85-
8647 it ( 'should include city in message' , ( ) => {
8748 const { mockServer, handlers } = createMockServer ( ) ;
8849 registerArgumentsPrompt ( mockServer ) ;
@@ -114,36 +75,12 @@ describe('Prompts', () => {
11475
11576 expect ( result . messages [ 0 ] . content . text ) . toBe ( "What's weather in New York?" ) ;
11677 expect ( result . messages [ 0 ] . content . text ) . not . toContain ( ',' ) ;
117- } ) ;
118-
119- it ( 'should return message with user role' , ( ) => {
120- const { mockServer, handlers } = createMockServer ( ) ;
121- registerArgumentsPrompt ( mockServer ) ;
122-
123- const handler = handlers . get ( 'args-prompt' ) ! ;
124- const result = handler ( { city : 'Boston' } ) ;
125-
126- expect ( result . messages ) . toHaveLength ( 1 ) ;
12778 expect ( result . messages [ 0 ] . role ) . toBe ( 'user' ) ;
12879 expect ( result . messages [ 0 ] . content . type ) . toBe ( 'text' ) ;
12980 } ) ;
13081 } ) ;
13182
13283 describe ( 'completable-prompt' , ( ) => {
133- it ( 'should register with correct name and config' , ( ) => {
134- const { mockServer } = createMockServer ( ) ;
135- registerPromptWithCompletions ( mockServer ) ;
136-
137- expect ( mockServer . registerPrompt ) . toHaveBeenCalledWith (
138- 'completable-prompt' ,
139- expect . objectContaining ( {
140- title : 'Team Management' ,
141- description : 'First argument choice narrows values for second argument.' ,
142- } ) ,
143- expect . any ( Function )
144- ) ;
145- } ) ;
146-
14784 it ( 'should generate promotion message with department and name' , ( ) => {
14885 const { mockServer, handlers } = createMockServer ( ) ;
14986 registerPromptWithCompletions ( mockServer ) ;
@@ -165,39 +102,15 @@ describe('Prompts', () => {
165102 const salesResult = handler ( { department : 'Sales' , name : 'David' } ) ;
166103 expect ( salesResult . messages [ 0 ] . content . text ) . toContain ( 'Sales' ) ;
167104 expect ( salesResult . messages [ 0 ] . content . text ) . toContain ( 'David' ) ;
105+ expect ( salesResult . messages [ 0 ] . role ) . toBe ( 'user' ) ;
168106
169107 const marketingResult = handler ( { department : 'Marketing' , name : 'Grace' } ) ;
170108 expect ( marketingResult . messages [ 0 ] . content . text ) . toContain ( 'Marketing' ) ;
171109 expect ( marketingResult . messages [ 0 ] . content . text ) . toContain ( 'Grace' ) ;
172110 } ) ;
173-
174- it ( 'should return message with user role' , ( ) => {
175- const { mockServer, handlers } = createMockServer ( ) ;
176- registerPromptWithCompletions ( mockServer ) ;
177-
178- const handler = handlers . get ( 'completable-prompt' ) ! ;
179- const result = handler ( { department : 'Support' , name : 'John' } ) ;
180-
181- expect ( result . messages ) . toHaveLength ( 1 ) ;
182- expect ( result . messages [ 0 ] . role ) . toBe ( 'user' ) ;
183- } ) ;
184111 } ) ;
185112
186113 describe ( 'resource-prompt' , ( ) => {
187- it ( 'should register with correct name and config' , ( ) => {
188- const { mockServer } = createMockServer ( ) ;
189- registerEmbeddedResourcePrompt ( mockServer ) ;
190-
191- expect ( mockServer . registerPrompt ) . toHaveBeenCalledWith (
192- 'resource-prompt' ,
193- expect . objectContaining ( {
194- title : 'Resource Prompt' ,
195- description : 'A prompt that includes an embedded resource reference' ,
196- } ) ,
197- expect . any ( Function )
198- ) ;
199- } ) ;
200-
201114 it ( 'should return text resource reference' , ( ) => {
202115 const { mockServer, handlers } = createMockServer ( ) ;
203116 registerEmbeddedResourcePrompt ( mockServer ) ;
0 commit comments