atrace/1.0/vts/functional/VtsHalAtraceV1_0TargetTest.cpp

/*
 * Copyright (C) 2018 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 <unordered_set>

#include <android/hardware/atrace/1.0/IAtraceDevice.h>

#include <VtsHalHidlTargetTestBase.h>
#include <VtsHalHidlTargetTestEnvBase.h>

using ::android::sp;
using ::android::hardware::hidl_string;
using ::android::hardware::hidl_vec;
using ::android::hardware::Return;
using ::android::hardware::atrace::V1_0::IAtraceDevice;
using ::android::hardware::atrace::V1_0::Status;

// Test environment for Boot HIDL HAL.
class AtraceHidlEnvironment : public ::testing::VtsHalHidlTargetTestEnvBase {
   public:
    // get the test environment singleton
    static AtraceHidlEnvironment* Instance() {
        static AtraceHidlEnvironment* instance = new AtraceHidlEnvironment;
        return instance;
    }

    virtual void registerTestServices() override { registerTestService<IAtraceDevice>(); }

   private:
    AtraceHidlEnvironment() {}
};

/**
 * There is no expected behaviour that can be tested so these tests check the
 * HAL doesn't crash with different execution orders.
 */
struct AtraceHidlTest : public ::testing::VtsHalHidlTargetTestBase {
    virtual void SetUp() override {
        atrace = ::testing::VtsHalHidlTargetTestBase::getService<IAtraceDevice>(
            AtraceHidlEnvironment::Instance()->getServiceName<IAtraceDevice>());
        ASSERT_NE(atrace, nullptr);
    }

    sp<IAtraceDevice> atrace;
};

hidl_vec<hidl_string> getVendorCategoryName(sp<IAtraceDevice> atrace) {
    std::unordered_set<std::string> categories_set;
    Return<void> ret = atrace->listCategories([&](const auto& list) {
        for (const auto& category : list) {
            std::string name = category.name;
            if (categories_set.count(name)) {
                ADD_FAILURE() << "Duplicate category: " << name;
            } else {
                categories_set.emplace(name);
            }
        }
    });
    if (!ret.isOk()) {
        ADD_FAILURE();
    }
    hidl_vec<hidl_string> categories;
    categories.resize(categories_set.size());
    std::size_t i = 0;
    for (auto& c : categories_set) {
        categories[i++] = c;
    }
    return categories;
}

/* list categories from vendors. */
TEST_F(AtraceHidlTest, listCategories) {
    hidl_vec<hidl_string> vnd_categories = getVendorCategoryName(atrace);
    EXPECT_NE(0, vnd_categories.size());
}

/* enable categories. */
TEST_F(AtraceHidlTest, enableCategories) {
    hidl_vec<hidl_string> vnd_categories = getVendorCategoryName(atrace);
    // empty Category with ERROR_INVALID_ARGUMENT
    hidl_vec<hidl_string> empty_categories;
    auto ret = atrace->enableCategories(empty_categories);
    ASSERT_TRUE(ret.isOk());
    EXPECT_EQ(Status::ERROR_INVALID_ARGUMENT, ret);
    // non-empty categories SUCCESS
    ret = atrace->enableCategories(vnd_categories);
    ASSERT_TRUE(ret.isOk());
    EXPECT_EQ(Status::SUCCESS, ret);
}

/* enable categories. */
TEST_F(AtraceHidlTest, disableAllCategories) {
    auto ret = atrace->disableAllCategories();
    ASSERT_TRUE(ret.isOk());
    EXPECT_EQ(Status::SUCCESS, ret);
}

int main(int argc, char** argv) {
    ::testing::AddGlobalTestEnvironment(AtraceHidlEnvironment::Instance());
    ::testing::InitGoogleTest(&argc, argv);
    AtraceHidlEnvironment::Instance()->init(&argc, argv);
    int status = RUN_ALL_TESTS();
    ALOGI("Test result = %d", status);
    return status;
}