src/content/browser/database_util_unittest.cc

   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
  11 using base::ASCIIToUTF16;
  12 using storage::DatabaseUtil;
  13
  14 static 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
  32 static GURL ToAndFromOriginIdentifier(const GURL origin_url) {
  33   std::string id = storage::GetIdentifierFromOrigin(origin_url);
  34   return storage::GetOriginFromIdentifier(id);
  35 }
  36
  37 static void TestValidOriginIdentifier(bool expected_result,
  38                                       const std::string& id) {
  39   EXPECT_EQ(expected_result,
  40             DatabaseUtil::IsValidOriginIdentifier(id));
  41 }
  42
  43 namespace content {
  44
  45 // Test DatabaseUtil::CrackVfsFilePath on various inputs.
  46 TEST(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
  62 TEST(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
  69 TEST(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