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: 27 void CheckOffset(off_t expected) { 28 off_t actual = output_stream_->Seek(0, kSeekCurrent); 29 EXPECT_EQ(expected, actual); 30 } 31 32 void SetOutputStream(OutputStream& output_stream) { 33 output_stream_ = &output_stream; 34 } 35 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 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 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 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 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
