1/* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include "cha.h" 18 19#include "common_runtime_test.h" 20 21namespace art { 22 23class CHATest : public CommonRuntimeTest {}; 24 25// Mocks some methods. 26#define METHOD1 (reinterpret_cast<ArtMethod*>(8u)) 27#define METHOD2 (reinterpret_cast<ArtMethod*>(16u)) 28#define METHOD3 (reinterpret_cast<ArtMethod*>(24u)) 29 30// Mocks some method headers. 31#define METHOD_HEADER1 (reinterpret_cast<OatQuickMethodHeader*>(128u)) 32#define METHOD_HEADER2 (reinterpret_cast<OatQuickMethodHeader*>(136u)) 33#define METHOD_HEADER3 (reinterpret_cast<OatQuickMethodHeader*>(144u)) 34 35TEST_F(CHATest, CHACheckDependency) { 36 ClassHierarchyAnalysis cha; 37 MutexLock cha_mu(Thread::Current(), *Locks::cha_lock_); 38 39 ASSERT_EQ(cha.GetDependents(METHOD1), nullptr); 40 ASSERT_EQ(cha.GetDependents(METHOD2), nullptr); 41 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 42 43 cha.AddDependency(METHOD1, METHOD2, METHOD_HEADER2); 44 ASSERT_EQ(cha.GetDependents(METHOD2), nullptr); 45 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 46 auto dependents = cha.GetDependents(METHOD1); 47 ASSERT_EQ(dependents->size(), 1u); 48 ASSERT_EQ(dependents->at(0).first, METHOD2); 49 ASSERT_EQ(dependents->at(0).second, METHOD_HEADER2); 50 51 cha.AddDependency(METHOD1, METHOD3, METHOD_HEADER3); 52 ASSERT_EQ(cha.GetDependents(METHOD2), nullptr); 53 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 54 dependents = cha.GetDependents(METHOD1); 55 ASSERT_EQ(dependents->size(), 2u); 56 ASSERT_EQ(dependents->at(0).first, METHOD2); 57 ASSERT_EQ(dependents->at(0).second, METHOD_HEADER2); 58 ASSERT_EQ(dependents->at(1).first, METHOD3); 59 ASSERT_EQ(dependents->at(1).second, METHOD_HEADER3); 60 61 std::unordered_set<OatQuickMethodHeader*> headers; 62 headers.insert(METHOD_HEADER2); 63 cha.RemoveDependentsWithMethodHeaders(headers); 64 ASSERT_EQ(cha.GetDependents(METHOD2), nullptr); 65 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 66 dependents = cha.GetDependents(METHOD1); 67 ASSERT_EQ(dependents->size(), 1u); 68 ASSERT_EQ(dependents->at(0).first, METHOD3); 69 ASSERT_EQ(dependents->at(0).second, METHOD_HEADER3); 70 71 cha.AddDependency(METHOD2, METHOD1, METHOD_HEADER1); 72 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 73 dependents = cha.GetDependents(METHOD1); 74 ASSERT_EQ(dependents->size(), 1u); 75 dependents = cha.GetDependents(METHOD2); 76 ASSERT_EQ(dependents->size(), 1u); 77 78 headers.insert(METHOD_HEADER3); 79 cha.RemoveDependentsWithMethodHeaders(headers); 80 ASSERT_EQ(cha.GetDependents(METHOD1), nullptr); 81 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 82 dependents = cha.GetDependents(METHOD2); 83 ASSERT_EQ(dependents->size(), 1u); 84 ASSERT_EQ(dependents->at(0).first, METHOD1); 85 ASSERT_EQ(dependents->at(0).second, METHOD_HEADER1); 86 87 cha.RemoveDependencyFor(METHOD2); 88 ASSERT_EQ(cha.GetDependents(METHOD1), nullptr); 89 ASSERT_EQ(cha.GetDependents(METHOD2), nullptr); 90 ASSERT_EQ(cha.GetDependents(METHOD3), nullptr); 91} 92 93} // namespace art 94