| Index: webrtc/base/file_unittest.cc
|
| diff --git a/webrtc/base/file_unittest.cc b/webrtc/base/file_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..458b6da75f2642dfead3d72786579903ca57e84a
|
| --- /dev/null
|
| +++ b/webrtc/base/file_unittest.cc
|
| @@ -0,0 +1,166 @@
|
| +/*
|
| + * Copyright 2016 The WebRTC Project Authors. All rights reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#include <sys/types.h>
|
| +#include <sys/stat.h>
|
| +#include <fcntl.h>
|
| +
|
| +#include <limits>
|
| +#include <memory>
|
| +#include <string>
|
| +
|
| +#include "webrtc/base/gunit.h"
|
| +#include "webrtc/base/file.h"
|
| +#include "webrtc/test/testsupport/fileutils.h"
|
| +
|
| +#if defined(WEBRTC_WIN)
|
| +
|
| +#include "webrtc/base/win32.h"
|
| +
|
| +#else // if defined(WEBRTC_WIN)
|
| +
|
| +#include <errno.h>
|
| +
|
| +#endif
|
| +
|
| +namespace rtc {
|
| +
|
| +void RemoveFile(const std::string& path) {
|
| +#if defined(WEBRTC_WIN)
|
| + ::DeleteFile(ToUtf16(path).c_str());
|
| +#else
|
| + ::unlink(path.c_str());
|
| +#endif
|
| +}
|
| +
|
| +int LastError() {
|
| +#if defined(WEBRTC_WIN)
|
| + return ::GetLastError();
|
| +#else
|
| + return errno;
|
| +#endif
|
| +}
|
| +
|
| +bool VerifyBuffer(uint8_t* buffer, size_t length, uint8_t start_value) {
|
| + for (size_t i = 0; i < length; ++i) {
|
| + uint8_t val = start_value++;
|
| + EXPECT_EQ(val, buffer[i]);
|
| + if (buffer[i] != val)
|
| + return false;
|
| + }
|
| + // Prevent the same buffer from being verified multiple times simply
|
| + // because some operation that should have written to it failed
|
| + memset(buffer, 0, length);
|
| + return true;
|
| +}
|
| +
|
| +class FileTest : public ::testing::Test {
|
| + protected:
|
| + std::string path_;
|
| + void SetUp() {
|
| + path_ = webrtc::test::TempFilename(webrtc::test::OutputPath(), "test_file");
|
| + ASSERT_FALSE(path_.empty());
|
| + }
|
| + rtc::File OpenTempFile() { return rtc::File::Open(path_); }
|
| + void TearDown() { RemoveFile(path_); }
|
| +};
|
| +
|
| +TEST_F(FileTest, DoubleClose) {
|
| + File file = OpenTempFile();
|
| + ASSERT_TRUE(file.IsOpen()) << "Error: " << LastError();
|
| +
|
| + EXPECT_TRUE(file.Close());
|
| + EXPECT_FALSE(file.Close());
|
| +}
|
| +
|
| +TEST_F(FileTest, SimpleReadWrite) {
|
| + File file = OpenTempFile();
|
| + ASSERT_TRUE(file.IsOpen()) << "Error: " << LastError();
|
| +
|
| + uint8_t data[100] = {0};
|
| + uint8_t out[100] = {0};
|
| + for (int i = 0; i < 100; ++i) {
|
| + data[i] = i;
|
| + }
|
| +
|
| + EXPECT_EQ(10u, file.Write(data, 10));
|
| +
|
| + EXPECT_TRUE(file.Seek(0));
|
| + EXPECT_EQ(10u, file.Read(out, 10));
|
| + EXPECT_TRUE(VerifyBuffer(out, 10, 0));
|
| +
|
| + EXPECT_TRUE(file.Seek(0));
|
| + EXPECT_EQ(100u, file.Write(data, 100));
|
| +
|
| + EXPECT_TRUE(file.Seek(0));
|
| + EXPECT_EQ(100u, file.Read(out, 100));
|
| + EXPECT_TRUE(VerifyBuffer(out, 100, 0));
|
| +
|
| + EXPECT_TRUE(file.Seek(1));
|
| + EXPECT_EQ(50u, file.Write(data, 50));
|
| + EXPECT_EQ(50u, file.Write(data + 50, 50));
|
| +
|
| + EXPECT_TRUE(file.Seek(1));
|
| + EXPECT_EQ(100u, file.Read(out, 100));
|
| + EXPECT_TRUE(VerifyBuffer(out, 100, 0));
|
| +}
|
| +
|
| +TEST_F(FileTest, ReadWriteClose) {
|
| + File file = OpenTempFile();
|
| + ASSERT_TRUE(file.IsOpen()) << "Error: " << LastError();
|
| +
|
| + uint8_t data[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
|
| + uint8_t out[10] = {0};
|
| + EXPECT_EQ(10u, file.Write(data, 10));
|
| + EXPECT_TRUE(file.Close());
|
| +
|
| + File file2 = OpenTempFile();
|
| + ASSERT_TRUE(file2.IsOpen()) << "Error: " << LastError();
|
| + EXPECT_EQ(10u, file2.Read(out, 10));
|
| + EXPECT_TRUE(VerifyBuffer(out, 10, 0));
|
| +}
|
| +
|
| +TEST_F(FileTest, RandomAccessRead) {
|
| + File file = OpenTempFile();
|
| + ASSERT_TRUE(file.IsOpen()) << "Error: " << LastError();
|
| +
|
| + uint8_t data[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
|
| + uint8_t out[10] = {0};
|
| + EXPECT_EQ(10u, file.Write(data, 10));
|
| +
|
| + EXPECT_EQ(4u, file.ReadAt(out, 4, 0));
|
| + EXPECT_TRUE(VerifyBuffer(out, 4, 0));
|
| +
|
| + EXPECT_EQ(4u, file.ReadAt(out, 4, 4));
|
| + EXPECT_TRUE(VerifyBuffer(out, 4, 4));
|
| +
|
| + EXPECT_EQ(5u, file.ReadAt(out, 5, 5));
|
| + EXPECT_TRUE(VerifyBuffer(out, 5, 5));
|
| +}
|
| +
|
| +TEST_F(FileTest, RandomAccessReadWrite) {
|
| + File file = OpenTempFile();
|
| + ASSERT_TRUE(file.IsOpen()) << "Error: " << LastError();
|
| +
|
| + uint8_t data[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
|
| + uint8_t out[10] = {0};
|
| + EXPECT_EQ(10u, file.Write(data, 10));
|
| + EXPECT_TRUE(file.Seek(4));
|
| +
|
| + EXPECT_EQ(4u, file.WriteAt(data, 4, 4));
|
| + EXPECT_EQ(4u, file.ReadAt(out, 4, 4));
|
| + EXPECT_TRUE(VerifyBuffer(out, 4, 0));
|
| +
|
| + EXPECT_EQ(2u, file.WriteAt(data, 2, 8));
|
| + EXPECT_EQ(2u, file.ReadAt(out, 2, 8));
|
| + EXPECT_TRUE(VerifyBuffer(out, 2, 0));
|
| +}
|
| +
|
| +} // namespace rtc
|
|
|