| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <string> | 5 #include <string> |
| 6 #include <utility> | |
| 7 | 6 |
| 8 #include "base/memory/scoped_ptr.h" | |
| 9 #include "base/strings/string_util.h" | 7 #include "base/strings/string_util.h" |
| 8 |
| 10 #include "courgette/assembly_program.h" | 9 #include "courgette/assembly_program.h" |
| 11 #include "courgette/courgette.h" | 10 #include "courgette/courgette.h" |
| 12 #include "courgette/encoded_program.h" | |
| 13 #include "courgette/streams.h" | 11 #include "courgette/streams.h" |
| 14 | 12 |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 14 |
| 17 class AdjustmentMethodTest : public testing::Test { | 15 class AdjustmentMethodTest : public testing::Test { |
| 18 public: | 16 public: |
| 19 void Test1() const; | 17 void Test1() const; |
| 20 | 18 |
| 21 private: | 19 private: |
| 22 void SetUp() { | 20 void SetUp() { |
| 23 } | 21 } |
| 24 | 22 |
| 25 void TearDown() { | 23 void TearDown() { |
| 26 } | 24 } |
| 27 | 25 |
| 28 // Returns one of two similar simple programs. These differ only in Label | 26 // Returns one of two similar a simple programs. They differ only in the |
| 29 // assignment, so it is possible to make them look identical. | 27 // label assignment, so that it is possible to make them look identical. |
| 30 scoped_ptr<courgette::AssemblyProgram> MakeProgram(int kind) const { | 28 courgette::AssemblyProgram* MakeProgram(int kind) const { |
| 31 scoped_ptr<courgette::AssemblyProgram> prog( | 29 courgette::AssemblyProgram* prog = |
| 32 new courgette::AssemblyProgram(courgette::EXE_WIN_32_X86)); | 30 new courgette::AssemblyProgram(courgette::EXE_WIN_32_X86); |
| 33 prog->set_image_base(0x00400000); | 31 prog->set_image_base(0x00400000); |
| 34 | 32 |
| 35 courgette::Label* labelA = prog->FindOrMakeAbs32Label(0x00410000); | 33 courgette::Label* labelA = prog->FindOrMakeAbs32Label(0x00410000); |
| 36 courgette::Label* labelB = prog->FindOrMakeAbs32Label(0x00410004); | 34 courgette::Label* labelB = prog->FindOrMakeAbs32Label(0x00410004); |
| 37 | 35 |
| 38 EXPECT_TRUE(prog->EmitAbs32(labelA)); | 36 EXPECT_TRUE(prog->EmitAbs32(labelA)); |
| 39 EXPECT_TRUE(prog->EmitAbs32(labelA)); | 37 EXPECT_TRUE(prog->EmitAbs32(labelA)); |
| 40 EXPECT_TRUE(prog->EmitAbs32(labelB)); | 38 EXPECT_TRUE(prog->EmitAbs32(labelB)); |
| 41 EXPECT_TRUE(prog->EmitAbs32(labelA)); | 39 EXPECT_TRUE(prog->EmitAbs32(labelA)); |
| 42 EXPECT_TRUE(prog->EmitAbs32(labelA)); | 40 EXPECT_TRUE(prog->EmitAbs32(labelA)); |
| 43 EXPECT_TRUE(prog->EmitAbs32(labelB)); | 41 EXPECT_TRUE(prog->EmitAbs32(labelB)); |
| 44 | 42 |
| 45 if (kind == 0) { | 43 if (kind == 0) { |
| 46 labelA->index_ = 0; | 44 labelA->index_ = 0; |
| 47 labelB->index_ = 1; | 45 labelB->index_ = 1; |
| 48 } else { | 46 } else { |
| 49 labelA->index_ = 1; | 47 labelA->index_ = 1; |
| 50 labelB->index_ = 0; | 48 labelB->index_ = 0; |
| 51 } | 49 } |
| 52 prog->AssignRemainingIndexes(); | 50 prog->AssignRemainingIndexes(); |
| 53 | 51 |
| 54 return prog; | 52 return prog; |
| 55 } | 53 } |
| 56 | 54 |
| 57 scoped_ptr<courgette::AssemblyProgram> MakeProgramA() const { | 55 courgette::AssemblyProgram* MakeProgramA() const { return MakeProgram(0); } |
| 58 return MakeProgram(0); | 56 courgette::AssemblyProgram* MakeProgramB() const { return MakeProgram(1); } |
| 59 } | |
| 60 scoped_ptr<courgette::AssemblyProgram> MakeProgramB() const { | |
| 61 return MakeProgram(1); | |
| 62 } | |
| 63 | 57 |
| 64 // Returns a string that is the serialized version of |program|. | 58 // Returns a string that is the serialized version of |program|. |
| 65 // Deletes |program|. | 59 // Deletes |program|. |
| 66 std::string Serialize(scoped_ptr<courgette::AssemblyProgram> program) const { | 60 std::string Serialize(courgette::AssemblyProgram *program) const { |
| 67 scoped_ptr<courgette::EncodedProgram> encoded; | 61 courgette::EncodedProgram* encoded = NULL; |
| 68 | 62 |
| 69 const courgette::Status encode_status = Encode(*program, &encoded); | 63 const courgette::Status encode_status = Encode(program, &encoded); |
| 70 EXPECT_EQ(courgette::C_OK, encode_status); | 64 EXPECT_EQ(courgette::C_OK, encode_status); |
| 71 | 65 |
| 72 program.reset(); | 66 DeleteAssemblyProgram(program); |
| 73 | 67 |
| 74 courgette::SinkStreamSet sinks; | 68 courgette::SinkStreamSet sinks; |
| 75 const courgette::Status write_status = | 69 const courgette::Status write_status = WriteEncodedProgram(encoded, &sinks); |
| 76 WriteEncodedProgram(encoded.get(), &sinks); | |
| 77 EXPECT_EQ(courgette::C_OK, write_status); | 70 EXPECT_EQ(courgette::C_OK, write_status); |
| 78 | 71 |
| 79 encoded.reset(); | 72 DeleteEncodedProgram(encoded); |
| 80 | 73 |
| 81 courgette::SinkStream sink; | 74 courgette::SinkStream sink; |
| 82 bool can_collect = sinks.CopyTo(&sink); | 75 bool can_collect = sinks.CopyTo(&sink); |
| 83 EXPECT_TRUE(can_collect); | 76 EXPECT_TRUE(can_collect); |
| 84 | 77 |
| 85 return std::string(reinterpret_cast<const char *>(sink.Buffer()), | 78 return std::string(reinterpret_cast<const char *>(sink.Buffer()), |
| 86 sink.Length()); | 79 sink.Length()); |
| 87 } | 80 } |
| 88 }; | 81 }; |
| 89 | 82 |
| 90 | 83 |
| 91 void AdjustmentMethodTest::Test1() const { | 84 void AdjustmentMethodTest::Test1() const { |
| 92 scoped_ptr<courgette::AssemblyProgram> prog1 = MakeProgramA(); | 85 courgette::AssemblyProgram* prog1 = MakeProgramA(); |
| 93 scoped_ptr<courgette::AssemblyProgram> prog2 = MakeProgramB(); | 86 courgette::AssemblyProgram* prog2 = MakeProgramB(); |
| 94 std::string s1 = Serialize(std::move(prog1)); | 87 std::string s1 = Serialize(prog1); |
| 95 std::string s2 = Serialize(std::move(prog2)); | 88 std::string s2 = Serialize(prog2); |
| 96 | 89 |
| 97 // Don't use EXPECT_EQ because strings are unprintable. | 90 // Don't use EXPECT_EQ because strings are unprintable. |
| 98 EXPECT_FALSE(s1 == s2); // Unadjusted A and B differ. | 91 EXPECT_FALSE(s1 == s2); // Unadjusted A and B differ. |
| 99 | 92 |
| 100 scoped_ptr<courgette::AssemblyProgram> prog5 = MakeProgramA(); | 93 courgette::AssemblyProgram* prog5 = MakeProgramA(); |
| 101 scoped_ptr<courgette::AssemblyProgram> prog6 = MakeProgramB(); | 94 courgette::AssemblyProgram* prog6 = MakeProgramB(); |
| 102 courgette::Status can_adjust = Adjust(*prog5, prog6.get()); | 95 courgette::Status can_adjust = Adjust(*prog5, prog6); |
| 103 EXPECT_EQ(courgette::C_OK, can_adjust); | 96 EXPECT_EQ(courgette::C_OK, can_adjust); |
| 104 std::string s5 = Serialize(std::move(prog5)); | 97 std::string s5 = Serialize(prog5); |
| 105 std::string s6 = Serialize(std::move(prog6)); | 98 std::string s6 = Serialize(prog6); |
| 106 | 99 |
| 107 EXPECT_TRUE(s1 == s5); // Adjustment did not change A (prog5) | 100 EXPECT_TRUE(s1 == s5); // Adjustment did not change A (prog5) |
| 108 EXPECT_TRUE(s5 == s6); // Adjustment did change B into A | 101 EXPECT_TRUE(s5 == s6); // Adjustment did change B into A |
| 109 } | 102 } |
| 110 | 103 |
| 111 | 104 |
| 112 TEST_F(AdjustmentMethodTest, All) { | 105 TEST_F(AdjustmentMethodTest, All) { |
| 113 Test1(); | 106 Test1(); |
| 114 } | 107 } |
| OLD | NEW |