1 /*
2  * Copyright (C) 2020 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 <gtest/gtest.h>
18 #include "stats_buffer_writer.h"
19 #include "stats_event.h"
20 #include "stats_socket.h"
21 
22 TEST(StatsWriterTest, TestSocketClose) {
23     AStatsEvent* event = AStatsEvent_obtain();
24     AStatsEvent_setAtomId(event, 100);
25     AStatsEvent_writeInt32(event, 5);
26     int successResult = AStatsEvent_write(event);
27     AStatsEvent_release(event);
28 
29     // In the case of a successful write, we return the number of bytes written.
30     EXPECT_GT(successResult, 0);
31     EXPECT_FALSE(stats_log_is_closed());
32 
33     AStatsSocket_close();
34 
35     EXPECT_TRUE(stats_log_is_closed());
36 }
37