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 "../libstatssocket_lazy.h"
18 
19 #include <gtest/gtest.h>
20 
21 #include "stats_event.h"
22 #include "stats_socket.h"
23 
24 // The tests here are just for the case when libstatssocket.so cannot be loaded by
25 // libstatssocket_lazy.
26 class LibstatssocketLazyTest : public ::testing::Test {
27   protected:
SetUp()28     virtual void SetUp() {
29         ::testing::Test::SetUp();
30         PreventLibstatssocketLazyLoadingForTests();
31     }
32 };
33 
34 static const char* kLoadFailed = "Failed to load libstatssocket.so";
35 
TEST_F(LibstatssocketLazyTest,NoLibstatssocketForStatsEvent)36 TEST_F(LibstatssocketLazyTest, NoLibstatssocketForStatsEvent) {
37     AStatsEvent* event = NULL;
38     EXPECT_DEATH(AStatsEvent_obtain(), kLoadFailed);
39     EXPECT_DEATH(AStatsEvent_build(event), kLoadFailed);
40     EXPECT_DEATH(AStatsEvent_write(event), kLoadFailed);
41     EXPECT_DEATH(AStatsEvent_release(event), kLoadFailed);
42 
43     EXPECT_DEATH(AStatsEvent_setAtomId(event, 0), kLoadFailed);
44     EXPECT_DEATH(AStatsEvent_writeInt32(event, 0), kLoadFailed);
45     EXPECT_DEATH(AStatsEvent_writeInt64(event, 0), kLoadFailed);
46     EXPECT_DEATH(AStatsEvent_writeFloat(event, 0), kLoadFailed);
47     EXPECT_DEATH(AStatsEvent_writeBool(event, false), kLoadFailed);
48     EXPECT_DEATH(AStatsEvent_writeByteArray(event, NULL, 0), kLoadFailed);
49     EXPECT_DEATH(AStatsEvent_writeString(event, NULL), kLoadFailed);
50     EXPECT_DEATH(AStatsEvent_writeAttributionChain(event, NULL, NULL, 0), kLoadFailed);
51 
52     EXPECT_DEATH(AStatsEvent_addBoolAnnotation(event, 0, false), kLoadFailed);
53     EXPECT_DEATH(AStatsEvent_addInt32Annotation(event, 0, 0), kLoadFailed);
54 }
55 
TEST_F(LibstatssocketLazyTest,NoLibstatssocketForStatsSocket)56 TEST_F(LibstatssocketLazyTest, NoLibstatssocketForStatsSocket) {
57     EXPECT_DEATH(AStatsSocket_close(), kLoadFailed);
58 }