diff options
author | sra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2009-05-08 23:00:29 +0000 |
---|---|---|
committer | sra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2009-05-08 23:00:29 +0000 |
commit | 04ca1bc65afb76ea30698c25f24599d20119e3d2 (patch) | |
tree | 2bb65e974d478f4d607b83f6a84a56c01f501ac0 /courgette/encode_decode_unittest.cc | |
parent | 9adf1dcf3281408560267787eddcc767566b425f (diff) | |
download | chromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.zip chromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.tar.gz chromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.tar.bz2 |
Move Courgette
from src\third_party\courgette
to src\courgette and src\courgette\third_party
Fixed #includes
Added properties to ignore generated files:
C:\c5\src>svn pg svn:ignore courgette
courgette.xcodeproj
courgette.sln
courgette_fuzz.vcproj
courgette_lib.vcproj
courgette_minimal_tool.vcproj
courgette_tool.vcproj
courgette.vcproj
courgette_unittests.vcproj
SConstruct
courgette_fuzz.scons
courgette_lib.scons
courgette_main.scons
courgette_minimal_tool.scons
courgette.scons
courgette_tool.scons
courgette_unittests.scons
Review URL: http://codereview.chromium.org/115062
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@15692 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'courgette/encode_decode_unittest.cc')
-rw-r--r-- | courgette/encode_decode_unittest.cc | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/courgette/encode_decode_unittest.cc b/courgette/encode_decode_unittest.cc new file mode 100644 index 0000000..436479d8 --- /dev/null +++ b/courgette/encode_decode_unittest.cc @@ -0,0 +1,105 @@ +// Copyright (c) 2009 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#include <string> + +#include "base/path_service.h" +#include "base/file_util.h" +#include "base/string_util.h" + +#include "courgette/courgette.h" +#include "courgette/streams.h" + +#include "testing/gtest/include/gtest/gtest.h" + +class EncodeDecodeTest : public testing::Test { + public: + void TestExe(const char *) const; + + private: + void SetUp() { + PathService::Get(base::DIR_SOURCE_ROOT, &testdata_dir_); + file_util::AppendToPath(&testdata_dir_, L"courgette"); + file_util::AppendToPath(&testdata_dir_, L"testdata"); + } + + void TearDown() { } + + // Returns contents of |file_name| as uninterprested bytes stored in a string. + std::string FileContents(const char* file_name) const; + + std::wstring testdata_dir_; // Full path name of testdata directory +}; + +// Reads a test file into a string. +std::string EncodeDecodeTest::FileContents(const char* file_name) const { + std::wstring file_path = testdata_dir_; + file_util::AppendToPath(&file_path, UTF8ToWide(file_name)); + std::string file_contents; + if (!file_util::ReadFileToString(file_path, &file_contents)) { + EXPECT_TRUE(!"Could not read test data"); + } + return file_contents; +} + +void EncodeDecodeTest::TestExe(const char* file_name) const { + // Test top-level Courgette API for converting an a file to a binary + // assembly representation and back. + std::string file1 = FileContents(file_name); + + const void* original_buffer = file1.c_str(); + size_t original_length = file1.size(); + + courgette::AssemblyProgram* program = NULL; + const courgette::Status parse_status = + courgette::ParseWin32X86PE(original_buffer, original_length, &program); + EXPECT_EQ(courgette::C_OK, parse_status); + + courgette::EncodedProgram* encoded = NULL; + + const courgette::Status encode_status = Encode(program, &encoded); + EXPECT_EQ(courgette::C_OK, encode_status); + + DeleteAssemblyProgram(program); + + courgette::SinkStreamSet sinks; + const courgette::Status write_status = WriteEncodedProgram(encoded, &sinks); + EXPECT_EQ(courgette::C_OK, write_status); + + DeleteEncodedProgram(encoded); + + courgette::SinkStream sink; + bool can_collect = sinks.CopyTo(&sink); + EXPECT_TRUE(can_collect); + + const void* buffer = sink.Buffer(); + size_t length = sink.Length(); + + EXPECT_EQ(971850, length); + + courgette::SourceStreamSet sources; + bool can_get_source_streams = sources.Init(buffer, length); + EXPECT_TRUE(can_get_source_streams); + + courgette::EncodedProgram *encoded2 = NULL; + const courgette::Status read_status = ReadEncodedProgram(&sources, &encoded2); + EXPECT_EQ(courgette::C_OK, read_status); + + courgette::SinkStream assembled; + const courgette::Status assemble_status = Assemble(encoded2, &assembled); + EXPECT_EQ(courgette::C_OK, assemble_status); + + const void* assembled_buffer = assembled.Buffer(); + size_t assembled_length = assembled.Length(); + + EXPECT_EQ(original_length, assembled_length); + EXPECT_EQ(0, memcmp(original_buffer, assembled_buffer, original_length)); + + DeleteEncodedProgram(encoded2); +} + + +TEST_F(EncodeDecodeTest, All) { + TestExe("setup1.exe"); +} |