Refactor notification-section tests (#6972)

zooniverse · Dec 4, 2023 · 2ff2f55 · 2ff2f55
1 parent 304aa9a
commit 2ff2f55
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 7 deletions.
diff --git a/app/pages/notifications/notification-section.jsx b/app/pages/notifications/notification-section.jsx
@@ -19,6 +19,7 @@ export default class NotificationSection extends Component {
       lastMeta: { },
       loading: false,
       notificationData: [],
+      notifications: [],
       notificationsMap: { },
       page: 1,
       project: null
@@ -318,5 +319,6 @@ NotificationSection.contextTypes = {
 
 NotificationSection.defaultProps = {
   expanded: false,
-  toggleSection: () => {}
+  toggleSection: () => {},
+  user: null
 };
diff --git a/app/pages/notifications/notification-section.spec.js b/app/pages/notifications/notification-section.spec.js
@@ -138,12 +138,26 @@ describe('Notification Section', function() {
     const notificationsCounter = {
       update: sinon.stub()
     };
-    wrapper = shallow(
-      <NotificationSection expanded={true} />,
-      { context: { notificationsCounter }, disableLifeCycleMethods: true }
-    );
-    wrapper.setState({ notifications: newNotifications });
-    wrapper.instance().markAsRead(newNotifications[0]);
+
+    before(function () {
+      wrapper = shallow(
+        <NotificationSection
+          expanded={true}
+          section="zooniverse"
+          user={{ id: '1' }}
+        />,
+        {
+          context: { notificationsCounter }, 
+          disableLifeCycleMethods: true
+        }
+      );
+      wrapper.setState({
+        loading: false,
+        notificationData: [],
+        notifications: newNotifications
+      });
+      wrapper.instance().markAsRead(newNotifications[0]);
+    });
 
     it('should update read notification as read (delivered)', function () {
       assert.equal(newNotifications[0].update.calledWith({ delivered: true }), true);