Add remove() to StatsWidget

modules/stats-widget/src/stats-widget.ts

@@ -58,6 +58,7 @@ export default class StatsWidget {
   _statsContainer: HTMLElement = null;
   _header: HTMLElement = null;
   _items = {};
+  _added: boolean = false;
 
   constructor(stats: Stats, options?: StatWidgetProps) {
     this.stats = stats;
@@ -104,6 +105,16 @@ export default class StatsWidget {
     this._update();
   }
 
+  /**
+   * Remove the stats widget from the container it was added to.
+   * The stats widget cannot be reused after this is called.
+   */
+  remove(): void {
+    // if re-adding the stats widget is needed, a code path to
+    // re-add the _innerContainer should be added, e.g. in _update.
+    this._container.removeChild(this._innerContainer);
+  }
+
   setCollapsed(collapsed: boolean): void {
     this.collapsed = collapsed;
     if (this._statsContainer) {

modules/stats-widget/test/stats-widget.spec.js

@@ -197,3 +197,14 @@ test('StatsWidget#resetOnUpdate', t => {
 
   t.end();
 });
+
+test('StatsWidget#remove', t => {
+  const container = _global.document.createElement('div');
+  container.id = 'test-stats-widget-container';
+  const statsWidget = new StatsWidget(null, {container});
+  t.ok(statsWidget._container === container);
+  t.equals(statsWidget._container.childNodes.length, 1, 'Should have 1 child node.');
+  statsWidget.remove();
+  t.equals(statsWidget._container.childNodes.length, 0, 'Should have 0 child nodes.');
+  t.end();
+});