Fix act warnings (#5811)

Author: mattseddon

webview/src/plots/components/App.test.tsx

@@ -3,6 +3,7 @@ import { configureStore } from '@reduxjs/toolkit'
 import React from 'react'
 import { Provider } from 'react-redux'
 import {
+  act,
   createEvent,
   fireEvent,
   render,
@@ -31,7 +32,6 @@ import {
   MessageFromWebviewType,
   MessageToWebviewType
 } from 'dvc/src/webview/contract'
-import { act } from 'react-dom/test-utils'
 import { App } from './App'
 import { NewSectionBlock } from './templatePlots/TemplatePlots'
 import {

webview/src/plots/components/comparisonTable/ComparisonTable.test.tsx

@@ -1,6 +1,7 @@
 import { configureStore } from '@reduxjs/toolkit'
 import '@testing-library/jest-dom'
 import {
+  act,
   fireEvent,
   render,
   RenderResult,
@@ -13,7 +14,6 @@ import comparisonTableFixture from 'dvc/src/test/fixtures/plotsDiff/comparison'
 import plotsRevisionsFixture from 'dvc/src/test/fixtures/plotsDiff/revisions'
 import React from 'react'
 import { Revision } from 'dvc/src/plots/webview/contract'
-import { act } from 'react-dom/test-utils'
 import { ComparisonTable } from './ComparisonTable'
 import {
   comparisonTableInitialState,

webview/src/test/dragDrop.ts

@@ -1,4 +1,4 @@
-import { act } from 'react-dom/test-utils'
+import { act } from '@testing-library/react'
 import { DragEnterDirection } from '../shared/components/dragDrop/util'
 import { idToNode } from '../util/helpers'