summaryrefslogtreecommitdiffstats
path: root/courgette/encode_decode_unittest.cc
diff options
context:
space:
mode:
authorsra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-05-08 23:00:29 +0000
committersra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-05-08 23:00:29 +0000
commit04ca1bc65afb76ea30698c25f24599d20119e3d2 (patch)
tree2bb65e974d478f4d607b83f6a84a56c01f501ac0 /courgette/encode_decode_unittest.cc
parent9adf1dcf3281408560267787eddcc767566b425f (diff)
downloadchromium_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.cc105
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");
+}