From 08727aafe580ff0f2b4653f6914f42c4c03e2059 Mon Sep 17 00:00:00 2001
From: Charles Bochet <charles@twenty.com>
Date: Sun, 10 Sep 2023 20:16:04 -0700
Subject: [PATCH] Fix tests

---
 .../companies/__stories__/Board.stories.tsx       | 15 +++++----------
 .../companies/board/components/CompanyBoard.tsx   | 15 ++++++++-------
 front/src/pages/opportunities/Opportunities.tsx   |  1 -
 3 files changed, 13 insertions(+), 18 deletions(-)

diff --git a/front/src/modules/companies/__stories__/Board.stories.tsx b/front/src/modules/companies/__stories__/Board.stories.tsx
index 1a51ab974..5c9b8b213 100644
--- a/front/src/modules/companies/__stories__/Board.stories.tsx
+++ b/front/src/modules/companies/__stories__/Board.stories.tsx
@@ -1,18 +1,17 @@
 import { MemoryRouter } from 'react-router-dom';
 import { Meta, StoryObj } from '@storybook/react';
 
-import { EntityBoard } from '@/ui/board/components/EntityBoard';
 import { RecoilScope } from '@/ui/utilities/recoil-scope/components/RecoilScope';
-import { opportunitiesBoardOptions } from '~/pages/opportunities/opportunitiesBoardOptions';
 import { ComponentDecorator } from '~/testing/decorators/ComponentDecorator';
 import { graphqlMocks } from '~/testing/graphqlMocks';
 
+import { CompanyBoard } from '../board/components/CompanyBoard';
 import { HooksCompanyBoard } from '../components/HooksCompanyBoard';
 import { CompanyBoardRecoilScopeContext } from '../states/recoil-scope-contexts/CompanyBoardRecoilScopeContext';
 
-const meta: Meta<typeof EntityBoard> = {
+const meta: Meta<typeof CompanyBoard> = {
   title: 'Modules/Companies/Board',
-  component: EntityBoard,
+  component: CompanyBoard,
   decorators: [
     (Story) => (
       <RecoilScope SpecificContext={CompanyBoardRecoilScopeContext}>
@@ -30,10 +29,6 @@ const meta: Meta<typeof EntityBoard> = {
 };
 
 export default meta;
-type Story = StoryObj<typeof EntityBoard>;
+type Story = StoryObj<typeof CompanyBoard>;
 
-export const OneColumnBoard: Story = {
-  render: (args) => (
-    <EntityBoard {...args} boardOptions={opportunitiesBoardOptions} />
-  ),
-};
+export const OneColumnBoard: Story = {};
diff --git a/front/src/modules/companies/board/components/CompanyBoard.tsx b/front/src/modules/companies/board/components/CompanyBoard.tsx
index f195140e2..6b97ebeb1 100644
--- a/front/src/modules/companies/board/components/CompanyBoard.tsx
+++ b/front/src/modules/companies/board/components/CompanyBoard.tsx
@@ -5,19 +5,20 @@ import {
 import { EntityBoardActionBar } from '@/ui/board/components/EntityBoardActionBar';
 import { EntityBoardContextMenu } from '@/ui/board/components/EntityBoardContextMenu';
 import { useBoardViews } from '@/views/hooks/useBoardViews';
+import { opportunitiesBoardOptions } from '~/pages/opportunities/opportunitiesBoardOptions';
 
 import { HooksCompanyBoard } from '../../components/HooksCompanyBoard';
 import { CompanyBoardRecoilScopeContext } from '../../states/recoil-scope-contexts/CompanyBoardRecoilScopeContext';
 
-type OwnProps = Pick<
+type CompanyBoardProps = Pick<
   EntityBoardProps,
-  'boardOptions' | 'onColumnAdd' | 'onColumnDelete' | 'onEditColumnTitle'
+  'onColumnAdd' | 'onColumnDelete' | 'onEditColumnTitle'
 >;
 
-export const CompanyBoard = ({ boardOptions, ...props }: OwnProps) => {
+export const CompanyBoard = ({ ...props }: CompanyBoardProps) => {
   const { handleViewsChange, handleViewSubmit } = useBoardViews({
-    availableFilters: boardOptions.filters,
-    availableSorts: boardOptions.sorts,
+    availableFilters: opportunitiesBoardOptions.filters,
+    availableSorts: opportunitiesBoardOptions.sorts,
     objectId: 'company',
     scopeContext: CompanyBoardRecoilScopeContext,
   });
@@ -26,12 +27,12 @@ export const CompanyBoard = ({ boardOptions, ...props }: OwnProps) => {
     <>
       <HooksCompanyBoard />
       <EntityBoard
-        boardOptions={boardOptions}
+        boardOptions={opportunitiesBoardOptions}
         defaultViewName="All opportunities"
         onViewsChange={handleViewsChange}
         onViewSubmit={handleViewSubmit}
         scopeContext={CompanyBoardRecoilScopeContext}
-        {...props}
+        onEditColumnTitle={props.onEditColumnTitle}
       />
       <EntityBoardActionBar />
       <EntityBoardContextMenu />
diff --git a/front/src/pages/opportunities/Opportunities.tsx b/front/src/pages/opportunities/Opportunities.tsx
index 2f364e6ba..d889e39a1 100644
--- a/front/src/pages/opportunities/Opportunities.tsx
+++ b/front/src/pages/opportunities/Opportunities.tsx
@@ -62,7 +62,6 @@ export function Opportunities() {
               SpecificContext={CompanyBoardRecoilScopeContext}
             >
               <CompanyBoard
-                boardOptions={opportunitiesBoardOptions}
                 onColumnAdd={handlePipelineStageAdd}
                 onColumnDelete={handlePipelineStageDelete}
                 onEditColumnTitle={handleEditColumnTitle}