From fcc658a085d8f267236bddf214dd7e2dac32173c Mon Sep 17 00:00:00 2001
From: ablanathtanalba <daly@eff.org>
Date: Wed, 3 Feb 2021 15:25:11 -0800
Subject: [PATCH] add test for tumblr link unwrapping firstparty script

---
 src/tests/tests/firstparties.js | 35 +++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/src/tests/tests/firstparties.js b/src/tests/tests/firstparties.js
index 3560e04ddc..df7ed4af38 100644
--- a/src/tests/tests/firstparties.js
+++ b/src/tests/tests/firstparties.js
@@ -5,6 +5,7 @@ let fb_wrap = 'https://facebook.com/l.php?u=' + destination;
 let fb_xss = 'https://facebook.com/l.php?u=javascript://bad.site/%250Aalert(1)';
 let g_wrap = 'https://www.google.com/url?q=' + destination;
 let g_ping = '/url?url=' + destination;
+let tumblr_wrap = 'https://t.umblr.com/redirect?z=' + destination;
 
 function makeLink(href) {
   let element = document.createElement('a');
@@ -164,4 +165,38 @@ QUnit.test('google search de-instrumentation', (assert) => {
   fixture.appendChild(util_script);
 });
 
+QUnit.test('tumblr link unwrapping', (assert) => {
+  const NUM_CHECKS = 2,
+    done = assert.async();
+  assert.expect(NUM_CHECKS);
+
+  let fixture = document.getElementById('qunit-fixture');
+  let tumblr_link = makeLink(tumblr_wrap);
+
+  // create first-party utility script
+  let util_script = document.createElement('script');
+  util_script.src = '../js/firstparties/lib/utils.js';
+
+  // create the content script
+  let tumblr_script = document.createElement('script');
+  tumblr_script.src = '../js/firstparties/tumblr.js';
+  tumblr_script.onload = function() {
+    assert.equal(tumblr_link.href, destination, 'unwrapped tumblr link');
+    assert.ok(tumblr_link.rel.includes('noreferrer'),
+      'added noreferrer to tumblr link');
+
+    unstub();
+    done();
+  };
+
+  // after the utility script has finished loading, add the content script
+  util_script.onload = function() {
+    fixture.append(tumblr_script);
+  };
+
+  stub([tumblr_link], '/url?');
+  fixture.appendChild(tumblr_link);
+  fixture.appendChild(util_script);
+});
+
 }());