blob: 5fa0ccb14318e4cdcedc15dab45d66b109926031 [file] [log] [blame]
Brian Carlstromcd60ac72013-01-20 17:09:51 -08001/*
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
Brian Carlstromcd60ac72013-01-20 17:09:51 -080017#include "file_output_stream.h"
18#include "vector_output_stream.h"
19
Brian Carlstroma1ce1fe2014-02-24 23:23:58 -080020#include "base/logging.h"
21#include "buffered_output_stream.h"
22#include "common_runtime_test.h"
23
Brian Carlstromcd60ac72013-01-20 17:09:51 -080024namespace art {
25
Brian Carlstroma1ce1fe2014-02-24 23:23:58 -080026class OutputStreamTest : public CommonRuntimeTest {
Brian Carlstromcd60ac72013-01-20 17:09:51 -080027 protected:
28 void CheckOffset(off_t expected) {
Brian Carlstrom49a0f152013-01-22 17:17:36 -080029 off_t actual = output_stream_->Seek(0, kSeekCurrent);
Ian Rogers8d3a1172013-06-04 01:13:28 -070030 EXPECT_EQ(expected, actual);
Brian Carlstromcd60ac72013-01-20 17:09:51 -080031 }
32
33 void SetOutputStream(OutputStream& output_stream) {
34 output_stream_ = &output_stream;
35 }
36
37 void GenerateTestOutput() {
Ian Rogers8d3a1172013-06-04 01:13:28 -070038 EXPECT_EQ(3, output_stream_->Seek(3, kSeekCurrent));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080039 CheckOffset(3);
Ian Rogers8d3a1172013-06-04 01:13:28 -070040 EXPECT_EQ(2, output_stream_->Seek(2, kSeekSet));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080041 CheckOffset(2);
42 uint8_t buf[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
Ian Rogers8d3a1172013-06-04 01:13:28 -070043 EXPECT_TRUE(output_stream_->WriteFully(buf, 2));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080044 CheckOffset(4);
Ian Rogers8d3a1172013-06-04 01:13:28 -070045 EXPECT_EQ(6, output_stream_->Seek(2, kSeekEnd));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080046 CheckOffset(6);
Ian Rogers8d3a1172013-06-04 01:13:28 -070047 EXPECT_TRUE(output_stream_->WriteFully(buf, 4));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080048 CheckOffset(10);
49 }
50
51 void CheckTestOutput(const std::vector<uint8_t>& actual) {
52 uint8_t expected[] = {
53 0, 0, 1, 2, 0, 0, 1, 2, 3, 4
54 };
Ian Rogers8d3a1172013-06-04 01:13:28 -070055 EXPECT_EQ(sizeof(expected), actual.size());
56 EXPECT_EQ(0, memcmp(expected, &actual[0], actual.size()));
Brian Carlstromcd60ac72013-01-20 17:09:51 -080057 }
58
59 OutputStream* output_stream_;
60};
61
62TEST_F(OutputStreamTest, File) {
63 ScratchFile tmp;
64 FileOutputStream output_stream(tmp.GetFile());
65 SetOutputStream(output_stream);
66 GenerateTestOutput();
Ian Rogers700a4022014-05-19 16:49:03 -070067 std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
Ian Rogers8d3a1172013-06-04 01:13:28 -070068 EXPECT_TRUE(in.get() != NULL);
Brian Carlstromcd60ac72013-01-20 17:09:51 -080069 std::vector<uint8_t> actual(in->GetLength());
70 bool readSuccess = in->ReadFully(&actual[0], actual.size());
Ian Rogers8d3a1172013-06-04 01:13:28 -070071 EXPECT_TRUE(readSuccess);
Brian Carlstromcd60ac72013-01-20 17:09:51 -080072 CheckTestOutput(actual);
73}
74
Brian Carlstromc6dfdac2013-08-26 18:57:31 -070075TEST_F(OutputStreamTest, Buffered) {
76 ScratchFile tmp;
Ian Rogers700a4022014-05-19 16:49:03 -070077 std::unique_ptr<FileOutputStream> file_output_stream(new FileOutputStream(tmp.GetFile()));
Brian Carlstromc6dfdac2013-08-26 18:57:31 -070078 CHECK(file_output_stream.get() != NULL);
79 BufferedOutputStream buffered_output_stream(file_output_stream.release());
80 SetOutputStream(buffered_output_stream);
81 GenerateTestOutput();
Ian Rogers700a4022014-05-19 16:49:03 -070082 std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
Brian Carlstromc6dfdac2013-08-26 18:57:31 -070083 EXPECT_TRUE(in.get() != NULL);
84 std::vector<uint8_t> actual(in->GetLength());
85 bool readSuccess = in->ReadFully(&actual[0], actual.size());
86 EXPECT_TRUE(readSuccess);
87 CheckTestOutput(actual);
88}
89
Brian Carlstromcd60ac72013-01-20 17:09:51 -080090TEST_F(OutputStreamTest, Vector) {
91 std::vector<uint8_t> output;
92 VectorOutputStream output_stream("test vector output", output);
93 SetOutputStream(output_stream);
94 GenerateTestOutput();
95 CheckTestOutput(output);
96}
97
Mathieu Chartier02e25112013-08-14 16:14:24 -070098} // namespace art