1 /*
2  * Copyright (C) 2021 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 "../libstatspull_lazy.h"
18 
19 #include <gtest/gtest.h>
20 
21 #include "stats_pull_atom_callback.h"
22 //#include "stats_event.h"
23 
24 // The tests here are just for the case when libstatspull.so cannot be loaded by
25 // libstatspull_lazy.
26 class LibstatspullLazyTest : public ::testing::Test {
27   protected:
SetUp()28     virtual void SetUp() {
29         ::testing::Test::SetUp();
30         PreventLibstatspullLazyLoadingForTests();
31     }
32 };
33 
34 static const char* kLoadFailed = "Failed to load libstatspull.so";
35 
TEST_F(LibstatspullLazyTest,NoLibstatspullForPullAtomMetadata)36 TEST_F(LibstatspullLazyTest, NoLibstatspullForPullAtomMetadata) {
37     AStatsManager_PullAtomMetadata* metadata = NULL;
38     EXPECT_DEATH(AStatsManager_PullAtomMetadata_obtain(), kLoadFailed);
39     EXPECT_DEATH(AStatsManager_PullAtomMetadata_release(metadata), kLoadFailed);
40     EXPECT_DEATH(AStatsManager_PullAtomMetadata_setCoolDownMillis(metadata, 0), kLoadFailed);
41     EXPECT_DEATH(AStatsManager_PullAtomMetadata_getCoolDownMillis(metadata), kLoadFailed);
42     EXPECT_DEATH(AStatsManager_PullAtomMetadata_setTimeoutMillis(metadata, 0), kLoadFailed);
43     EXPECT_DEATH(AStatsManager_PullAtomMetadata_getTimeoutMillis(metadata), kLoadFailed);
44     EXPECT_DEATH(AStatsManager_PullAtomMetadata_setAdditiveFields(metadata, NULL, 0), kLoadFailed);
45     EXPECT_DEATH(AStatsManager_PullAtomMetadata_getNumAdditiveFields(metadata), kLoadFailed);
46     EXPECT_DEATH(AStatsManager_PullAtomMetadata_getAdditiveFields(metadata, NULL), kLoadFailed);
47 }
48 
TEST_F(LibstatspullLazyTest,NoLibstatspullForAStatsEventList)49 TEST_F(LibstatspullLazyTest, NoLibstatspullForAStatsEventList) {
50     AStatsEventList* event_list = NULL;
51     EXPECT_DEATH(AStatsEventList_addStatsEvent(event_list), kLoadFailed);
52 }
53 
TEST_F(LibstatspullLazyTest,NoLibstatspullForPullAtomCallback)54 TEST_F(LibstatspullLazyTest, NoLibstatspullForPullAtomCallback) {
55     AStatsManager_PullAtomCallback callback = NULL;
56     EXPECT_DEATH(AStatsManager_setPullAtomCallback(0, NULL, callback, NULL), kLoadFailed);
57     EXPECT_DEATH(AStatsManager_clearPullAtomCallback(0), kLoadFailed);
58 }