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 |