From d5e94b5a5e0a02295e31f4cbac41be0e00991804 Mon Sep 17 00:00:00 2001
From: rjmacarthy <richardmacarthy@hotmail.com>
Date: Tue, 31 Jan 2023 10:04:13 +0000
Subject: [PATCH] Fix Messages stories wrap with SessionProvider

---
 website/src/components/Messages/MessageTable.stories.tsx  | 7 ++++++-
 .../src/components/Messages/MessageTableEntry.stories.tsx | 7 ++++++-
 .../components/Messages/MessageWithChildren.stories.tsx   | 8 +++++++-
 3 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/website/src/components/Messages/MessageTable.stories.tsx b/website/src/components/Messages/MessageTable.stories.tsx
index bc03aed1..772d1b52 100644
--- a/website/src/components/Messages/MessageTable.stories.tsx
+++ b/website/src/components/Messages/MessageTable.stories.tsx
@@ -1,3 +1,4 @@
+import { SessionProvider } from "next-auth/react";
 import React from "react";
 import { Message } from "src/types/Conversation";
 
@@ -18,7 +19,11 @@ const Template = ({
   enableLink: boolean;
   highlightLastMessage: boolean;
 }) => {
-  return <MessageTable messages={messages} enableLink={enableLink} highlightLastMessage={highlightLastMessage} />;
+  return (
+    <SessionProvider>
+      <MessageTable messages={messages} enableLink={enableLink} highlightLastMessage={highlightLastMessage} />;
+    </SessionProvider>
+  );
 };
 
 export const Default = Template.bind({});
diff --git a/website/src/components/Messages/MessageTableEntry.stories.tsx b/website/src/components/Messages/MessageTableEntry.stories.tsx
index 3550d00f..7288fb6f 100644
--- a/website/src/components/Messages/MessageTableEntry.stories.tsx
+++ b/website/src/components/Messages/MessageTableEntry.stories.tsx
@@ -1,3 +1,4 @@
+import { SessionProvider } from "next-auth/react";
 import React from "react";
 import { Message } from "src/types/Conversation";
 
@@ -10,7 +11,11 @@ export default {
 };
 
 const Template = ({ enabled, highlight, ...message }) => {
-  return <MessageTableEntry message={message as Message} enabled={enabled} highlight={highlight} />;
+  return (
+    <SessionProvider>
+      <MessageTableEntry message={message as Message} enabled={enabled} highlight={highlight} />;
+    </SessionProvider>
+  );
 };
 
 export const Default = Template.bind({});
diff --git a/website/src/components/Messages/MessageWithChildren.stories.tsx b/website/src/components/Messages/MessageWithChildren.stories.tsx
index 3ad9583f..55472c20 100644
--- a/website/src/components/Messages/MessageWithChildren.stories.tsx
+++ b/website/src/components/Messages/MessageWithChildren.stories.tsx
@@ -1,4 +1,6 @@
 import { rest } from "msw";
+import { SessionProvider } from "next-auth/react";
+
 import { MessageWithChildren } from "./MessageWithChildren";
 
 // eslint-disable-next-line import/no-anonymous-default-export
@@ -28,7 +30,11 @@ export default {
   },
 };
 
-const Template = (args) => <MessageWithChildren {...args} />;
+const Template = (args) => (
+  <SessionProvider>
+    <MessageWithChildren {...args} />;
+  </SessionProvider>
+);
 
 export const NoChildren = Template.bind({});
 NoChildren.args = {