remoting/host/setup/oauth_helper_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "remoting/host/setup/oauth_helper.h"
   6
   7 #include "testing/gtest/include/gtest/gtest.h"
   8
   9 namespace {
  10
  11 std::string Replace(const std::string& s, const std::string& old_substr,
  12                     const std::string& new_substr) {
  13   size_t pos = s.find(old_substr);
  14   if (pos == std::string::npos) {
  15     return s;
  16   }
  17   return s.substr(0, pos) + new_substr +
  18       s.substr(pos + old_substr.length(), std::string::npos);
  19 }
  20
  21 std::string GetTestRedirectUrl() {
  22   return std::string("https://google.com/redirect");
  23 }
  24
  25 }  // namespace
  26
  27 namespace remoting {
  28
  29 TEST(OauthHelperTest, TestNotCode) {
  30   ASSERT_EQ("", GetOauthCodeInUrl("notURL", GetTestRedirectUrl()));
  31 }
  32
  33 TEST(OauthHelperTest, TestVeryShort) {
  34   ASSERT_EQ("", GetOauthCodeInUrl(GetTestRedirectUrl(), GetTestRedirectUrl()));
  35 }
  36
  37 TEST(OauthHelperTest, TestEmptyQuery) {
  38   ASSERT_EQ("", GetOauthCodeInUrl(GetTestRedirectUrl() + "?",
  39                                   GetTestRedirectUrl()));
  40 }
  41
  42 TEST(OauthHelperTest, TestNoQueryValue) {
  43   ASSERT_EQ("", GetOauthCodeInUrl(GetTestRedirectUrl() + "?code",
  44                                   GetTestRedirectUrl()));
  45 }
  46
  47 TEST(OauthHelperTest, TestEmptyCode) {
  48   ASSERT_EQ("", GetOauthCodeInUrl(GetTestRedirectUrl() + "?code=",
  49                                   GetTestRedirectUrl()));
  50 }
  51
  52 TEST(OauthHelperTest, TestCode) {
  53   ASSERT_EQ("Dummy", GetOauthCodeInUrl(GetTestRedirectUrl() + "?code=Dummy",
  54                                        GetTestRedirectUrl()));
  55 }
  56
  57 TEST(OauthHelperTest, TestCodeInLongQuery) {
  58   ASSERT_EQ("Dummy", GetOauthCodeInUrl(GetTestRedirectUrl() +
  59                                            "?x=1&code=Dummy&y=2",
  60                                        GetTestRedirectUrl()));
  61 }
  62
  63 TEST(OauthHelperTest, TestBadScheme) {
  64   std::string url = GetTestRedirectUrl() + "?code=Dummy";
  65   url = Replace(url, "https:", "http");
  66   ASSERT_EQ("", GetOauthCodeInUrl(url, GetTestRedirectUrl()));
  67 }
  68
  69 TEST(OauthHelperTest, TestBadHost) {
  70   std::string url = GetTestRedirectUrl() + "?code=Dummy";
  71   url = Replace(url, "google", "goggle");
  72   ASSERT_EQ("", GetOauthCodeInUrl(url, GetTestRedirectUrl()));
  73 }
  74
  75 }  // namespace remoting