summaryrefslogtreecommitdiffstats
path: root/chrome/browser/safe_browsing/ping_manager_unittest.cc
blob: 95a8686dd7309784f17c1dec4f74da739cc51781 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//

#include "base/logging.h"
#include "base/strings/stringprintf.h"
#include "base/time/time.h"
#include "chrome/browser/safe_browsing/ping_manager.h"
#include "google_apis/google_api_keys.h"
#include "net/base/escape.h"
#include "testing/gtest/include/gtest/gtest.h"

using base::Time;
using base::TimeDelta;

static const char kUrlPrefix[] = "https://prefix.com/foo";
static const char kClient[] = "unittest";
static const char kAppVer[] = "1.0";

class SafeBrowsingPingManagerTest : public testing::Test {
 protected:
  std::string key_param_;

  void SetUp() override {
    std::string key = google_apis::GetAPIKey();
    if (!key.empty()) {
      key_param_ = base::StringPrintf(
          "&key=%s",
          net::EscapeQueryParamValue(key, true).c_str());
    }
  }
};

TEST_F(SafeBrowsingPingManagerTest, TestSafeBrowsingHitUrl) {
  SafeBrowsingProtocolConfig config;
  config.client_name = kClient;
  config.url_prefix = kUrlPrefix;
  SafeBrowsingPingManager pm(NULL, config);

  pm.version_ = kAppVer;

  GURL malicious_url("http://malicious.url.com");
  GURL page_url("http://page.url.com");
  GURL referrer_url("http://referrer.url.com");
  EXPECT_EQ(
      "https://prefix.com/foo/report?client=unittest&appver=1.0&"
      "pver=3.0" +
          key_param_ +
          "&ext=1&evts=malblhit&evtd=http%3A%2F%2Fmalicious.url.com%2F&"
          "evtr=http%3A%2F%2Fpage.url.com%2F&evhr=http%3A%2F%2Freferrer."
          "url.com%2F&evtb=1",
      pm.SafeBrowsingHitUrl(malicious_url, page_url, referrer_url, true,
                            SB_THREAT_TYPE_URL_MALWARE, true).spec());

  EXPECT_EQ(
      "https://prefix.com/foo/report?client=unittest&appver=1.0&"
      "pver=3.0" +
          key_param_ +
          "&ext=1&evts=phishblhit&"
          "evtd=http%3A%2F%2Fmalicious.url.com%2F&"
          "evtr=http%3A%2F%2Fpage.url.com%2F&evhr=http%3A%2F%2Freferrer."
          "url.com%2F&evtb=0",
      pm.SafeBrowsingHitUrl(malicious_url, page_url, referrer_url, false,
                            SB_THREAT_TYPE_URL_PHISHING, true).spec());

  EXPECT_EQ(
      "https://prefix.com/foo/report?client=unittest&appver=1.0&"
      "pver=3.0" +
          key_param_ +
          "&ext=0&evts=binurlhit&"
          "evtd=http%3A%2F%2Fmalicious.url.com%2F&"
          "evtr=http%3A%2F%2Fpage.url.com%2F&evhr=http%3A%2F%2Freferrer."
          "url.com%2F&evtb=0",
      pm.SafeBrowsingHitUrl(malicious_url, page_url, referrer_url, false,
                            SB_THREAT_TYPE_BINARY_MALWARE_URL, false).spec());

  EXPECT_EQ(
      "https://prefix.com/foo/report?client=unittest&appver=1.0&"
      "pver=3.0" +
          key_param_ +
          "&ext=0&evts=phishcsdhit&"
          "evtd=http%3A%2F%2Fmalicious.url.com%2F&"
          "evtr=http%3A%2F%2Fpage.url.com%2F&evhr=http%3A%2F%2Freferrer."
          "url.com%2F&evtb=0",
      pm.SafeBrowsingHitUrl(malicious_url, page_url, referrer_url, false,
                            SB_THREAT_TYPE_CLIENT_SIDE_PHISHING_URL,
                            false).spec());

  EXPECT_EQ(
      "https://prefix.com/foo/report?client=unittest&appver=1.0&"
      "pver=3.0" +
          key_param_ +
          "&ext=0&evts=malcsdhit&"
          "evtd=http%3A%2F%2Fmalicious.url.com%2F&"
          "evtr=http%3A%2F%2Fpage.url.com%2F&evhr=http%3A%2F%2Freferrer."
          "url.com%2F&evtb=1",
      pm.SafeBrowsingHitUrl(malicious_url, page_url, referrer_url, true,
                            SB_THREAT_TYPE_CLIENT_SIDE_MALWARE_URL,
                            false).spec());
}

TEST_F(SafeBrowsingPingManagerTest, TestMalwareDetailsUrl) {
  SafeBrowsingProtocolConfig config;
  config.client_name = kClient;
  config.url_prefix = kUrlPrefix;
  SafeBrowsingPingManager pm(NULL, config);

  pm.version_ = kAppVer;
  EXPECT_EQ("https://prefix.com/foo/clientreport/malware?"
            "client=unittest&appver=1.0&pver=1.0" + key_param_,
            pm.MalwareDetailsUrl().spec());
}