forked from mirrors/gecko-dev
63aefee16d
There were compiler errors while I added a gtest testcase. This patch updates the Test* files to include a common header, remove redundant headers, and include missing headers. Differential Revision: https://phabricator.services.mozilla.com/D49880 --HG-- extra : moz-landing-system : lando
59 lines
2.3 KiB
C++
59 lines
2.3 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "nsUrlClassifierDBService.h"
|
|
|
|
#include "Common.h"
|
|
|
|
static void TestResponseCode(const char* table, nsresult result) {
|
|
nsCString tableName(table);
|
|
ASSERT_EQ(TablesToResponse(tableName), result);
|
|
}
|
|
|
|
TEST(UrlClassifierTable, ResponseCode)
|
|
{
|
|
// malware URIs.
|
|
TestResponseCode("goog-malware-shavar", NS_ERROR_MALWARE_URI);
|
|
TestResponseCode("test-malware-simple", NS_ERROR_MALWARE_URI);
|
|
TestResponseCode("goog-phish-shavar,test-malware-simple",
|
|
NS_ERROR_MALWARE_URI);
|
|
TestResponseCode(
|
|
"test-malware-simple,mozstd-track-digest256,mozplugin-block-digest256",
|
|
NS_ERROR_MALWARE_URI);
|
|
|
|
// phish URIs.
|
|
TestResponseCode("goog-phish-shavar", NS_ERROR_PHISHING_URI);
|
|
TestResponseCode("test-phish-simple", NS_ERROR_PHISHING_URI);
|
|
TestResponseCode("test-phish-simple,mozplugin-block-digest256",
|
|
NS_ERROR_PHISHING_URI);
|
|
TestResponseCode(
|
|
"mozstd-track-digest256,test-phish-simple,goog-unwanted-shavar",
|
|
NS_ERROR_PHISHING_URI);
|
|
|
|
// unwanted URIs.
|
|
TestResponseCode("goog-unwanted-shavar", NS_ERROR_UNWANTED_URI);
|
|
TestResponseCode("test-unwanted-simple", NS_ERROR_UNWANTED_URI);
|
|
TestResponseCode("mozplugin-unwanted-digest256,mozfull-track-digest256",
|
|
NS_ERROR_UNWANTED_URI);
|
|
TestResponseCode(
|
|
"test-block-simple,mozfull-track-digest256,test-unwanted-simple",
|
|
NS_ERROR_UNWANTED_URI);
|
|
|
|
// track URIs.
|
|
TestResponseCode("test-track-simple", NS_ERROR_TRACKING_URI);
|
|
TestResponseCode("mozstd-track-digest256", NS_ERROR_TRACKING_URI);
|
|
TestResponseCode("test-block-simple,mozstd-track-digest256",
|
|
NS_ERROR_TRACKING_URI);
|
|
|
|
// block URIs
|
|
TestResponseCode("test-block-simple", NS_ERROR_BLOCKED_URI);
|
|
TestResponseCode("mozplugin-block-digest256", NS_ERROR_BLOCKED_URI);
|
|
TestResponseCode("mozplugin2-block-digest256", NS_ERROR_BLOCKED_URI);
|
|
|
|
TestResponseCode("test-trackwhite-simple", NS_OK);
|
|
TestResponseCode("mozstd-trackwhite-digest256", NS_OK);
|
|
TestResponseCode("goog-badbinurl-shavar", NS_OK);
|
|
TestResponseCode("goog-downloadwhite-digest256", NS_OK);
|
|
}
|