1 files changed, 104 insertions, 0 deletions
diff --git a/browser/components/newtab/test/unit/lib/ShortUrl.test.js b/browser/components/newtab/test/unit/lib/ShortUrl.test.js
new file mode 100644
index 0000000000..e0f6688db8
--- /dev/null
+++ b/browser/components/newtab/test/unit/lib/ShortUrl.test.js
@@ -0,0 +1,104 @@
+import { GlobalOverrider } from "test/unit/utils";
+import { shortURL } from "lib/ShortURL.jsm";
+
+const puny = "xn--kpry57d";
+const idn = "台灣";
+
+describe("shortURL", () => {
+  let globals;
+  let IDNStub;
+  let getPublicSuffixFromHostStub;
+
+  beforeEach(() => {
+    IDNStub = sinon.stub().callsFake(host => host.replace(puny, idn));
+    getPublicSuffixFromHostStub = sinon.stub().returns("com");
+
+    globals = new GlobalOverrider();
+    globals.set("IDNService", { convertToDisplayIDN: IDNStub });
+    globals.set("Services", {
+      eTLD: { getPublicSuffixFromHost: getPublicSuffixFromHostStub },
+    });
+  });
+
+  afterEach(() => {
+    globals.restore();
+  });
+
+  it("should return a blank string if url is falsey", () => {
+    assert.equal(shortURL({ url: false }), "");
+    assert.equal(shortURL({ url: "" }), "");
+    assert.equal(shortURL({}), "");
+  });
+
+  it("should return the 'url' if not a valid url", () => {
+    const checkInvalid = url => assert.equal(shortURL({ url }), url);
+    checkInvalid(true);
+    checkInvalid("something");
+    checkInvalid("http:");
+    checkInvalid("http::double");
+    checkInvalid("http://badport:65536/");
+  });
+
+  it("should remove the eTLD", () => {
+    assert.equal(shortURL({ url: "http://com.blah.com" }), "com.blah");
+  });
+
+  it("should convert host to idn when calling shortURL", () => {
+    assert.equal(shortURL({ url: `http://${puny}.blah.com` }), `${idn}.blah`);
+  });
+
+  it("should get the hostname from .url", () => {
+    assert.equal(shortURL({ url: "http://bar.com" }), "bar");
+  });
+
+  it("should not strip out www if not first subdomain", () => {
+    assert.equal(shortURL({ url: "http://foo.www.com" }), "foo.www");
+  });
+
+  it("should convert to lowercase", () => {
+    assert.equal(shortURL({ url: "HTTP://FOO.COM" }), "foo");
+  });
+
+  it("should not include the port", () => {
+    assert.equal(shortURL({ url: "http://foo.com:8888" }), "foo");
+  });
+
+  it("should return hostname for localhost", () => {
+    getPublicSuffixFromHostStub.throws("insufficient domain levels");
+
+    assert.equal(shortURL({ url: "http://localhost:8000/" }), "localhost");
+  });
+
+  it("should return hostname for ip address", () => {
+    getPublicSuffixFromHostStub.throws("host is ip address");
+
+    assert.equal(shortURL({ url: "http://127.0.0.1/foo" }), "127.0.0.1");
+  });
+
+  it("should return etld for www.gov.uk (www-only non-etld)", () => {
+    getPublicSuffixFromHostStub.returns("gov.uk");
+
+    assert.equal(
+      shortURL({ url: "https://www.gov.uk/countersigning" }),
+      "gov.uk"
+    );
+  });
+
+  it("should return idn etld for www-only non-etld", () => {
+    getPublicSuffixFromHostStub.returns(puny);
+
+    assert.equal(shortURL({ url: `https://www.${puny}/foo` }), idn);
+  });
+
+  it("should return not the protocol for file:", () => {
+    assert.equal(shortURL({ url: "file:///foo/bar.txt" }), "/foo/bar.txt");
+  });
+
+  it("should return not the protocol for about:", () => {
+    assert.equal(shortURL({ url: "about:newtab" }), "newtab");
+  });
+
+  it("should fall back to full url as a last resort", () => {
+    assert.equal(shortURL({ url: "about:" }), "about:");
+  });
+});