From 34f6fa4fab502497b6ffd5d04cf0547bb9569fc2 Mon Sep 17 00:00:00 2001
From: Brian Vaughn <bvaughn@fb.com>
Date: Thu, 23 May 2019 11:22:02 -0700
Subject: [PATCH] Added a sanity test for Bridge shutdown

---
 src/__tests__/bridge-test.js | 42 ++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 src/__tests__/bridge-test.js

diff --git a/src/__tests__/bridge-test.js b/src/__tests__/bridge-test.js
new file mode 100644
index 0000000000..1a9b4b949a
--- /dev/null
+++ b/src/__tests__/bridge-test.js
@@ -0,0 +1,42 @@
+// @flow
+
+describe('Bridge', () => {
+  let Bridge;
+
+  beforeEach(() => {
+    Bridge = require('src/bridge').default;
+  });
+
+  it('should shutdown properly', () => {
+    const wall = {
+      listen: jest.fn(() => () => {}),
+      send: jest.fn(),
+    };
+    const bridge = new Bridge(wall);
+
+    // Check that we're wired up correctly.
+    bridge.send('init');
+    jest.runAllTimers();
+    expect(wall.send).toHaveBeenCalledWith('init', undefined, undefined);
+
+    // Should flush pending messages and then shut down.
+    wall.send.mockClear();
+    bridge.send('update', '1');
+    bridge.send('update', '2');
+    bridge.shutdown();
+    jest.runAllTimers();
+    expect(wall.send).toHaveBeenCalledWith('update', '1', undefined);
+    expect(wall.send).toHaveBeenCalledWith('update', '2', undefined);
+    expect(wall.send).toHaveBeenCalledWith('shutdown', undefined, undefined);
+
+    // Verify that the Bridge doesn't send messages after shutdown.
+    spyOn(console, 'warn');
+    wall.send.mockClear();
+    bridge.send('should not send');
+    jest.runAllTimers();
+    expect(wall.send).not.toHaveBeenCalled();
+    expect(console.warn).toHaveBeenCalledWith(
+      'Cannot send message "should not send" through a Bridge that has been shutdown.'
+    );
+  });
+});