Index: third_party/courgette/encode_decode_unittest.cc |
=================================================================== |
--- third_party/courgette/encode_decode_unittest.cc (revision 15668) |
+++ third_party/courgette/encode_decode_unittest.cc (working copy) |
@@ -1,106 +0,0 @@ |
-// 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 "third_party/courgette/courgette.h" |
-#include "third_party/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"third_party"); |
- 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"); |
-} |