@@ -14,6 +14,7 @@ import { Title } from '../../webComponents/Title/index.js';
1414import { FlexBox } from '../FlexBox/index.js' ;
1515import { MessageViewButton } from '../MessageViewButton/index.js' ;
1616import { MessageItem } from './MessageItem.js' ;
17+ import type { MessageViewDomRef } from './index.js' ;
1718import { MessageView } from './index.js' ;
1819
1920// TODO: check docs for outdated info
@@ -88,7 +89,7 @@ export const MessageViewInDialog: Story = {
8889 name : 'MessageView in Dialog' ,
8990 render ( args ) {
9091 const [ open , setOpen ] = useState ( false ) ;
91- const messageViewRef = useRef ( null ) ;
92+ const messageViewRef = useRef < MessageViewDomRef > ( null ) ;
9293 const [ isOnDetailsPage , setIsOnDetailsPage ] = useState ( false ) ;
9394 return (
9495 < >
@@ -106,6 +107,7 @@ export const MessageViewInDialog: Story = {
106107 open = { open }
107108 onClose = { ( ) => {
108109 setOpen ( false ) ;
110+ messageViewRef . current . navigateBack ( ) ;
109111 } }
110112 header = {
111113 < Bar
@@ -165,7 +167,7 @@ export const WithMessageViewButton: Story = {
165167 name : 'with MessageViewButton & Popover' ,
166168 render ( ) {
167169 const ref = useRef ( null ) ;
168- const messageViewRef = useRef ( null ) ;
170+ const messageViewRef = useRef < MessageViewDomRef > ( null ) ;
169171 const [ isOnDetailsPage , setIsOnDetailsPage ] = useState ( false ) ;
170172 const [ open , setOpen ] = useState ( false ) ;
171173 const numberOfItems = {
@@ -190,6 +192,7 @@ export const WithMessageViewButton: Story = {
190192 headerText = "Messages"
191193 className = "contentPartNoPadding headerPartNoPadding"
192194 onClose = { ( ) => {
195+ messageViewRef . current . navigateBack ( ) ;
193196 setOpen ( false ) ;
194197 } }
195198 header = {
0 commit comments