OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 <string> | |
6 | |
7 #include "base/file_path.h" | |
8 #include "base/files/scoped_temp_dir.h" | |
9 #include "chrome/browser/media_gallery/media_gallery_database.h" | |
10 #include "chrome/browser/media_gallery/media_gallery_database_types.h" | |
11 #include "sql/connection.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 namespace chrome { | |
15 | |
16 class MediaGalleryDatabaseTest : public testing::Test, | |
17 public MediaGalleryDatabase { | |
18 public: | |
19 MediaGalleryDatabaseTest() { } | |
20 | |
21 protected: | |
22 virtual sql::Connection& GetDB() OVERRIDE { | |
23 return db_; | |
24 } | |
25 | |
26 private: | |
27 // Test setup. | |
28 virtual void SetUp() { | |
29 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
30 base::FilePath db_file = | |
31 temp_dir_.path().AppendASCII("MediaGalleryTest.db"); | |
32 | |
33 ASSERT_TRUE(db_.Open(db_file)); | |
34 | |
35 // Initialize the tables for this test. | |
36 ASSERT_EQ(sql::INIT_OK, InitInternal(&db_)); | |
37 } | |
38 | |
39 virtual void TearDown() { | |
40 db_.Close(); | |
41 } | |
42 | |
43 base::ScopedTempDir temp_dir_; | |
44 sql::Connection db_; | |
45 }; | |
46 | |
47 TEST_F(MediaGalleryDatabaseTest, Init) { | |
48 EXPECT_TRUE(DoesCollectionsTableExist(&GetDB())); | |
49 } | |
50 | |
51 TEST_F(MediaGalleryDatabaseTest, AddCollection) { | |
52 CollectionRow row1(base::FilePath(FILE_PATH_LITERAL("path1")), | |
53 base::Time::FromDoubleT(12345), | |
54 123, | |
55 false); | |
56 CollectionId rowid = CreateCollectionRow(&row1); | |
57 EXPECT_EQ(rowid, row1.id); | |
58 | |
59 CollectionRow row2; | |
60 EXPECT_TRUE(GetCollectionRow(rowid, &row2)); | |
61 EXPECT_TRUE(row1 == row2); | |
62 } | |
63 | |
64 TEST_F(MediaGalleryDatabaseTest, StandardizePath) { | |
65 base::FilePath path(FILE_PATH_LITERAL("path1")); | |
66 path = path.AppendASCII("path2"); | |
67 CollectionRow row1(path, | |
68 base::Time::FromDoubleT(12345), | |
69 123, | |
70 false); | |
71 CollectionId rowid = CreateCollectionRow(&row1); | |
72 EXPECT_EQ(rowid, row1.id); | |
73 | |
74 CollectionRow row2; | |
75 EXPECT_TRUE(GetCollectionRow(rowid, &row2)); | |
76 EXPECT_EQ(FILE_PATH_LITERAL("path1/path2"), row2.path.value()); | |
77 } | |
78 | |
79 } // namespace chrome | |
OLD | NEW |