1 /*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "base/logging.h"
18 #include "buffered_output_stream.h"
19 #include "common_test.h"
20 #include "file_output_stream.h"
21 #include "vector_output_stream.h"
22
23 namespace art {
24
25 class OutputStreamTest : public CommonTest {
26 protected:
CheckOffset(off_t expected)27 void CheckOffset(off_t expected) {
28 off_t actual = output_stream_->Seek(0, kSeekCurrent);
29 EXPECT_EQ(expected, actual);
30 }
31
SetOutputStream(OutputStream & output_stream)32 void SetOutputStream(OutputStream& output_stream) {
33 output_stream_ = &output_stream;
34 }
35
GenerateTestOutput()36 void GenerateTestOutput() {
37 EXPECT_EQ(3, output_stream_->Seek(3, kSeekCurrent));
38 CheckOffset(3);
39 EXPECT_EQ(2, output_stream_->Seek(2, kSeekSet));
40 CheckOffset(2);
41 uint8_t buf[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
42 EXPECT_TRUE(output_stream_->WriteFully(buf, 2));
43 CheckOffset(4);
44 EXPECT_EQ(6, output_stream_->Seek(2, kSeekEnd));
45 CheckOffset(6);
46 EXPECT_TRUE(output_stream_->WriteFully(buf, 4));
47 CheckOffset(10);
48 }
49
CheckTestOutput(const std::vector<uint8_t> & actual)50 void CheckTestOutput(const std::vector<uint8_t>& actual) {
51 uint8_t expected[] = {
52 0, 0, 1, 2, 0, 0, 1, 2, 3, 4
53 };
54 EXPECT_EQ(sizeof(expected), actual.size());
55 EXPECT_EQ(0, memcmp(expected, &actual[0], actual.size()));
56 }
57
58 OutputStream* output_stream_;
59 };
60
TEST_F(OutputStreamTest,File)61 TEST_F(OutputStreamTest, File) {
62 ScratchFile tmp;
63 FileOutputStream output_stream(tmp.GetFile());
64 SetOutputStream(output_stream);
65 GenerateTestOutput();
66 UniquePtr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
67 EXPECT_TRUE(in.get() != NULL);
68 std::vector<uint8_t> actual(in->GetLength());
69 bool readSuccess = in->ReadFully(&actual[0], actual.size());
70 EXPECT_TRUE(readSuccess);
71 CheckTestOutput(actual);
72 }
73
TEST_F(OutputStreamTest,Buffered)74 TEST_F(OutputStreamTest, Buffered) {
75 ScratchFile tmp;
76 UniquePtr<FileOutputStream> file_output_stream(new FileOutputStream(tmp.GetFile()));
77 CHECK(file_output_stream.get() != NULL);
78 BufferedOutputStream buffered_output_stream(file_output_stream.release());
79 SetOutputStream(buffered_output_stream);
80 GenerateTestOutput();
81 UniquePtr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
82 EXPECT_TRUE(in.get() != NULL);
83 std::vector<uint8_t> actual(in->GetLength());
84 bool readSuccess = in->ReadFully(&actual[0], actual.size());
85 EXPECT_TRUE(readSuccess);
86 CheckTestOutput(actual);
87 }
88
TEST_F(OutputStreamTest,Vector)89 TEST_F(OutputStreamTest, Vector) {
90 std::vector<uint8_t> output;
91 VectorOutputStream output_stream("test vector output", output);
92 SetOutputStream(output_stream);
93 GenerateTestOutput();
94 CheckTestOutput(output);
95 }
96
97 } // namespace art
98