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 "testing/gtest/include/gtest/gtest.h" 8 #include "webkit/browser/database/database_util.h" 9 #include "webkit/common/database/database_identifier.h" 10 11 using base::ASCIIToUTF16; 12 using webkit_database::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 = webkit_database::GetIdentifierFromOrigin(origin_url); 34 return webkit_database::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 80
