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