1 files changed, 16 insertions, 31 deletions

diff --git a/test/shared/versions/index.test.js b/test/shared/versions/index.test.js
index 3bb307a..d90f04c 100644
--- a/test/shared/versions/index.test.js
+++ b/test/shared/versions/index.test.js
@@ -7,32 +7,21 @@ describe("shared/versions/storage", () => {
       return browser.storage.local.remove('version');
     });
 
-    it('return true if no previous versions', () => {
-      return Promise.resolve().then(() => {
-        return versions.checkUpdated();
-      }).then((updated) => {
-        expect(updated).to.be.true;
-      });
+    it('return true if no previous versions', async() => {
+      let updated = await versions.checkUpdated();
+      expect(updated).to.be.true;
     });
 
-    it('return true if updated', () => {
-      return Promise.resolve().then(() => {
-        return browser.storage.local.set({ version: '0.001' });
-      }).then(() => {
-        return versions.checkUpdated();
-      }).then((updated) => {
-        expect(updated).to.be.true;
-      });
+    it('return true if updated', async() => {
+      await browser.storage.local.set({ version: '0.001' });
+      let updated = await versions.checkUpdated();
+      expect(updated).to.be.true;
     });
 
-    it('return false if not updated', () => {
-      return Promise.resolve().then(() => {
-        return browser.storage.local.set({ version: manifest.version });
-      }).then(() => {
-        return versions.checkUpdated();
-      }).then((updated) => {
-        expect(updated).to.be.false;
-      });
+    it('return false if not updated', async() => {
+      await browser.storage.local.set({ version: manifest.version });
+      let updated = await versions.checkUpdated();
+      expect(updated).to.be.false;
     });
   });
 
@@ -41,15 +30,11 @@ describe("shared/versions/storage", () => {
       return browser.storage.local.remove('version');
     });
 
-    it('saves current version from manifest.json', () => {
-      return Promise.resolve().then(() => {
-        return versions.commit();
-      }).then(() => {
-        return browser.storage.local.get('version');
-      }).then(({version}) => {
-        expect(version).to.be.a('string');
-        expect(version).to.equal(manifest.version);
-      });
+    it('saves current version from manifest.json', async() => {
+      await versions.commit();
+      let { version } = await browser.storage.local.get('version');
+      expect(version).to.be.a('string');
+      expect(version).to.equal(manifest.version);
     });
   });
 });