88 */
99
1010import { AbstractChatModel , IChatContext , IChatModel } from '../model' ;
11- import { IMessageContent , INewMessage } from '../types' ;
11+ import { IMessage , IMessageContent , INewMessage } from '../types' ;
1212import { MockChatModel , MockChatContext } from './mocks' ;
1313
1414describe ( 'test chat model' , ( ) => {
@@ -43,7 +43,7 @@ describe('test chat model', () => {
4343 }
4444
4545 let model : IChatModel ;
46- let messages : IMessageContent [ ] ;
46+ let messages : IMessage [ ] ;
4747 const msg = {
4848 type : 'msg' ,
4949 id : 'message1' ,
@@ -64,7 +64,7 @@ describe('test chat model', () => {
6464 } ) ;
6565 model . messageAdded ( msg ) ;
6666 expect ( messages ) . toHaveLength ( 1 ) ;
67- expect ( messages [ 0 ] ) . toBe ( msg ) ;
67+ expect ( messages [ 0 ] . content ) . toBe ( msg ) ;
6868 } ) ;
6969
7070 it ( 'should format message' , ( ) => {
@@ -75,8 +75,8 @@ describe('test chat model', () => {
7575 } ) ;
7676 model . messageAdded ( { ...msg } ) ;
7777 expect ( messages ) . toHaveLength ( 1 ) ;
78- expect ( messages [ 0 ] ) . not . toBe ( msg ) ;
79- expect ( ( messages [ 0 ] as IMessageContent ) . body ) . toBe ( 'formatted msg' ) ;
78+ expect ( messages [ 0 ] . content ) . not . toBe ( msg ) ;
79+ expect ( messages [ 0 ] . body ) . toBe ( 'formatted msg' ) ;
8080 } ) ;
8181 } ) ;
8282
0 commit comments