Enable tests from before

These tests were written with replaying in mind and now we can properly
enable them.

packages/react-dom/src/__tests__/ReactDOMServerPartialHydration-test.internal.js

@@ -1805,12 +1805,6 @@ describe('ReactDOMServerPartialHydration', () => {
     Scheduler.unstable_flushAll();
     jest.runAllTimers();
 
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      a.click();
-    });
-
     expect(clicks).toBe(1);
 
     expect(container.textContent).toBe('Hello');
@@ -1885,12 +1879,6 @@ describe('ReactDOMServerPartialHydration', () => {
     Scheduler.unstable_flushAll();
     jest.runAllTimers();
 
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      a.click();
-    });
-
     expect(onEvent).toHaveBeenCalledTimes(1);
 
     document.body.removeChild(container);
@@ -1963,12 +1951,6 @@ describe('ReactDOMServerPartialHydration', () => {
     Scheduler.unstable_flushAll();
     jest.runAllTimers();
 
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      span.click();
-    });
-
     expect(clicksOnChild).toBe(1);
     // This will be zero due to the stopPropagation.
     expect(clicksOnParent).toBe(0);
@@ -2047,11 +2029,6 @@ describe('ReactDOMServerPartialHydration', () => {
     jest.runAllTimers();
 
     // We're now full hydrated.
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      a.click();
-    });
 
     expect(clicks).toBe(1);
 

packages/react-dom/src/__tests__/ReactServerRenderingHydration-test.js

@@ -13,7 +13,6 @@ let React;
 let ReactDOM;
 let ReactDOMServer;
 let Scheduler;
-let act;
 
 // These tests rely both on ReactDOMServer and ReactDOM.
 // If a test only needs ReactDOMServer, put it in ReactServerRendering-test instead.
@@ -24,7 +23,6 @@ describe('ReactDOMServerHydration', () => {
     ReactDOM = require('react-dom');
     ReactDOMServer = require('react-dom/server');
     Scheduler = require('scheduler');
-    act = require('react-dom/test-utils').act;
   });
 
   it('should have the correct mounting behavior (old hydrate API)', () => {
@@ -572,13 +570,12 @@ describe('ReactDOMServerHydration', () => {
     expect(clicks).toBe(0);
 
     // Finish the rest of the hydration.
-    expect(Scheduler).toFlushAndYield(['Sibling2']);
-
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      a.click();
-    });
+    if (__DEV__) {
+      // In DEV effects gets double invoked.
+      expect(Scheduler).toFlushAndYield(['Sibling2', 'Button', 'Button']);
+    } else {
+      expect(Scheduler).toFlushAndYield(['Sibling2', 'Button']);
+    }
 
     expect(clicks).toBe(1);
 
@@ -634,11 +631,6 @@ describe('ReactDOMServerHydration', () => {
     Scheduler.unstable_flushAll();
 
     // We're now full hydrated.
-    // TODO: With selective hydration the event should've been replayed
-    // but for now we'll have to issue it again.
-    act(() => {
-      a.click();
-    });
 
     expect(clicks).toBe(1);