| /* |
| * Copyright (C) 2022 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include <android/binder_auto_utils.h> |
| #include <android/binder_manager.h> |
| #include <gmock/gmock.h> |
| #include <gtest/gtest.h> |
| |
| #include <aidl/android/aidl/tests/BnTestService.h> |
| #include <android-base/logging.h> |
| |
| using aidl::android::aidl::tests::BackendType; |
| using aidl::android::aidl::tests::ITestService; |
| using aidl::android::aidl::tests::ITestServiceDelegator; |
| |
| static constexpr int8_t kCustomByte = 8; |
| |
| struct CustomDelegator : public ITestServiceDelegator { |
| public: |
| CustomDelegator(std::shared_ptr<ITestService>& impl) : ITestServiceDelegator(impl) {} |
| |
| // Change RepeatByte to always return the same byte. |
| ndk::ScopedAStatus RepeatByte(int8_t /* token */, int8_t* _aidl_return) override { |
| *_aidl_return = kCustomByte; |
| return ndk::ScopedAStatus::ok(); |
| } |
| }; |
| |
| struct AidlDelegatorTest : testing::Test { |
| template <typename T> |
| std::shared_ptr<T> getService() { |
| ndk::SpAIBinder binder = ndk::SpAIBinder(AServiceManager_getService(T::descriptor)); |
| return T::fromBinder(binder); |
| } |
| void SetUp() override { service = getService<ITestService>(); } |
| std::shared_ptr<ITestService> service; |
| }; |
| |
| TEST_F(AidlDelegatorTest, SimpleDelegator) { |
| auto delegator = ndk::SharedRefBase::make<ITestServiceDelegator>(service); |
| int8_t returned_value; |
| auto status = delegator->RepeatByte(12, &returned_value); |
| ASSERT_TRUE(status.isOk()) << status.getMessage(); |
| EXPECT_EQ(12, returned_value); |
| } |
| |
| TEST_F(AidlDelegatorTest, CustomDelegator) { |
| auto delegator = ndk::SharedRefBase::make<CustomDelegator>(service); |
| int8_t returned_value; |
| auto status = delegator->RepeatByte(12, &returned_value); |
| ASSERT_TRUE(status.isOk()) << status.getMessage(); |
| EXPECT_EQ(kCustomByte, returned_value); |
| } |
| |
| TEST_F(AidlDelegatorTest, SendDelegator) { |
| auto delegator = ndk::SharedRefBase::make<ITestServiceDelegator>(service); |
| auto fromAsBinder = ITestServiceDelegator::fromBinder(delegator->asBinder()); |
| // Make sure the delegator works after asBinder -> fromBinder conversions |
| int8_t returned_value = 0; |
| auto status = fromAsBinder->RepeatByte(12, &returned_value); |
| ASSERT_TRUE(status.isOk()) << status.getDescription(); |
| EXPECT_EQ(12, returned_value); |
| } |