1// Copyright 2014 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 "base/strings/string_piece.h"
6#include "base/strings/utf_string_conversions.h"
7#include "storage/browser/database/database_util.h"
8#include "storage/common/database/database_identifier.h"
9#include "testing/gtest/include/gtest/gtest.h"
10
11using base::ASCIIToUTF16;
12using storage::DatabaseUtil;
13
14static void TestVfsFilePath(bool expected_result,
15                            const char* vfs_file_name,
16                            const char* expected_origin_identifier = "",
17                            const char* expected_database_name = "",
18                            const char* expected_sqlite_suffix = "") {
19  std::string origin_identifier;
20  base::string16 database_name;
21  base::string16 sqlite_suffix;
22  EXPECT_EQ(expected_result,
23            DatabaseUtil::CrackVfsFileName(ASCIIToUTF16(vfs_file_name),
24                                           &origin_identifier,
25                                           &database_name,
26                                           &sqlite_suffix));
27  EXPECT_EQ(expected_origin_identifier, origin_identifier);
28  EXPECT_EQ(ASCIIToUTF16(expected_database_name), database_name);
29  EXPECT_EQ(ASCIIToUTF16(expected_sqlite_suffix), sqlite_suffix);
30}
31
32static GURL ToAndFromOriginIdentifier(const GURL origin_url) {
33  std::string id = storage::GetIdentifierFromOrigin(origin_url);
34  return storage::GetOriginFromIdentifier(id);
35}
36
37static void TestValidOriginIdentifier(bool expected_result,
38                                      const std::string& id) {
39  EXPECT_EQ(expected_result,
40            DatabaseUtil::IsValidOriginIdentifier(id));
41}
42
43namespace content {
44
45// Test DatabaseUtil::CrackVfsFilePath on various inputs.
46TEST(DatabaseUtilTest, CrackVfsFilePathTest) {
47  TestVfsFilePath(true, "http_origin_0/#", "http_origin_0", "", "");
48  TestVfsFilePath(true,
49      "http_origin_0/#suffix", "http_origin_0", "", "suffix");
50  TestVfsFilePath(true,
51      "http_origin_0/db_name#", "http_origin_0", "db_name", "");
52  TestVfsFilePath(true,
53      "http_origin_0/db_name#suffix", "http_origin_0", "db_name", "suffix");
54  TestVfsFilePath(false, "http_origin_0db_name#");
55  TestVfsFilePath(false, "http_origin_0db_name#suffix");
56  TestVfsFilePath(false, "http_origin_0/db_name");
57  TestVfsFilePath(false, "http_origin_0#db_name/suffix");
58  TestVfsFilePath(false, "/db_name#");
59  TestVfsFilePath(false, "/db_name#suffix");
60}
61
62TEST(DatabaseUtilTest, OriginIdentifiers) {
63  const GURL kFileOrigin(GURL("file:///").GetOrigin());
64  const GURL kHttpOrigin(GURL("http://bar/").GetOrigin());
65  EXPECT_EQ(kFileOrigin, ToAndFromOriginIdentifier(kFileOrigin));
66  EXPECT_EQ(kHttpOrigin, ToAndFromOriginIdentifier(kHttpOrigin));
67}
68
69TEST(DatabaseUtilTest, IsValidOriginIdentifier) {
70  TestValidOriginIdentifier(true,  "http_bar_0");
71  TestValidOriginIdentifier(false,  "");
72  TestValidOriginIdentifier(false, "bad..id");
73  TestValidOriginIdentifier(false, "bad/id");
74  TestValidOriginIdentifier(false, "bad\\id");
75  TestValidOriginIdentifier(false, "http_bad:0_2");
76  TestValidOriginIdentifier(false, std::string("bad\0id", 6));
77}
78
79}  // namespace content
80