| /* |
| * Copyright (C) 2013 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "file_output_stream.h" |
| #include "vector_output_stream.h" |
| |
| #include "base/logging.h" |
| #include "base/unix_file/fd_file.h" |
| #include "buffered_output_stream.h" |
| #include "common_runtime_test.h" |
| |
| namespace art { |
| |
| class OutputStreamTest : public CommonRuntimeTest { |
| protected: |
| void CheckOffset(off_t expected) { |
| off_t actual = output_stream_->Seek(0, kSeekCurrent); |
| EXPECT_EQ(expected, actual); |
| } |
| |
| void SetOutputStream(OutputStream& output_stream) { |
| output_stream_ = &output_stream; |
| } |
| |
| void GenerateTestOutput() { |
| EXPECT_EQ(3, output_stream_->Seek(3, kSeekCurrent)); |
| CheckOffset(3); |
| EXPECT_EQ(2, output_stream_->Seek(2, kSeekSet)); |
| CheckOffset(2); |
| uint8_t buf[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 }; |
| EXPECT_TRUE(output_stream_->WriteFully(buf, 2)); |
| CheckOffset(4); |
| EXPECT_EQ(6, output_stream_->Seek(2, kSeekEnd)); |
| CheckOffset(6); |
| EXPECT_TRUE(output_stream_->WriteFully(buf, 4)); |
| CheckOffset(10); |
| EXPECT_TRUE(output_stream_->WriteFully(buf, 6)); |
| EXPECT_TRUE(output_stream_->Flush()); |
| } |
| |
| void CheckTestOutput(const std::vector<uint8_t>& actual) { |
| uint8_t expected[] = { |
| 0, 0, 1, 2, 0, 0, 1, 2, 3, 4, 1, 2, 3, 4, 5, 6 |
| }; |
| EXPECT_EQ(sizeof(expected), actual.size()); |
| EXPECT_EQ(0, memcmp(expected, &actual[0], actual.size())); |
| } |
| |
| OutputStream* output_stream_; |
| }; |
| |
| TEST_F(OutputStreamTest, File) { |
| ScratchFile tmp; |
| FileOutputStream output_stream(tmp.GetFile()); |
| SetOutputStream(output_stream); |
| GenerateTestOutput(); |
| std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str())); |
| EXPECT_TRUE(in.get() != nullptr); |
| std::vector<uint8_t> actual(in->GetLength()); |
| bool readSuccess = in->ReadFully(&actual[0], actual.size()); |
| EXPECT_TRUE(readSuccess); |
| CheckTestOutput(actual); |
| } |
| |
| TEST_F(OutputStreamTest, Buffered) { |
| ScratchFile tmp; |
| { |
| BufferedOutputStream buffered_output_stream(std::make_unique<FileOutputStream>(tmp.GetFile())); |
| SetOutputStream(buffered_output_stream); |
| GenerateTestOutput(); |
| } |
| std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str())); |
| EXPECT_TRUE(in.get() != nullptr); |
| std::vector<uint8_t> actual(in->GetLength()); |
| bool readSuccess = in->ReadFully(&actual[0], actual.size()); |
| EXPECT_TRUE(readSuccess); |
| CheckTestOutput(actual); |
| } |
| |
| TEST_F(OutputStreamTest, Vector) { |
| std::vector<uint8_t> output; |
| VectorOutputStream output_stream("test vector output", &output); |
| SetOutputStream(output_stream); |
| GenerateTestOutput(); |
| CheckTestOutput(output); |
| } |
| |
| TEST_F(OutputStreamTest, BufferedFlush) { |
| struct CheckingOutputStream : OutputStream { |
| CheckingOutputStream() |
| : OutputStream("dummy"), |
| flush_called(false) { } |
| ~CheckingOutputStream() OVERRIDE {} |
| |
| bool WriteFully(const void* buffer ATTRIBUTE_UNUSED, |
| size_t byte_count ATTRIBUTE_UNUSED) OVERRIDE { |
| LOG(FATAL) << "UNREACHABLE"; |
| UNREACHABLE(); |
| } |
| |
| off_t Seek(off_t offset ATTRIBUTE_UNUSED, Whence whence ATTRIBUTE_UNUSED) OVERRIDE { |
| LOG(FATAL) << "UNREACHABLE"; |
| UNREACHABLE(); |
| } |
| |
| bool Flush() OVERRIDE { |
| flush_called = true; |
| return true; |
| } |
| |
| bool flush_called; |
| }; |
| |
| std::unique_ptr<CheckingOutputStream> cos = std::make_unique<CheckingOutputStream>(); |
| CheckingOutputStream* checking_output_stream = cos.get(); |
| BufferedOutputStream buffered(std::move(cos)); |
| ASSERT_FALSE(checking_output_stream->flush_called); |
| bool flush_result = buffered.Flush(); |
| ASSERT_TRUE(flush_result); |
| ASSERT_TRUE(checking_output_stream->flush_called); |
| } |
| |
| } // namespace art |