Files.app: Dispatch 'drive-connection-changed' event on initialization of VolumeManag...
[chromium-blink-merge.git] / extensions / browser / computed_hashes_unittest.cc
blobaecbd650d537f9fdcf26909b88ef054182939315
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.
5 #include "base/base64.h"
6 #include "base/files/file_path.h"
7 #include "base/files/scoped_temp_dir.h"
8 #include "crypto/sha2.h"
9 #include "extensions/browser/computed_hashes.h"
10 #include "testing/gtest/include/gtest/gtest.h"
12 namespace {
14 // Helper to return base64 encode result by value.
15 std::string Base64Encode(const std::string& data) {
16 std::string result;
17 base::Base64Encode(data, &result);
18 return result;
21 } // namespace
23 namespace extensions {
25 TEST(ComputedHashes, ComputedHashes) {
26 base::ScopedTempDir scoped_dir;
27 ASSERT_TRUE(scoped_dir.CreateUniqueTempDir());
28 base::FilePath computed_hashes =
29 scoped_dir.path().AppendASCII("computed_hashes.json");
31 // We'll add hashes for 2 files, one of which uses a subdirectory
32 // path. The first file will have a list of 1 block hash, and the
33 // second file will have 2 block hashes.
34 base::FilePath path1(FILE_PATH_LITERAL("foo.txt"));
35 base::FilePath path2 =
36 base::FilePath(FILE_PATH_LITERAL("foo")).AppendASCII("bar.txt");
37 std::vector<std::string> hashes1;
38 std::vector<std::string> hashes2;
39 hashes1.push_back(crypto::SHA256HashString("first"));
40 hashes2.push_back(crypto::SHA256HashString("second"));
41 hashes2.push_back(crypto::SHA256HashString("third"));
43 // Write them into the file.
44 ComputedHashes::Writer writer;
45 writer.AddHashes(path1, 4096, hashes1);
46 writer.AddHashes(path2, 2048, hashes2);
47 EXPECT_TRUE(writer.WriteToFile(computed_hashes));
49 // Now read them back again and assert that we got what we wrote.
50 ComputedHashes::Reader reader;
51 std::vector<std::string> read_hashes1;
52 std::vector<std::string> read_hashes2;
53 EXPECT_TRUE(reader.InitFromFile(computed_hashes));
55 int block_size = 0;
56 EXPECT_TRUE(reader.GetHashes(path1, &block_size, &read_hashes1));
57 EXPECT_EQ(block_size, 4096);
58 block_size = 0;
59 EXPECT_TRUE(reader.GetHashes(path2, &block_size, &read_hashes2));
60 EXPECT_EQ(block_size, 2048);
62 EXPECT_EQ(hashes1, read_hashes1);
63 EXPECT_EQ(hashes2, read_hashes2);
65 // Make sure we can lookup hashes for a file using incorrect case
66 base::FilePath path1_badcase(FILE_PATH_LITERAL("FoO.txt"));
67 std::vector<std::string> read_hashes1_badcase;
68 EXPECT_TRUE(
69 reader.GetHashes(path1_badcase, &block_size, &read_hashes1_badcase));
70 EXPECT_EQ(block_size, 4096);
71 EXPECT_EQ(hashes1, read_hashes1_badcase);
73 // Finally make sure that we can retrieve the hashes for the subdir
74 // path even when that path contains forward slashes (on windows).
75 base::FilePath path2_fwd_slashes =
76 base::FilePath::FromUTF8Unsafe("foo/bar.txt");
77 block_size = 0;
78 EXPECT_TRUE(reader.GetHashes(path2_fwd_slashes, &block_size, &read_hashes2));
79 EXPECT_EQ(hashes2, read_hashes2);
82 // Note: the expected hashes used in this test were generated using linux
83 // command line tools. E.g., from a bash prompt:
84 // $ printf "hello world" | openssl dgst -sha256 -binary | base64
86 // The file with multiple-blocks expectations were generated by doing:
87 // $ for i in `seq 500 ; do printf "hello world" ; done > hello.txt
88 // $ dd if=hello.txt bs=4096 count=1 | openssl dgst -sha256 -binary | base64
89 // $ dd if=hello.txt skip=1 bs=4096 count=1 |
90 // openssl dgst -sha256 -binary | base64
91 TEST(ComputedHashes, ComputeHashesForContent) {
92 const int block_size = 4096;
94 // Simple short input.
95 std::string content1 = "hello world";
96 std::string content1_expected_hash =
97 "uU0nuZNNPgilLlLX2n2r+sSE7+N6U4DukIj3rOLvzek=";
98 std::vector<std::string> hashes1;
99 ComputedHashes::ComputeHashesForContent(content1, block_size, &hashes1);
100 ASSERT_EQ(1u, hashes1.size());
101 EXPECT_EQ(content1_expected_hash, Base64Encode(hashes1[0]));
103 // Multiple blocks input.
104 std::string content2;
105 for (int i = 0; i < 500; i++)
106 content2 += "hello world";
107 const char* content2_expected_hashes[] = {
108 "bvtt5hXo8xvHrlzGAhhoqPL/r+4zJXHx+6wAvkv15V8=",
109 "lTD45F7P6I/HOdi8u7FLRA4qzAYL+7xSNVeusG6MJI0="};
110 std::vector<std::string> hashes2;
111 ComputedHashes::ComputeHashesForContent(content2, block_size, &hashes2);
112 ASSERT_EQ(2u, hashes2.size());
113 EXPECT_EQ(content2_expected_hashes[0], Base64Encode(hashes2[0]));
114 EXPECT_EQ(content2_expected_hashes[1], Base64Encode(hashes2[1]));
116 // Now an empty input.
117 std::string content3;
118 std::vector<std::string> hashes3;
119 ComputedHashes::ComputeHashesForContent(content3, block_size, &hashes3);
120 ASSERT_EQ(1u, hashes3.size());
121 ASSERT_EQ(std::string("47DEQpj8HBSa+/TImW+5JCeuQeRkm5NMpJWZG3hSuFU="),
122 Base64Encode(hashes3[0]));
125 } // namespace extensions