From 2c2e8317de3f84a2465603f814d88f4023822880 Mon Sep 17 00:00:00 2001
From: Shelley Vohr <shelley.vohr@gmail.com>
Date: Sun, 17 Jun 2018 16:00:28 -0700
Subject: [PATCH] spec: convert desktop capturer to expect

---
 spec/api-desktop-capturer-spec.js | 49 ++++++++++++++++++-------------
 1 file changed, 29 insertions(+), 20 deletions(-)

diff --git a/spec/api-desktop-capturer-spec.js b/spec/api-desktop-capturer-spec.js
index 6bc0b230bb40..93e370b52a84 100644
--- a/spec/api-desktop-capturer-spec.js
+++ b/spec/api-desktop-capturer-spec.js
@@ -1,7 +1,11 @@
-const assert = require('assert')
+const chai = require('chai')
+const dirtyChai = require('dirty-chai')
 const {desktopCapturer, remote, screen} = require('electron')
 const features = process.atomBinding('features')
 
+const {expect} = chai
+chai.use(dirtyChai)
+
 const isCI = remote.getGlobal('isCi')
 
 describe('desktopCapturer', () => {
@@ -17,19 +21,19 @@ describe('desktopCapturer', () => {
     }
   })
 
-  it('should return a non-empty array of sources', (done) => {
+  it('should return a non-empty array of sources', done => {
     desktopCapturer.getSources({
       types: ['window', 'screen']
     }, (error, sources) => {
-      assert.equal(error, null)
-      assert.notEqual(sources.length, 0)
+      expect(error).to.be.null()
+      expect(sources.length).to.not.equal(0)
       done()
     })
   })
 
-  it('throws an error for invalid options', (done) => {
-    desktopCapturer.getSources(['window', 'screen'], (error) => {
-      assert.equal(error.message, 'Invalid options')
+  it('throws an error for invalid options', done => {
+    desktopCapturer.getSources(['window', 'screen'], error => {
+      expect(error.message).to.equal('Invalid options')
       done()
     })
   })
@@ -38,8 +42,8 @@ describe('desktopCapturer', () => {
     let callCount = 0
     const callback = (error, sources) => {
       callCount++
-      assert.equal(error, null)
-      assert.notEqual(sources.length, 0)
+      expect(error).to.be.null()
+      expect(sources.length).to.not.equal(0)
       if (callCount === 2) done()
     }
 
@@ -47,11 +51,11 @@ describe('desktopCapturer', () => {
     desktopCapturer.getSources({types: ['window', 'screen']}, callback)
   })
 
-  it('responds to subsequent calls of different options', (done) => {
+  it('responds to subsequent calls of different options', done => {
     let callCount = 0
     const callback = (error, sources) => {
       callCount++
-      assert.equal(error, null)
+      expect(error).to.be.null()
       if (callCount === 2) done()
     }
 
@@ -59,30 +63,35 @@ describe('desktopCapturer', () => {
     desktopCapturer.getSources({types: ['screen']}, callback)
   })
 
-  it('returns an empty display_id for window sources on Windows and Mac', (done) => {
+  it('returns an empty display_id for window sources on Windows and Mac', done => {
     // Linux doesn't return any window sources.
     if (process.platform !== 'win32' && process.platform !== 'darwin') {
       return done()
     }
+
     desktopCapturer.getSources({types: ['window']}, (error, sources) => {
-      assert.equal(error, null)
-      assert.notEqual(sources.length, 0)
-      sources.forEach((source) => { assert.equal(source.display_id.length, 0) })
+      expect(error).to.be.null()
+      expect(sources.length).to.not.equal(0)
+      sources.forEach((source) => {
+        expect(source.display_id.length).to.equal(0)
+      })
       done()
     })
   })
 
-  it('returns display_ids matching the Screen API on Windows and Mac', (done) => {
+  it('returns display_ids matching the Screen API on Windows and Mac', done => {
     if (process.platform !== 'win32' && process.platform !== 'darwin') {
       return done()
     }
+
     const displays = screen.getAllDisplays()
     desktopCapturer.getSources({types: ['screen']}, (error, sources) => {
-      assert.equal(error, null)
-      assert.notEqual(sources.length, 0)
-      assert.equal(sources.length, displays.length)
+      expect(error).to.be.null()
+      expect(sources.length).to.not.equal(0)
+      expect(sources.length).to.equal(displays.length)
+
       for (let i = 0; i < sources.length; i++) {
-        assert.equal(sources[i].display_id, displays[i].id)
+        expect(sources[i].display_id).to.equal(displays[i].id.toString())
       }
       done()
     })