webkit/database/database_util_unittest.cc

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