firefox-main/browser/components/urlbar/tests/unit/test_UrlbarUtils_autofillBlock.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* Any copyright is dedicated to the Public Domain.
// Tests UrlbarUtils.blockAutofill, blockOriginAutofill,
// blockOriginPageAutofill, and clearAutofillBlock by calling them and verifying
// moz_origins column state.
"use strict";
const BLOCK_UNTIL_MS = Date.now();
add_task(async function blockOriginAutofill_sets_block_until_ms() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginAutofill(url, BLOCK_UNTIL_MS);
let stored = await getOriginColumn(url, "block_until_ms");
Assert.equal(
stored,
BLOCK_UNTIL_MS,
"block_until_ms should match the value we set"
);
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_updates_existing_value() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
let first = Date.now() + 1000;
await UrlbarUtils.blockOriginAutofill(url, first);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
first,
"First value should be stored"
);
let second = Date.now() + 99000;
await UrlbarUtils.blockOriginAutofill(url, second);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
second,
"Value should be overwritten on second call"
);
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_updates_when_visit_differs() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginAutofill("https://example.com/", BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
"block_until_ms should be stored"
);
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_noop_for_unknown_url() {
// Should not throw when the URL is not in moz_places.
await UrlbarUtils.blockOriginAutofill(
Date.now() + 1000
);
Assert.ok(true, "No exception for a URL that does not exist in Places");
});
add_task(async function blockOriginAutofill_affects_correct_origin() {
await PlacesTestUtils.addVisits([urlA, urlB]);
await UrlbarUtils.blockOriginAutofill(urlA, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(urlA, "block_until_ms"),
BLOCK_UNTIL_MS,
"alpha origin should be blocked"
);
Assert.equal(
await getOriginColumn(urlB, "block_until_ms"),
null,
"beta origin should remain NULL (unblocked)"
);
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_blocks_all_scheme_www_variants() {
// Add all four variations of example.com.
let urls = [
];
await PlacesTestUtils.addVisits(urls);
// Block via just one variation.
await UrlbarUtils.blockOriginAutofill("https://example.com/", BLOCK_UNTIL_MS);
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
`${url} should be blocked after blocking https://example.com/`
);
}
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_from_www_blocks_all_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
// Block via the www + http variation.
await UrlbarUtils.blockOriginAutofill(
BLOCK_UNTIL_MS
);
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
`${url} should be blocked after blocking http://www.example.com/`
);
}
await PlacesUtils.history.clear();
});
add_task(
async function blockOriginAutofill_works_when_only_some_variants_exist() {
// Only two of the four variations have visits.
let present = "https://example.com/";
let alsoPresent = "http://www.example.com/";
let absent = "http://example.com/";
await PlacesTestUtils.addVisits([present, alsoPresent]);
await UrlbarUtils.blockOriginAutofill(absent, BLOCK_UNTIL_MS);
// The two that exist should be blocked; the absent one is a no-op
// (no moz_origins row to update).
Assert.equal(
await getOriginColumn(present, "block_until_ms"),
BLOCK_UNTIL_MS,
"https://example.com/ should be blocked"
);
Assert.equal(
await getOriginColumn(alsoPresent, "block_until_ms"),
BLOCK_UNTIL_MS,
"http://www.example.com/ should be blocked"
);
Assert.equal(
await getOriginColumn(absent, "block_until_ms"),
null,
"http://example.com/ doesn't exist"
);
await PlacesUtils.history.clear();
}
);
add_task(async function blockOriginPageAutofill_sets_block_pages_until_ms() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginPageAutofill(url, BLOCK_UNTIL_MS);
let stored = await getOriginColumn(url, "block_pages_until_ms");
Assert.equal(
stored,
BLOCK_UNTIL_MS,
"block_pages_until_ms should match the value we set"
);
await PlacesUtils.history.clear();
});
add_task(
async function blockOriginPageAutofill_does_not_touch_block_until_ms() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginPageAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
null,
"block_until_ms should remain NULL when only page autofill is blocked"
);
await PlacesUtils.history.clear();
}
);
add_task(async function blockOriginPageAutofill_noop_for_unknown_url() {
await UrlbarUtils.blockOriginPageAutofill(
BLOCK_UNTIL_MS
);
Assert.ok(true, "No exception for a URL that does not exist in Places");
});
add_task(async function blockOriginPageAutofill_shared_across_paths() {
// Two URLs under the same origin share a single moz_origins row.
await PlacesTestUtils.addVisits([urlA, urlB]);
await UrlbarUtils.blockOriginPageAutofill(urlA, BLOCK_UNTIL_MS);
// Both URLs resolve to the same origin row, so querying via urlB should
// show the same block.
Assert.equal(
await getOriginColumn(urlB, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"Block set via urlA should be visible when queried via urlB (same origin)"
);
await PlacesUtils.history.clear();
});
add_task(
async function blockOriginPageAutofill_blocks_all_scheme_www_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
await UrlbarUtils.blockOriginPageAutofill(
BLOCK_UNTIL_MS
);
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
`${url} should be blocked after blocking https://example.com/page`
);
}
await PlacesUtils.history.clear();
}
);
add_task(async function blockOriginPageAutofill_from_www_blocks_all_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
await UrlbarUtils.blockOriginPageAutofill(
BLOCK_UNTIL_MS
);
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
`${url} should be blocked after blocking http://www.example.com/page`
);
}
await PlacesUtils.history.clear();
});
add_task(
async function blockOriginPageAutofill_works_when_only_some_variants_exist() {
let present = "https://example.com/page";
let alsoPresent = "http://www.example.com/page";
let absent = "http://example.com/page";
await PlacesTestUtils.addVisits([present, alsoPresent]);
await UrlbarUtils.blockOriginPageAutofill(absent, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(present, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"https://example.com/page should be blocked"
);
Assert.equal(
await getOriginColumn(alsoPresent, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"http://www.example.com/page should be blocked"
);
Assert.equal(
await getOriginColumn(absent, "block_pages_until_ms"),
null,
"http://example.com/page doesn't exist"
);
await PlacesUtils.history.clear();
}
);
add_task(async function clearOriginPageAutofillBlock_clears_origin_column() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginAutofill(url, BLOCK_UNTIL_MS);
await UrlbarUtils.blockOriginPageAutofill(url, BLOCK_UNTIL_MS);
// Check that both columns are populated.
Assert.equal(await getOriginColumn(url, "block_until_ms"), BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS
);
let didUnblock = await UrlbarUtils.clearOriginAutofillBlock(url);
Assert.ok(didUnblock, "Did unblock");
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
null,
"block_until_ms should be NULL after clearing"
);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"block_pages_until_ms should still have stored value"
);
await PlacesUtils.history.clear();
});
add_task(async function clearOriginPageAutofillBlock_clears_page_column() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginAutofill(url, BLOCK_UNTIL_MS);
await UrlbarUtils.blockOriginPageAutofill(url, BLOCK_UNTIL_MS);
// Check that both columns are populated.
Assert.equal(await getOriginColumn(url, "block_until_ms"), BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS
);
let didUnblock = await UrlbarUtils.clearOriginPageAutofillBlock(url);
Assert.ok(didUnblock, "Did unblock");
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
"block_until_ms should still have stored value"
);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
null,
"block_pages_until_ms should be NULL after clearing"
);
await PlacesUtils.history.clear();
});
add_task(async function clearOriginPageAutofillBlock_clears_all_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
await UrlbarUtils.blockOriginPageAutofill(
BLOCK_UNTIL_MS
);
let didUnblock = await UrlbarUtils.clearOriginPageAutofillBlock(
);
Assert.ok(didUnblock, "Should report that blocks were cleared");
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
null,
`${url} should be unblocked after clearing via http://www.example.com/page`
);
}
await PlacesUtils.history.clear();
});
add_task(async function clearOriginPageAutofillBlock_noop_for_unknown_url() {
let didUnblock = await UrlbarUtils.clearOriginAutofillBlock(
);
Assert.ok(
!didUnblock,
"Did not clear an origin that did not exist in Places"
);
didUnblock = await UrlbarUtils.clearOriginPageAutofillBlock(
);
Assert.ok(!didUnblock, "Did not clear a URL that did not exist in Places");
});
add_task(async function clearOriginAutofillBlock_only_affects_target_origin() {
await PlacesTestUtils.addVisits([urlA, urlB]);
await UrlbarUtils.blockOriginAutofill(urlA, BLOCK_UNTIL_MS);
await UrlbarUtils.blockOriginAutofill(urlB, BLOCK_UNTIL_MS);
// Clear only alpha.
await UrlbarUtils.clearOriginAutofillBlock(urlA);
Assert.equal(
await getOriginColumn(urlA, "block_until_ms"),
null,
"alpha should be cleared"
);
Assert.equal(
await getOriginColumn(urlB, "block_until_ms"),
BLOCK_UNTIL_MS,
"beta should remain blocked"
);
await PlacesUtils.history.clear();
});
add_task(async function clearOriginAutofillBlock_clears_all_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
// Block all variants.
await UrlbarUtils.blockOriginAutofill("https://example.com/", BLOCK_UNTIL_MS);
// Clear via a different variation than the one used to block.
let didUnblock = await UrlbarUtils.clearOriginAutofillBlock(
);
Assert.ok(didUnblock, "Should report that blocks were cleared");
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
null,
`${url} should be unblocked after clearing via http://www.example.com/`
);
}
await PlacesUtils.history.clear();
});
add_task(async function clearOriginAutofillBlock_roundtrip() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
// Start clean.
Assert.equal(await getOriginColumn(url, "block_until_ms"), null);
// Block.
let t1 = Date.now() + 10000;
await UrlbarUtils.blockOriginAutofill(url, t1);
Assert.equal(await getOriginColumn(url, "block_until_ms"), t1);
// Clear.
await UrlbarUtils.clearOriginAutofillBlock(url);
Assert.equal(await getOriginColumn(url, "block_until_ms"), null);
// Block again with a different timestamp.
let t2 = Date.now() + 99000;
await UrlbarUtils.blockOriginAutofill(url, t2);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
t2,
"Should accept a new block after clearing"
);
await PlacesUtils.history.clear();
});
add_task(async function clearOriginPageAutofillBlock_roundtrip() {
await PlacesTestUtils.addVisits(url);
// Start clean.
Assert.equal(await getOriginColumn(url, "block_pages_until_ms"), null);
// Block.
let t1 = Date.now() + 10000;
await UrlbarUtils.blockOriginPageAutofill(url, t1);
Assert.equal(await getOriginColumn(url, "block_pages_until_ms"), t1);
// Clear.
await UrlbarUtils.clearOriginPageAutofillBlock(url);
Assert.equal(await getOriginColumn(url, "block_pages_until_ms"), null);
// Block again with a different timestamp.
let t2 = Date.now() + 99000;
await UrlbarUtils.blockOriginPageAutofill(url, t2);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
t2,
"Should accept a new block after clearing"
);
await PlacesUtils.history.clear();
});
// blockAutofill triggers blockOriginAutofill for origin URLs and
// blockOriginPageAutofill for non-origin URLs.
add_task(async function blockAutofill_origin_sets_block_until_ms() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
"blockAutofill with an origin URL should set block_until_ms"
);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
null,
"block_pages_until_ms should remain NULL for an origin URL"
);
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_page_sets_block_pages_until_ms() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"blockAutofill with a non-origin URL should set block_pages_until_ms"
);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
null,
"block_until_ms should remain NULL for a non-origin URL"
);
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_origin_with_trailing_slash() {
let url = "https://example.com/";
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
"Origin URL with trailing slash should be treated as origin"
);
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_url_with_query_is_page() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"URL with query string should be treated as a page URL"
);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
null,
"block_until_ms should remain NULL for a URL with query string"
);
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_url_with_hash_is_page() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_pages_until_ms"),
BLOCK_UNTIL_MS,
"URL with hash should be treated as a page URL"
);
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_origin_blocks_all_scheme_www_variants() {
let urls = [
];
await PlacesTestUtils.addVisits(urls);
await UrlbarUtils.blockAutofill("https://example.com/", BLOCK_UNTIL_MS);
for (let url of urls) {
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
`${url} should be blocked via blockAutofill origin`
);
}
await PlacesUtils.history.clear();
});
add_task(async function blockAutofill_noop_for_unknown_origin() {
await UrlbarUtils.blockAutofill("https://nonexistent.test/", BLOCK_UNTIL_MS);
Assert.ok(true, "No exception for an unknown origin URL");
});
add_task(async function blockAutofill_noop_for_unknown_page() {
await UrlbarUtils.blockAutofill(
BLOCK_UNTIL_MS
);
Assert.ok(true, "No exception for an unknown page URL");
});
add_task(async function blockOriginAutofill_origin_with_port() {
await PlacesTestUtils.addVisits(url);
await UrlbarUtils.blockOriginAutofill(url, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url, "block_until_ms"),
BLOCK_UNTIL_MS,
"block_until_ms should be set for an origin with a non-standard port"
);
await PlacesUtils.history.clear();
});
add_task(async function blockOriginAutofill_port_does_not_affect_other_ports() {
let url8080 = "https://example.com:8080/";
let url9090 = "https://example.com:9090/";
await PlacesTestUtils.addVisits([url8080, url9090]);
await UrlbarUtils.blockOriginAutofill(url8080, BLOCK_UNTIL_MS);
Assert.equal(
await getOriginColumn(url8080, "block_until_ms"),
BLOCK_UNTIL_MS,
"port 8080 origin should be blocked"
);
Assert.equal(
await getOriginColumn(url9090, "block_until_ms"),
null,
"port 9090 origin should remain unblocked"
);
await PlacesUtils.history.clear();
});