Differential D580 Diff 1490 src/leveldb/util/env_posix_test.cc

Changeset View

Standalone View

src/leveldb/util/env_posix_test.cc

This file was added.

				// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
				// Use of this source code is governed by a BSD-style license that can be
				// found in the LICENSE file. See the AUTHORS file for names of contributors.

				#include "leveldb/env.h"

				#include "port/port.h"
				#include "util/testharness.h"
				#include "util/env_posix_test_helper.h"

				namespace leveldb {

				static const int kDelayMicros = 100000;
				static const int kReadOnlyFileLimit = 4;
				static const int kMMapLimit = 4;

				class EnvPosixTest {
				public:
				Env* env_;
				EnvPosixTest() : env_(Env::Default()) { }

				static void SetFileLimits(int read_only_file_limit, int mmap_limit) {
				EnvPosixTestHelper::SetReadOnlyFDLimit(read_only_file_limit);
				EnvPosixTestHelper::SetReadOnlyMMapLimit(mmap_limit);
				}
				};

				TEST(EnvPosixTest, TestOpenOnRead) {
				// Write some test data to a single file that will be opened \|n\| times.
				std::string test_dir;
				ASSERT_OK(env_->GetTestDirectory(&test_dir));
				std::string test_file = test_dir + "/open_on_read.txt";

				FILE* f = fopen(test_file.c_str(), "w");
				ASSERT_TRUE(f != NULL);
				const char kFileData[] = "abcdefghijklmnopqrstuvwxyz";
				fputs(kFileData, f);
				fclose(f);

				// Open test file some number above the sum of the two limits to force
				// open-on-read behavior of POSIX Env leveldb::RandomAccessFile.
				const int kNumFiles = kReadOnlyFileLimit + kMMapLimit + 5;
				leveldb::RandomAccessFile* files[kNumFiles] = {0};
				for (int i = 0; i < kNumFiles; i++) {
				ASSERT_OK(env_->NewRandomAccessFile(test_file, &files[i]));
				}
				char scratch;
				Slice read_result;
				for (int i = 0; i < kNumFiles; i++) {
				ASSERT_OK(files[i]->Read(i, 1, &read_result, &scratch));
				ASSERT_EQ(kFileData[i], read_result[0]);
				}
				for (int i = 0; i < kNumFiles; i++) {
				delete files[i];
				}
				ASSERT_OK(env_->DeleteFile(test_file));
				}

				} // namespace leveldb

				int main(int argc, char** argv) {
				// All tests currently run with the same read-only file limits.
				leveldb::EnvPosixTest::SetFileLimits(leveldb::kReadOnlyFileLimit,
				leveldb::kMMapLimit);
				return leveldb::test::RunAllTests();
				}