Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f11e0a39 authored by Tej Singh's avatar Tej Singh
Browse files

Tests for libstatspull 

Sets up a new test suite in frameworks/base/apex/statsd. Creates a JNI
wrapper around libstatspull so that configs can be added and removed
from java, but the pulls happen in native code.

Bug: 145310729
Test: atest LibStatsPullTests
Change-Id: Ib3df74cac12f4276892be3f4a060a4a9dd120464
parent 55c70caa

apex/statsd/tests/libstatspull/Android.bp

0 → 100644
+56 −0
Original line number Diff line number Diff line 
// Copyright (C) 2019 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.



android_test {

    name: "LibStatsPullTests",

    static_libs: [

        "androidx.test.rules",

        "platformprotoslite",

        "statsdprotolite",

        "truth-prebuilt",

    ],

    libs: [

        "android.test.runner.stubs",

        "android.test.base.stubs",

    ],

    jni_libs: [

        "libstatspull_testhelper",

    ],

    srcs: [

        "src/**/*.java",

        "protos/**/*.proto",

        ],

    test_suites: [

        "general-tests",

    ],

    platform_apis: true,

    privileged: true,

    certificate: "platform",

    compile_multilib: "both",

}



cc_library_shared {

    name: "libstatspull_testhelper",

    srcs: ["jni/stats_pull_helper.cpp"],

    cflags: [

        "-Wall",

        "-Werror",

    ],

    shared_libs: [

        "libbinder",

        "libutils",

        "libstatspull",

        "libstatssocket",

    ],

}
 
 No newline at end of file

apex/statsd/tests/libstatspull/AndroidManifest.xml

0 → 100644
+30 −0
Original line number Diff line number Diff line 
<?xml version="1.0" encoding="utf-8"?>

<!--

 * Copyright (C) 2020 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.

 -->



<manifest xmlns:android="http://schemas.android.com/apk/res/android"

          package="com.android.internal.os.statsd.libstats" >





    <uses-permission android:name="android.permission.DUMP" />

    <uses-permission android:name="android.permission.PACKAGE_USAGE_STATS" />



    <instrumentation android:name="androidx.test.runner.AndroidJUnitRunner"

                     android:targetPackage="com.android.internal.os.statsd.libstats"

                     android:label="Tests for libstatspull">

    </instrumentation>

</manifest>

apex/statsd/tests/libstatspull/TEST_MAPPING

0 → 100644
+7 −0
Original line number Diff line number Diff line 
{

  "presubmit" : [

    {

      "name" : "LibStatsPullTests"

    }

  ]

}
 
 No newline at end of file

apex/statsd/tests/libstatspull/jni/stats_pull_helper.cpp

0 → 100644
+88 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2019, 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 <binder/ProcessState.h>

#include <jni.h>

#include <log/log.h>

#include <stats_event.h>

#include <stats_pull_atom_callback.h>



#include <chrono>

#include <thread>



using std::this_thread::sleep_for;

using namespace android;



namespace {

static int32_t sAtomTag;

static int32_t sPullReturnVal;

static int64_t sLatencyMillis;

static int32_t sAtomsPerPull;

static int32_t sNumPulls = 0;



static bool initialized = false;



static void init() {

    if (!initialized) {

        initialized = true;

        // Set up the binder

        sp<ProcessState> ps(ProcessState::self());

        ps->setThreadPoolMaxThreadCount(9);

        ps->startThreadPool();

        ps->giveThreadPoolName();

    }

}



static status_pull_atom_return_t pullAtomCallback(int32_t atomTag, pulled_stats_event_list* data,

                                                  void* /*cookie*/) {

    sNumPulls++;

    sleep_for(std::chrono::milliseconds(sLatencyMillis));

    for (int i = 0; i < sAtomsPerPull; i++) {

        stats_event* event = add_stats_event_to_pull_data(data);

        stats_event_set_atom_id(event, atomTag);

        stats_event_write_int64(event, (int64_t) sNumPulls);

        stats_event_build(event);

    }

    return sPullReturnVal;

}



extern "C"

JNIEXPORT void JNICALL

Java_com_android_internal_os_statsd_libstats_LibStatsPullTests_registerStatsPuller(

        JNIEnv* /*env*/, jobject /* this */, jint atomTag, jlong timeoutNs, jlong coolDownNs,

        jint pullRetVal, jlong latencyMillis, int atomsPerPull)

{

    init();

    sAtomTag = atomTag;

    sPullReturnVal = pullRetVal;

    sLatencyMillis = latencyMillis;

    sAtomsPerPull = atomsPerPull;

    sNumPulls = 0;

    pull_atom_metadata metadata = {.cool_down_ns = coolDownNs,

                                   .timeout_ns = timeoutNs,

                                   .additive_fields = nullptr,

                                   .additive_fields_size = 0};

    register_stats_pull_atom_callback(sAtomTag, &pullAtomCallback, &metadata, nullptr);

}



extern "C"

JNIEXPORT void JNICALL

Java_com_android_internal_os_statsd_libstats_LibStatsPullTests_unregisterStatsPuller(

        JNIEnv* /*env*/, jobject /* this */, jint /*atomTag*/)

{

    unregister_stats_pull_atom_callback(sAtomTag);

}

} // namespace
 
 No newline at end of file

apex/statsd/tests/libstatspull/protos/test_atoms.proto

0 → 100644
+32 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2020 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.

 */

syntax = "proto2";



package com.android.internal.os.statsd.protos;



option java_package = "com.android.internal.os.statsd.protos";

option java_outer_classname = "TestAtoms";



message PullCallbackAtomWrapper {

  optional PullCallbackAtom pull_callback_atom = 150030;

}



message PullCallbackAtom {

  optional int64 long_val = 1;

}