webpack/scenes/Subscriptions/__tests__/SubscriptionsPage.test.js in katello-3.16.0.rc2 vs webpack/scenes/Subscriptions/__tests__/SubscriptionsPage.test.js in katello-3.16.0.rc2.1

@@ -14,15 +14,17 @@
 const loadTables = () => new Promise((resolve) => {
   resolve();
 });
 
 const pollTasks = jest.fn();
-const handleTask = jest.fn();
+const handleStartTask = jest.fn();
+const handleFinishedTask = jest.fn();
 
 afterEach(() => {
   pollTasks.mockClear();
-  handleTask.mockClear();
+  handleStartTask.mockClear();
+  handleFinishedTask.mockClear();
 });
 
 describe('subscriptions page', () => {
   const noop = () => {};
   const organization = { owner_details: { upstreamConsumer: {} } };
@@ -37,11 +39,12 @@
     loadTableColumns={loadTableColumns}
     createColumns={createColumns}
     updateColumns={updateColumns}
     loadSubscriptions={loadSubscriptions}
     updateQuantity={updateQuantity}
-    handleTask={handleTask}
+    handleStartTask={handleStartTask}
+    handleFinishedTask={handleFinishedTask}
     pollTaskUntilDone={noop}
     pollBulkSearch={noop}
     pollTasks={pollTasks}
     cancelPollTasks={noop}
     deleteSubscriptions={() => {}}
@@ -69,11 +72,12 @@
     loadTableColumns={loadTableColumns}
     createColumns={createColumns}
     updateColumns={updateColumns}
     loadSubscriptions={loadSubscriptions}
     updateQuantity={updateQuantity}
-    handleTask={handleTask}
+    handleStartTask={handleStartTask}
+    handleFinishedTask={handleFinishedTask}
     pollTaskUntilDone={noop}
     pollBulkSearch={noop}
     pollTasks={pollTasks}
     cancelPollTasks={noop}
     deleteSubscriptions={() => {}}
@@ -116,10 +120,11 @@
       humanized: {
         action: 'Manifest Refresh',
       },
     };
 
-    page.setProps({ task: mockTask });
+    page.setProps({ isTaskPending: true, isPollingTask: true });
+    page.setProps({ task: mockTask, isPollingTask: true, isTaskPending: false });
 
-    expect(handleTask).toHaveBeenCalledWith(mockTask);
+    expect(handleFinishedTask).toHaveBeenCalledWith(mockTask);
   });
 });