test: add driver.close assertions (#479)

* test: add driver.close assertions

* test: play nice with the rest of tests

packages/browserless/package.json

@@ -44,6 +44,7 @@
     "superlock": "~1.0.1"
   },
   "devDependencies": {
+    "ps-list": "7",
     "@browserless/test": "^9.11.0",
     "ava": "latest"
   },
@@ -62,4 +63,4 @@
     "timeout": "30s",
     "workerThreads": false
   }
-}
+}

packages/browserless/test/driver.js

@@ -0,0 +1,43 @@
+'use strict'
+
+const { createBrowser } = require('@browserless/test/util')
+const psList = require('ps-list')
+const isCI = require('is-ci')
+const test = require('ava')
+
+const getChromiumPs = async () => {
+  const ps = await psList()
+  return ps.filter(ps => ps.name.includes('Chromium'))
+}
+
+;(isCI ? test.skip : test.serial)('.close() will kill process and subprocess', async t => {
+  const initialPs = await getChromiumPs()
+
+  const browserlessFactory = createBrowser()
+  t.teardown(() => browserlessFactory.close())
+
+  t.is((await getChromiumPs()).length, 1)
+  t.is((await getChromiumPs())[0].pid, (await browserlessFactory.browser()).process().pid)
+
+  const browserless = await browserlessFactory.createContext()
+  t.is((await getChromiumPs()).length, 3)
+
+  await browserless.destroyContext()
+  t.is((await getChromiumPs()).length, 3)
+
+  await browserlessFactory.close()
+  t.is((await getChromiumPs()).length, initialPs)
+})
+
+;(isCI ? test.skip : test.serial)('.close() is idempotency', async t => {
+  const initialPs = await getChromiumPs()
+
+  const browserlessFactory = createBrowser()
+  t.is((await getChromiumPs()).length, 1)
+
+  await browserlessFactory.close()
+  t.is((await getChromiumPs()).length, initialPs)
+
+  await browserlessFactory.close()
+  t.is((await getChromiumPs()).length, initialPs)
+})