1df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck/*
2df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * Copyright (C) 2016 The Android Open Source Project
3df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck *
4df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * Licensed under the Apache License, Version 2.0 (the "License");
5df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * you may not use this file except in compliance with the License.
6df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * You may obtain a copy of the License at
7df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck *
8df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck *      http://www.apache.org/licenses/LICENSE-2.0
9df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck *
10df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * Unless required by applicable law or agreed to in writing, software
11df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * distributed under the License is distributed on an "AS IS" BASIS,
12df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * See the License for the specific language governing permissions and
14df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck * limitations under the License.
15df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck */
16df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
17df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <gtest/gtest.h>
18df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
19df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include "service/GraphicsStatsService.h"
20df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
21df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <frameworks/base/core/proto/android/service/graphicsstats.pb.h>
22df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
23df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <sys/types.h>
24df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <sys/stat.h>
25df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <stdio.h>
26df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <stdlib.h>
27df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck#include <unistd.h>
28df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
29df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reckusing namespace android;
30df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reckusing namespace android::uirenderer;
31df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
32df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reckstd::string findRootPath() {
33df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    char path[1024];
34df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    ssize_t r = readlink("/proc/self/exe", path, 1024);
35df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // < 1023 because we need room for the null terminator
36df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    if (r <= 0 || r > 1023) {
37df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        int err = errno;
38df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        fprintf(stderr, "Failed to read from /proc/self/exe; r=%zd, err=%d (%s)\n",
39df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck                r, err, strerror(err));
40df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        exit(EXIT_FAILURE);
41df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
42df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    while (--r > 0) {
43df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        if (path[r] == '/') {
44df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            path[r] = '\0';
45df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            return std::string(path);
46df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        }
47df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
48df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    return std::string();
49df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck}
50df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
51df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck// No code left untested
52df1742ed47da1e9b61afeae16fa448d5302a8aa0John ReckTEST(GraphicsStats, findRootPath) {
53df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    std::string expected = "/data/nativetest/hwui_unit_tests";
54df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(expected, findRootPath());
55df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck}
56df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
57df1742ed47da1e9b61afeae16fa448d5302a8aa0John ReckTEST(GraphicsStats, saveLoad) {
58df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    std::string path = findRootPath() + "/test_saveLoad";
59df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    std::string packageName = "com.test.saveLoad";
60df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    ProfileData mockData;
61df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.jankFrameCount = 20;
62df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.totalFrameCount = 100;
63df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.statStartTime = 10000;
64df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // Fill with patterned data we can recognize but which won't map to a
65df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // memset or basic for iteration count
66df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.frameCounts.size(); i++) {
67df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.frameCounts[i] = ((i % 10) + 1) * 2;
68df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
69df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.slowFrameCounts.size(); i++) {
70df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.slowFrameCounts[i] = (i % 5) + 1;
71df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
72df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    GraphicsStatsService::saveBuffer(path, packageName, 5, 3000, 7000, &mockData);
73df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    service::GraphicsStatsProto loadedProto;
74df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_TRUE(GraphicsStatsService::parseFromFile(path, &loadedProto));
75df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // Clean up the file
76df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    unlink(path.c_str());
77df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
78df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(packageName, loadedProto.package_name());
79df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(5, loadedProto.version_code());
80df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(3000, loadedProto.stats_start());
81df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(7000, loadedProto.stats_end());
82df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // ASSERT here so we don't continue with a nullptr deref crash if this is false
83df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    ASSERT_TRUE(loadedProto.has_summary());
84df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(20, loadedProto.summary().janky_frames());
85df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(100, loadedProto.summary().total_frames());
86df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(mockData.frameCounts.size() + mockData.slowFrameCounts.size(),
87df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            (size_t) loadedProto.histogram_size());
88df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < (size_t) loadedProto.histogram_size(); i++) {
89df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        int expectedCount, expectedBucket;
90df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        if (i < mockData.frameCounts.size()) {
91df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount = ((i % 10) + 1) * 2;
92df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedBucket = JankTracker::frameTimeForFrameCountIndex(i);
93df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        } else {
94df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            int temp = i - mockData.frameCounts.size();
95df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount = (temp % 5) + 1;
96df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedBucket = JankTracker::frameTimeForSlowFrameCountIndex(temp);
97df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        }
98df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        EXPECT_EQ(expectedCount, loadedProto.histogram().Get(i).frame_count());
99df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        EXPECT_EQ(expectedBucket, loadedProto.histogram().Get(i).render_millis());
100df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
101df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck}
102df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
103df1742ed47da1e9b61afeae16fa448d5302a8aa0John ReckTEST(GraphicsStats, merge) {
104df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    std::string path = findRootPath() + "/test_merge";
105df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    std::string packageName = "com.test.merge";
106df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    ProfileData mockData;
107df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.jankFrameCount = 20;
108df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.totalFrameCount = 100;
109df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.statStartTime = 10000;
110df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // Fill with patterned data we can recognize but which won't map to a
111df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // memset or basic for iteration count
112df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.frameCounts.size(); i++) {
113df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.frameCounts[i] = ((i % 10) + 1) * 2;
114df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
115df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.slowFrameCounts.size(); i++) {
116df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.slowFrameCounts[i] = (i % 5) + 1;
117df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
118df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    GraphicsStatsService::saveBuffer(path, packageName, 5, 3000, 7000, &mockData);
119df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.jankFrameCount = 50;
120df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    mockData.totalFrameCount = 500;
121df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.frameCounts.size(); i++) {
122df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.frameCounts[i] = (i % 5) + 1;
123df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
124df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < mockData.slowFrameCounts.size(); i++) {
125df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        mockData.slowFrameCounts[i] = ((i % 10) + 1) * 2;
126df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
127df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    GraphicsStatsService::saveBuffer(path, packageName, 5, 7050, 10000, &mockData);
128df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
129df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    service::GraphicsStatsProto loadedProto;
130df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_TRUE(GraphicsStatsService::parseFromFile(path, &loadedProto));
131df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // Clean up the file
132df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    unlink(path.c_str());
133df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck
134df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(packageName, loadedProto.package_name());
135df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(5, loadedProto.version_code());
136df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(3000, loadedProto.stats_start());
137df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(10000, loadedProto.stats_end());
138df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    // ASSERT here so we don't continue with a nullptr deref crash if this is false
139df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    ASSERT_TRUE(loadedProto.has_summary());
140df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(20 + 50, loadedProto.summary().janky_frames());
141df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(100 + 500, loadedProto.summary().total_frames());
142df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    EXPECT_EQ(mockData.frameCounts.size() + mockData.slowFrameCounts.size(),
143df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            (size_t) loadedProto.histogram_size());
144df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    for (size_t i = 0; i < (size_t) loadedProto.histogram_size(); i++) {
145df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        int expectedCount, expectedBucket;
146df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        if (i < mockData.frameCounts.size()) {
147df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount = ((i % 10) + 1) * 2;
148df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount += (i % 5) + 1;
149df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedBucket = JankTracker::frameTimeForFrameCountIndex(i);
150df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        } else {
151df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            int temp = i - mockData.frameCounts.size();
152df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount = (temp % 5) + 1;
153df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedCount += ((temp % 10) + 1) * 2;
154df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck            expectedBucket = JankTracker::frameTimeForSlowFrameCountIndex(temp);
155df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        }
156df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        EXPECT_EQ(expectedCount, loadedProto.histogram().Get(i).frame_count());
157df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck        EXPECT_EQ(expectedBucket, loadedProto.histogram().Get(i).render_millis());
158df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck    }
159df1742ed47da1e9b61afeae16fa448d5302a8aa0John Reck}