@@ -35,8 +35,11 @@ const expectActionButtonsExists = () => {
3535} ;
3636const afterClickDropDownButton = async ( ) => {
3737 const dropDownButton = screen . getAllByRole ( 'button' ) ;
38- expect ( dropDownButton . length ) . toEqual ( 1 ) ;
39- await userEvent . click ( dropDownButton [ 0 ] ) ;
38+ await userEvent . click ( dropDownButton [ 1 ] ) ;
39+ } ;
40+ const afterClickRestartButton = async ( ) => {
41+ const dropDownButton = screen . getByText ( 'Restart' ) ;
42+ await userEvent . click ( dropDownButton ) ;
4043} ;
4144describe ( 'Actions' , ( ) => {
4245 afterEach ( ( ) => {
@@ -66,8 +69,8 @@ describe('Actions', () => {
6669 data : set ( { ...connector } , 'status.state' , ConnectorState . PAUSED ) ,
6770 } ) ) ;
6871 renderComponent ( ) ;
69- await afterClickDropDownButton ( ) ;
70- expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 5 ) ;
72+ await afterClickRestartButton ( ) ;
73+ expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 4 ) ;
7174 expect ( screen . getByText ( 'Resume' ) ) . toBeInTheDocument ( ) ;
7275 expect ( screen . queryByText ( 'Pause' ) ) . not . toBeInTheDocument ( ) ;
7376 expectActionButtonsExists ( ) ;
@@ -78,8 +81,8 @@ describe('Actions', () => {
7881 data : set ( { ...connector } , 'status.state' , ConnectorState . FAILED ) ,
7982 } ) ) ;
8083 renderComponent ( ) ;
81- await afterClickDropDownButton ( ) ;
82- expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 4 ) ;
84+ await afterClickRestartButton ( ) ;
85+ expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 3 ) ;
8386 expect ( screen . queryByText ( 'Resume' ) ) . not . toBeInTheDocument ( ) ;
8487 expect ( screen . queryByText ( 'Pause' ) ) . not . toBeInTheDocument ( ) ;
8588 expectActionButtonsExists ( ) ;
@@ -90,8 +93,8 @@ describe('Actions', () => {
9093 data : set ( { ...connector } , 'status.state' , ConnectorState . UNASSIGNED ) ,
9194 } ) ) ;
9295 renderComponent ( ) ;
93- await afterClickDropDownButton ( ) ;
94- expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 4 ) ;
96+ await afterClickRestartButton ( ) ;
97+ expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 3 ) ;
9598 expect ( screen . queryByText ( 'Resume' ) ) . not . toBeInTheDocument ( ) ;
9699 expect ( screen . queryByText ( 'Pause' ) ) . not . toBeInTheDocument ( ) ;
97100 expectActionButtonsExists ( ) ;
@@ -102,8 +105,8 @@ describe('Actions', () => {
102105 data : set ( { ...connector } , 'status.state' , ConnectorState . RUNNING ) ,
103106 } ) ) ;
104107 renderComponent ( ) ;
105- await afterClickDropDownButton ( ) ;
106- expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 5 ) ;
108+ await afterClickRestartButton ( ) ;
109+ expect ( screen . getAllByRole ( 'menuitem' ) . length ) . toEqual ( 4 ) ;
107110 expect ( screen . queryByText ( 'Resume' ) ) . not . toBeInTheDocument ( ) ;
108111 expect ( screen . getByText ( 'Pause' ) ) . toBeInTheDocument ( ) ;
109112 expectActionButtonsExists ( ) ;
@@ -131,7 +134,7 @@ describe('Actions', () => {
131134 mutateAsync : restartConnector ,
132135 } ) ) ;
133136 renderComponent ( ) ;
134- await afterClickDropDownButton ( ) ;
137+ await afterClickRestartButton ( ) ;
135138 await userEvent . click (
136139 screen . getByRole ( 'menuitem' , { name : 'Restart Connector' } )
137140 ) ;
@@ -144,7 +147,7 @@ describe('Actions', () => {
144147 mutateAsync : restartAllTasks ,
145148 } ) ) ;
146149 renderComponent ( ) ;
147- await afterClickDropDownButton ( ) ;
150+ await afterClickRestartButton ( ) ;
148151 await userEvent . click (
149152 screen . getByRole ( 'menuitem' , { name : 'Restart All Tasks' } )
150153 ) ;
@@ -159,7 +162,7 @@ describe('Actions', () => {
159162 mutateAsync : restartFailedTasks ,
160163 } ) ) ;
161164 renderComponent ( ) ;
162- await afterClickDropDownButton ( ) ;
165+ await afterClickRestartButton ( ) ;
163166 await userEvent . click (
164167 screen . getByRole ( 'menuitem' , { name : 'Restart Failed Tasks' } )
165168 ) ;
@@ -174,7 +177,7 @@ describe('Actions', () => {
174177 mutateAsync : pauseConnector ,
175178 } ) ) ;
176179 renderComponent ( ) ;
177- await afterClickDropDownButton ( ) ;
180+ await afterClickRestartButton ( ) ;
178181 await userEvent . click ( screen . getByRole ( 'menuitem' , { name : 'Pause' } ) ) ;
179182 expect ( pauseConnector ) . toHaveBeenCalledWith ( ConnectorAction . PAUSE ) ;
180183 } ) ;
@@ -188,7 +191,7 @@ describe('Actions', () => {
188191 mutateAsync : resumeConnector ,
189192 } ) ) ;
190193 renderComponent ( ) ;
191- await afterClickDropDownButton ( ) ;
194+ await afterClickRestartButton ( ) ;
192195 await userEvent . click ( screen . getByRole ( 'menuitem' , { name : 'Resume' } ) ) ;
193196 expect ( resumeConnector ) . toHaveBeenCalledWith ( ConnectorAction . RESUME ) ;
194197 } ) ;
0 commit comments