Add {get,set}CookieValue

testiumjs · Oct 27, 2015 · 7721cfa · 7721cfa
1 parent e447b6f
commit 7721cfa
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/lib/browser/cookie.js b/lib/browser/cookie.js
@@ -10,6 +10,20 @@ exports.clearCookies = function clearCookies() {
   return this.deleteAllCookies();
 };
 
+exports.setCookieValue = function setCookieValue(name, value, options) {
+  var cookie = _.extend({
+    name: name,
+    value: value,
+    path: '/',
+    secure: false,
+  }, options || {});
+  return this.setCookie(cookie);
+};
+
+exports.getCookieValue = function getCookieValue(name) {
+  return this.getCookie(name).then(_.property('value'));
+};
+
 exports.setCookies = function setCookies(cookies) {
   return Bluebird.all(cookies.map(this.setCookie, this));
 };

diff --git a/test/integration/cookie.test.js b/test/integration/cookie.test.js
@@ -50,4 +50,22 @@ describe('cookie', () => {
     const cookie = await browser.getCookie('test_cookie');
     assert.falsey(cookie);
   });
+
+  describe('setCookieValue', () => {
+    before(() =>
+      browser
+        .navigateTo('/some/nested/url')
+        .setCookie({ name: 'non_root', value: 'a' })
+        .setCookieValue('root', 'b')
+        .navigateTo('/'));
+
+    it('can\'t read the non-root cookie', async () => {
+      assert.falsey(await browser.getCookie('non_root'));
+    });
+
+    it('can read the root cookie', async () => {
+      assert.truthy(await browser.getCookie('root'));
+      assert.equal('b', await browser.getCookieValue('root'));
+    });
+  });
 });