Update tests

test/src/notebook/cells/widget.spec.ts

@@ -803,7 +803,7 @@ describe('notebook/cells/widget', () => {
     describe('#constructor()', () => {
 
       it('should create a raw cell widget', () => {
-        let widget = new RawCellWidget({renderer:CodeMirrorNotebookRenderer.defaultRawCellRenderer});
+        let widget = new RawCellWidget({renderer: CodeMirrorNotebookRenderer.defaultRawCellRenderer});
         expect(widget).to.be.a(RawCellWidget);
       });
 

test/src/notebook/notebook/widget.spec.ts

@@ -709,12 +709,11 @@ describe('notebook/notebook/widget', () => {
         Widget.attach(widget, document.body);
         widget.mode = 'edit';
         let cell = widget.childAt(widget.activeCellIndex);
-        // Notebook activates cell.
+        // Wait for update-request.
         requestAnimationFrame(() => {
-          // Cell activates editor.
+          // Notebook activates the editor.
           requestAnimationFrame(() => {
             expect(cell.node.contains(document.activeElement)).to.be(true);
-            widget.dispose();
             done();
           });
         });
@@ -1044,13 +1043,16 @@ describe('notebook/notebook/widget', () => {
 
     describe('#onActivateRequest()', () => {
 
-      it('should focus the node', () => {
+      it('should focus the node after an update', (done) => {
         let widget = createActiveWidget();
         Widget.attach(widget, document.body);
         sendMessage(widget, WidgetMessage.ActivateRequest);
         expect(widget.methods).to.contain('onActivateRequest');
-        expect(document.activeElement).to.be(widget.node);
-        widget.dispose();
+        requestAnimationFrame(() => {
+          expect(document.activeElement).to.be(widget.node);
+          widget.dispose();
+          done();
+        });
       });
 
       it('should post an `update-request', (done) => {