1// Copyright (c) 2011 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 "net/ftp/ftp_auth_cache.h"
6
7#include "base/strings/string_number_conversions.h"
8#include "base/strings/string_util.h"
9#include "base/strings/utf_string_conversions.h"
10#include "net/base/auth.h"
11#include "testing/gtest/include/gtest/gtest.h"
12#include "url/gurl.h"
13
14using base::ASCIIToUTF16;
15using net::FtpAuthCache;
16
17namespace {
18
19const base::string16 kBogus(ASCIIToUTF16("bogus"));
20const base::string16 kOthername(ASCIIToUTF16("othername"));
21const base::string16 kOtherword(ASCIIToUTF16("otherword"));
22const base::string16 kPassword(ASCIIToUTF16("password"));
23const base::string16 kPassword1(ASCIIToUTF16("password1"));
24const base::string16 kPassword2(ASCIIToUTF16("password2"));
25const base::string16 kPassword3(ASCIIToUTF16("password3"));
26const base::string16 kUsername(ASCIIToUTF16("username"));
27const base::string16 kUsername1(ASCIIToUTF16("username1"));
28const base::string16 kUsername2(ASCIIToUTF16("username2"));
29const base::string16 kUsername3(ASCIIToUTF16("username3"));
30
31}  // namespace
32
33TEST(FtpAuthCacheTest, LookupAddRemove) {
34  FtpAuthCache cache;
35
36  GURL origin1("ftp://foo1");
37  GURL origin2("ftp://foo2");
38
39  // Lookup non-existent entry.
40  EXPECT_TRUE(cache.Lookup(origin1) == NULL);
41
42  // Add entry for origin1.
43  cache.Add(origin1, net::AuthCredentials(kUsername1, kPassword1));
44  FtpAuthCache::Entry* entry1 = cache.Lookup(origin1);
45  ASSERT_TRUE(entry1);
46  EXPECT_EQ(origin1, entry1->origin);
47  EXPECT_EQ(kUsername1, entry1->credentials.username());
48  EXPECT_EQ(kPassword1, entry1->credentials.password());
49
50  // Add an entry for origin2.
51  cache.Add(origin2, net::AuthCredentials(kUsername2, kPassword2));
52  FtpAuthCache::Entry* entry2 = cache.Lookup(origin2);
53  ASSERT_TRUE(entry2);
54  EXPECT_EQ(origin2, entry2->origin);
55  EXPECT_EQ(kUsername2, entry2->credentials.username());
56  EXPECT_EQ(kPassword2, entry2->credentials.password());
57
58  // The original entry1 should still be there.
59  EXPECT_EQ(entry1, cache.Lookup(origin1));
60
61  // Overwrite the entry for origin1.
62  cache.Add(origin1, net::AuthCredentials(kUsername3, kPassword3));
63  FtpAuthCache::Entry* entry3 = cache.Lookup(origin1);
64  ASSERT_TRUE(entry3);
65  EXPECT_EQ(origin1, entry3->origin);
66  EXPECT_EQ(kUsername3, entry3->credentials.username());
67  EXPECT_EQ(kPassword3, entry3->credentials.password());
68
69  // Remove entry of origin1.
70  cache.Remove(origin1, net::AuthCredentials(kUsername3, kPassword3));
71  EXPECT_TRUE(cache.Lookup(origin1) == NULL);
72
73  // Remove non-existent entry.
74  cache.Remove(origin1, net::AuthCredentials(kUsername3, kPassword3));
75  EXPECT_TRUE(cache.Lookup(origin1) == NULL);
76}
77
78// Check that if the origin differs only by port number, it is considered
79// a separate origin.
80TEST(FtpAuthCacheTest, LookupWithPort) {
81  FtpAuthCache cache;
82
83  GURL origin1("ftp://foo:80");
84  GURL origin2("ftp://foo:21");
85
86  cache.Add(origin1, net::AuthCredentials(kUsername, kPassword));
87  cache.Add(origin2, net::AuthCredentials(kUsername, kPassword));
88
89  EXPECT_NE(cache.Lookup(origin1), cache.Lookup(origin2));
90}
91
92TEST(FtpAuthCacheTest, NormalizedKey) {
93  // GURL is automatically canonicalized. Hence the following variations in
94  // url format should all map to the same entry (case insensitive host,
95  // default port of 21).
96
97  FtpAuthCache cache;
98
99  // Add.
100  cache.Add(GURL("ftp://HoSt:21"), net::AuthCredentials(kUsername, kPassword));
101
102  // Lookup.
103  FtpAuthCache::Entry* entry1 = cache.Lookup(GURL("ftp://HoSt:21"));
104  ASSERT_TRUE(entry1);
105  EXPECT_EQ(entry1, cache.Lookup(GURL("ftp://host:21")));
106  EXPECT_EQ(entry1, cache.Lookup(GURL("ftp://host")));
107
108  // Overwrite.
109  cache.Add(GURL("ftp://host"), net::AuthCredentials(kOthername, kOtherword));
110  FtpAuthCache::Entry* entry2 = cache.Lookup(GURL("ftp://HoSt:21"));
111  ASSERT_TRUE(entry2);
112  EXPECT_EQ(GURL("ftp://host"), entry2->origin);
113  EXPECT_EQ(kOthername, entry2->credentials.username());
114  EXPECT_EQ(kOtherword, entry2->credentials.password());
115
116  // Remove
117  cache.Remove(GURL("ftp://HOsT"),
118               net::AuthCredentials(kOthername, kOtherword));
119  EXPECT_TRUE(cache.Lookup(GURL("ftp://host")) == NULL);
120}
121
122TEST(FtpAuthCacheTest, OnlyRemoveMatching) {
123  FtpAuthCache cache;
124
125  cache.Add(GURL("ftp://host"), net::AuthCredentials(kUsername, kPassword));
126  EXPECT_TRUE(cache.Lookup(GURL("ftp://host")));
127
128  // Auth data doesn't match, shouldn't remove.
129  cache.Remove(GURL("ftp://host"), net::AuthCredentials(kBogus, kBogus));
130  EXPECT_TRUE(cache.Lookup(GURL("ftp://host")));
131
132  // Auth data matches, should remove.
133  cache.Remove(GURL("ftp://host"), net::AuthCredentials(kUsername, kPassword));
134  EXPECT_TRUE(cache.Lookup(GURL("ftp://host")) == NULL);
135}
136
137TEST(FtpAuthCacheTest, EvictOldEntries) {
138  FtpAuthCache cache;
139
140  for (size_t i = 0; i < FtpAuthCache::kMaxEntries; i++) {
141    cache.Add(GURL("ftp://host" + base::IntToString(i)),
142              net::AuthCredentials(kUsername, kPassword));
143  }
144
145  // No entries should be evicted before reaching the limit.
146  for (size_t i = 0; i < FtpAuthCache::kMaxEntries; i++) {
147    EXPECT_TRUE(cache.Lookup(GURL("ftp://host" + base::IntToString(i))));
148  }
149
150  // Adding one entry should cause eviction of the first entry.
151  cache.Add(GURL("ftp://last_host"),
152            net::AuthCredentials(kUsername, kPassword));
153  EXPECT_TRUE(cache.Lookup(GURL("ftp://host0")) == NULL);
154
155  // Remaining entries should not get evicted.
156  for (size_t i = 1; i < FtpAuthCache::kMaxEntries; i++) {
157    EXPECT_TRUE(cache.Lookup(GURL("ftp://host" + base::IntToString(i))));
158  }
159  EXPECT_TRUE(cache.Lookup(GURL("ftp://last_host")));
160}
161