Test yellowlist additions getting cookieblocked.

Author: ghostwords

src/tests/tests/yellowlist.js

@@ -113,4 +113,34 @@
     });
   });
 
+  QUnit.test("Added domains get cookieblocked", (assert) => {
+    const DOMAIN = "example.com";
+
+    let done = assert.async();
+    assert.expect(2);
+
+    // mark domain for blocking
+    badger.storage.setupHeuristicAction(DOMAIN, constants.BLOCK);
+
+    // respond with this domain added
+    let ylist = get_ylist();
+    ylist[DOMAIN] = true;
+    server.respondWith("GET", constants.YELLOWLIST_URL,
+      [200, {}, Object.keys(ylist).join("\n")]);
+
+    // update yellowlist
+    badger.updateYellowlist(function (success) {
+      assert.ok(success, "Callback status indicates success");
+
+      // check that the domain got cookieblocked
+      assert.equal(
+        badger.storage.getAction(DOMAIN),
+        constants.COOKIEBLOCK,
+        "domain is marked for blocking"
+      );
+
+      done();
+    });
+  });
+
 }());