cmds/statsd/tests/state/StateTracker_test.cpp

/*
 * 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 <gtest/gtest.h>
#include "state/StateManager.h"
#include "state/StateTracker.h"
#include "state/StateListener.h"

#include "tests/statsd_test_util.h"

#ifdef __ANDROID__

namespace android {
namespace os {
namespace statsd {

/**
 * Mock StateListener class for testing.
 * Stores primary key and state pairs.
 */
class TestStateListener : public virtual StateListener {
public:
    TestStateListener(){};

    virtual ~TestStateListener(){};

    struct Update {
        Update(const HashableDimensionKey& key, int state) : mKey(key), mState(state){};
        HashableDimensionKey mKey;
        int mState;
    };

    std::vector<Update> updates;

    void onStateChanged(const int64_t eventTimeNs, const int32_t atomId,
                        const HashableDimensionKey& primaryKey, int oldState, int newState) {
        updates.emplace_back(primaryKey, newState);
    }
};

int getStateInt(StateManager& mgr, int atomId, const HashableDimensionKey& queryKey) {
    FieldValue output;
    mgr.getStateValue(atomId, queryKey, &output);
    return output.mValue.int_value;
}

// TODO(b/149590301): Update these helpers to use new socket schema.
//// START: build event functions.
//// State with no primary fields - ScreenStateChanged
//std::shared_ptr<LogEvent> buildScreenEvent(int state) {
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::SCREEN_STATE_CHANGED, 1000 /*timestamp*/);
//    event->write((int32_t)state);
//    event->init();
//    return event;
//}
//
//// State with one primary field - UidProcessStateChanged
//std::shared_ptr<LogEvent> buildUidProcessEvent(int uid, int state) {
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::UID_PROCESS_STATE_CHANGED, 1000 /*timestamp*/);
//    event->write((int32_t)uid);
//    event->write((int32_t)state);
//    event->init();
//    return event;
//}
//
//// State with first uid in attribution chain as primary field - WakelockStateChanged
//std::shared_ptr<LogEvent> buildPartialWakelockEvent(int uid, const std::string& tag, bool acquire) {
//    std::vector<AttributionNodeInternal> chain;
//    chain.push_back(AttributionNodeInternal());
//    AttributionNodeInternal& attr = chain.back();
//    attr.set_uid(uid);
//
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::WAKELOCK_STATE_CHANGED, 1000 /* timestamp */);
//    event->write(chain);
//    event->write((int32_t)1);  // PARTIAL_WAKE_LOCK
//    event->write(tag);
//    event->write(acquire ? 1 : 0);
//    event->init();
//    return event;
//}
//
//// State with multiple primary fields - OverlayStateChanged
//std::shared_ptr<LogEvent> buildOverlayEvent(int uid, const std::string& packageName, int state) {
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::OVERLAY_STATE_CHANGED, 1000 /*timestamp*/);
//    event->write((int32_t)uid);
//    event->write(packageName);
//    event->write(true);  // using_alert_window
//    event->write((int32_t)state);
//    event->init();
//    return event;
//}
//
//// Incorrect event - missing fields
//std::shared_ptr<LogEvent> buildIncorrectOverlayEvent(int uid, const std::string& packageName, int state) {
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::OVERLAY_STATE_CHANGED, 1000 /*timestamp*/);
//    event->write((int32_t)uid);
//    event->write(packageName);
//    event->write((int32_t)state);
//    event->init();
//    return event;
//}
//
//// Incorrect event - exclusive state has wrong type
//std::shared_ptr<LogEvent> buildOverlayEventBadStateType(int uid, const std::string& packageName) {
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::OVERLAY_STATE_CHANGED, 1000 /*timestamp*/);
//    event->write((int32_t)uid);
//    event->write(packageName);
//    event->write(true);
//    event->write("string");  // exclusive state: string instead of int
//    event->init();
//    return event;
//}
//
//std::shared_ptr<LogEvent> buildBleScanEvent(int uid, bool acquire, bool reset) {
//    std::vector<AttributionNodeInternal> chain;
//    chain.push_back(AttributionNodeInternal());
//    AttributionNodeInternal& attr = chain.back();
//    attr.set_uid(uid);
//
//    std::shared_ptr<LogEvent> event =
//            std::make_shared<LogEvent>(android::util::BLE_SCAN_STATE_CHANGED, 1000);
//    event->write(chain);
//    event->write(reset ? 2 : acquire ? 1 : 0);  // PARTIAL_WAKE_LOCK
//    event->write(0);                            // filtered
//    event->write(0);                            // first match
//    event->write(0);                            // opportunistic
//    event->init();
//    return event;
//}
// END: build event functions.

// START: get primary key functions
void getUidProcessKey(int uid, HashableDimensionKey* key) {
    int pos1[] = {1, 0, 0};
    Field field1(27 /* atom id */, pos1, 0 /* depth */);
    Value value1((int32_t)uid);

    key->addValue(FieldValue(field1, value1));
}

void getOverlayKey(int uid, string packageName, HashableDimensionKey* key) {
    int pos1[] = {1, 0, 0};
    int pos2[] = {2, 0, 0};

    Field field1(59 /* atom id */, pos1, 0 /* depth */);
    Field field2(59 /* atom id */, pos2, 0 /* depth */);

    Value value1((int32_t)uid);
    Value value2(packageName);

    key->addValue(FieldValue(field1, value1));
    key->addValue(FieldValue(field2, value2));
}

void getPartialWakelockKey(int uid, const std::string& tag, HashableDimensionKey* key) {
    int pos1[] = {1, 1, 1};
    int pos3[] = {2, 0, 0};
    int pos4[] = {3, 0, 0};

    Field field1(10 /* atom id */, pos1, 2 /* depth */);

    Field field3(10 /* atom id */, pos3, 0 /* depth */);
    Field field4(10 /* atom id */, pos4, 0 /* depth */);

    Value value1((int32_t)uid);
    Value value3((int32_t)1 /*partial*/);
    Value value4(tag);

    key->addValue(FieldValue(field1, value1));
    key->addValue(FieldValue(field3, value3));
    key->addValue(FieldValue(field4, value4));
}

void getPartialWakelockKey(int uid, HashableDimensionKey* key) {
    int pos1[] = {1, 1, 1};
    int pos3[] = {2, 0, 0};

    Field field1(10 /* atom id */, pos1, 2 /* depth */);
    Field field3(10 /* atom id */, pos3, 0 /* depth */);

    Value value1((int32_t)uid);
    Value value3((int32_t)1 /*partial*/);

    key->addValue(FieldValue(field1, value1));
    key->addValue(FieldValue(field3, value3));
}
// END: get primary key functions

TEST(StateListenerTest, TestStateListenerWeakPointer) {
    sp<TestStateListener> listener = new TestStateListener();
    wp<TestStateListener> wListener = listener;
    listener = nullptr;  // let go of listener
    EXPECT_TRUE(wListener.promote() == nullptr);
}

TEST(StateManagerTest, TestStateManagerGetInstance) {
    sp<TestStateListener> listener1 = new TestStateListener();
    StateManager& mgr = StateManager::getInstance();
    mgr.clear();

    mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener1);
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(1, StateManager::getInstance().getStateTrackersCount());
}

/**
 * Test registering listeners to StateTrackers
 *
 * - StateManager will create a new StateTracker if it doesn't already exist
 * and then register the listener to the StateTracker.
 * - If a listener is already registered to a StateTracker, it is not added again.
 * - StateTrackers are only created for atoms that are state atoms.
 */
TEST(StateTrackerTest, TestRegisterListener) {
    sp<TestStateListener> listener1 = new TestStateListener();
    sp<TestStateListener> listener2 = new TestStateListener();
    StateManager mgr;

    // Register listener to non-existing StateTracker
    EXPECT_EQ(0, mgr.getStateTrackersCount());
    EXPECT_TRUE(mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener1));
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Register listener to existing StateTracker
    EXPECT_TRUE(mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener2));
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(2, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Register already registered listener to existing StateTracker
    EXPECT_TRUE(mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener2));
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(2, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Register listener to non-state atom
    EXPECT_FALSE(mgr.registerListener(android::util::BATTERY_LEVEL_CHANGED, listener2));
    EXPECT_EQ(1, mgr.getStateTrackersCount());
}

/**
 * Test unregistering listeners from StateTrackers
 *
 * - StateManager will unregister listeners from a StateTracker only if the
 * StateTracker exists and the listener is registered to the StateTracker.
 * - Once all listeners are removed from a StateTracker, the StateTracker
 * is also removed.
 */
TEST(StateTrackerTest, TestUnregisterListener) {
    sp<TestStateListener> listener1 = new TestStateListener();
    sp<TestStateListener> listener2 = new TestStateListener();
    StateManager mgr;

    // Unregister listener from non-existing StateTracker
    EXPECT_EQ(0, mgr.getStateTrackersCount());
    mgr.unregisterListener(android::util::SCREEN_STATE_CHANGED, listener1);
    EXPECT_EQ(0, mgr.getStateTrackersCount());
    EXPECT_EQ(-1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Unregister non-registered listener from existing StateTracker
    mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener1);
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));
    mgr.unregisterListener(android::util::SCREEN_STATE_CHANGED, listener2);
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Unregister second-to-last listener from StateTracker
    mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener2);
    mgr.unregisterListener(android::util::SCREEN_STATE_CHANGED, listener1);
    EXPECT_EQ(1, mgr.getStateTrackersCount());
    EXPECT_EQ(1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));

    // Unregister last listener from StateTracker
    mgr.unregisterListener(android::util::SCREEN_STATE_CHANGED, listener2);
    EXPECT_EQ(0, mgr.getStateTrackersCount());
    EXPECT_EQ(-1, mgr.getListenersCount(android::util::SCREEN_STATE_CHANGED));
}
// TODO(b/149590301): Update these tests to use new socket schema.
///**
// * Test a binary state atom with nested counting.
// *
// * To go from an "ON" state to an "OFF" state with nested counting, we must see
// * an equal number of "OFF" events as "ON" events.
// * For example, ACQUIRE, ACQUIRE, RELEASE will still be in the ACQUIRE state.
// * ACQUIRE, ACQUIRE, RELEASE, RELEASE will be in the RELEASE state.
// */
//TEST(StateTrackerTest, TestStateChangeNested) {
//    sp<TestStateListener> listener = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::WAKELOCK_STATE_CHANGED, listener);
//
//    std::shared_ptr<LogEvent> event1 =
//            buildPartialWakelockEvent(1000 /* uid */, "tag", true /*acquire*/);
//    mgr.onLogEvent(*event1);
//    EXPECT_EQ(1, listener->updates.size());
//    EXPECT_EQ(1000, listener->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(1, listener->updates[0].mState);
//    listener->updates.clear();
//
//    std::shared_ptr<LogEvent> event2 =
//            buildPartialWakelockEvent(1000 /* uid */, "tag", true /*acquire*/);
//    mgr.onLogEvent(*event2);
//    EXPECT_EQ(0, listener->updates.size());
//
//    std::shared_ptr<LogEvent> event3 =
//            buildPartialWakelockEvent(1000 /* uid */, "tag", false /*release*/);
//    mgr.onLogEvent(*event3);
//    EXPECT_EQ(0, listener->updates.size());
//
//    std::shared_ptr<LogEvent> event4 =
//            buildPartialWakelockEvent(1000 /* uid */, "tag", false /*release*/);
//    mgr.onLogEvent(*event4);
//    EXPECT_EQ(1, listener->updates.size());
//    EXPECT_EQ(1000, listener->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(0, listener->updates[0].mState);
//}
//
///**
// * Test a state atom with a reset state.
// *
// * If the reset state value is seen, every state in the map is set to the default
// * state and every listener is notified.
// */
//TEST(StateTrackerTest, TestStateChangeReset) {
//    sp<TestStateListener> listener = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::BLE_SCAN_STATE_CHANGED, listener);
//
//    std::shared_ptr<LogEvent> event1 =
//            buildBleScanEvent(1000 /* uid */, true /*acquire*/, false /*reset*/);
//    mgr.onLogEvent(*event1);
//    EXPECT_EQ(1, listener->updates.size());
//    EXPECT_EQ(1000, listener->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(BleScanStateChanged::ON, listener->updates[0].mState);
//    listener->updates.clear();
//
//    std::shared_ptr<LogEvent> event2 =
//            buildBleScanEvent(2000 /* uid */, true /*acquire*/, false /*reset*/);
//    mgr.onLogEvent(*event2);
//    EXPECT_EQ(1, listener->updates.size());
//    EXPECT_EQ(2000, listener->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(BleScanStateChanged::ON, listener->updates[0].mState);
//    listener->updates.clear();
//
//    std::shared_ptr<LogEvent> event3 =
//            buildBleScanEvent(2000 /* uid */, false /*acquire*/, true /*reset*/);
//    mgr.onLogEvent(*event3);
//    EXPECT_EQ(2, listener->updates.size());
//    EXPECT_EQ(BleScanStateChanged::OFF, listener->updates[0].mState);
//    EXPECT_EQ(BleScanStateChanged::OFF, listener->updates[1].mState);
//}
//
///**
// * Test StateManager's onLogEvent and StateListener's onStateChanged correctly
// * updates listener for states without primary keys.
// */
//TEST(StateTrackerTest, TestStateChangeNoPrimaryFields) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener1);
//
//    // log event
//    std::shared_ptr<LogEvent> event =
//            buildScreenEvent(android::view::DisplayStateEnum::DISPLAY_STATE_ON);
//    mgr.onLogEvent(*event);
//
//    // check listener was updated
//    EXPECT_EQ(1, listener1->updates.size());
//    EXPECT_EQ(DEFAULT_DIMENSION_KEY, listener1->updates[0].mKey);
//    EXPECT_EQ(2, listener1->updates[0].mState);
//
//    // check StateTracker was updated by querying for state
//    HashableDimensionKey queryKey = DEFAULT_DIMENSION_KEY;
//    EXPECT_EQ(android::view::DisplayStateEnum::DISPLAY_STATE_ON,
//              getStateInt(mgr, android::util::SCREEN_STATE_CHANGED, queryKey));
//}
//
///**
// * Test StateManager's onLogEvent and StateListener's onStateChanged correctly
// * updates listener for states with one primary key.
// */
//TEST(StateTrackerTest, TestStateChangeOnePrimaryField) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::UID_PROCESS_STATE_CHANGED, listener1);
//
//    // log event
//    std::shared_ptr<LogEvent> event =
//            buildUidProcessEvent(1000 /* uid */, android::app::ProcessStateEnum::PROCESS_STATE_TOP);
//    mgr.onLogEvent(*event);
//
//    // check listener was updated
//    EXPECT_EQ(1, listener1->updates.size());
//    EXPECT_EQ(1000, listener1->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(1002, listener1->updates[0].mState);
//
//    // check StateTracker was updated by querying for state
//    HashableDimensionKey queryKey;
//    getUidProcessKey(1000 /* uid */, &queryKey);
//    EXPECT_EQ(android::app::ProcessStateEnum::PROCESS_STATE_TOP,
//              getStateInt(mgr, android::util::UID_PROCESS_STATE_CHANGED, queryKey));
//}
//
//TEST(StateTrackerTest, TestStateChangePrimaryFieldAttrChain) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::WAKELOCK_STATE_CHANGED, listener1);
//
//    // Log event.
//    std::shared_ptr<LogEvent> event =
//            buildPartialWakelockEvent(1001 /* uid */, "tag1", true /* acquire */);
//    mgr.onLogEvent(*event);
//
//    EXPECT_EQ(1, mgr.getStateTrackersCount());
//    EXPECT_EQ(1, mgr.getListenersCount(android::util::WAKELOCK_STATE_CHANGED));
//
//    // Check listener was updated.
//    EXPECT_EQ(1, listener1->updates.size());
//    EXPECT_EQ(3, listener1->updates[0].mKey.getValues().size());
//    EXPECT_EQ(1001, listener1->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(1, listener1->updates[0].mKey.getValues()[1].mValue.int_value);
//    EXPECT_EQ("tag1", listener1->updates[0].mKey.getValues()[2].mValue.str_value);
//    EXPECT_EQ(WakelockStateChanged::ACQUIRE, listener1->updates[0].mState);
//
//    // Check StateTracker was updated by querying for state.
//    HashableDimensionKey queryKey;
//    getPartialWakelockKey(1001 /* uid */, "tag1", &queryKey);
//    EXPECT_EQ(WakelockStateChanged::ACQUIRE,
//              getStateInt(mgr, android::util::WAKELOCK_STATE_CHANGED, queryKey));
//
//    // No state stored for this query key.
//    HashableDimensionKey queryKey2;
//    getPartialWakelockKey(1002 /* uid */, "tag1", &queryKey2);
//    EXPECT_EQ(WakelockStateChanged::RELEASE,
//              getStateInt(mgr, android::util::WAKELOCK_STATE_CHANGED, queryKey2));
//
//    // Partial query fails.
//    HashableDimensionKey queryKey3;
//    getPartialWakelockKey(1001 /* uid */, &queryKey3);
//    EXPECT_EQ(WakelockStateChanged::RELEASE,
//              getStateInt(mgr, android::util::WAKELOCK_STATE_CHANGED, queryKey3));
//}
//
///**
// * Test StateManager's onLogEvent and StateListener's onStateChanged correctly
// * updates listener for states with multiple primary keys.
// */
//TEST(StateTrackerTest, TestStateChangeMultiplePrimaryFields) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::OVERLAY_STATE_CHANGED, listener1);
//
//    // log event
//    std::shared_ptr<LogEvent> event =
//            buildOverlayEvent(1000 /* uid */, "package1", 1);  // state: ENTERED
//    mgr.onLogEvent(*event);
//
//    // check listener was updated
//    EXPECT_EQ(1, listener1->updates.size());
//    EXPECT_EQ(1000, listener1->updates[0].mKey.getValues()[0].mValue.int_value);
//    EXPECT_EQ(1, listener1->updates[0].mState);
//
//    // check StateTracker was updated by querying for state
//    HashableDimensionKey queryKey;
//    getOverlayKey(1000 /* uid */, "package1", &queryKey);
//    EXPECT_EQ(OverlayStateChanged::ENTERED,
//              getStateInt(mgr, android::util::OVERLAY_STATE_CHANGED, queryKey));
//}
//
///**
// * Test StateManager's onLogEvent and StateListener's onStateChanged
// * when there is an error extracting state from log event. Listener is not
// * updated of state change.
// */
//TEST(StateTrackerTest, TestStateChangeEventError) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::OVERLAY_STATE_CHANGED, listener1);
//
//    // log event
//    std::shared_ptr<LogEvent> event1 =
//            buildIncorrectOverlayEvent(1000 /* uid */, "package1", 1 /* state */);
//    std::shared_ptr<LogEvent> event2 = buildOverlayEventBadStateType(1001 /* uid */, "package2");
//
//    // check listener was updated
//    mgr.onLogEvent(*event1);
//    EXPECT_EQ(0, listener1->updates.size());
//    mgr.onLogEvent(*event2);
//    EXPECT_EQ(0, listener1->updates.size());
//}
//
//TEST(StateTrackerTest, TestStateQuery) {
//    sp<TestStateListener> listener1 = new TestStateListener();
//    sp<TestStateListener> listener2 = new TestStateListener();
//    sp<TestStateListener> listener3 = new TestStateListener();
//    sp<TestStateListener> listener4 = new TestStateListener();
//    StateManager mgr;
//    mgr.registerListener(android::util::SCREEN_STATE_CHANGED, listener1);
//    mgr.registerListener(android::util::UID_PROCESS_STATE_CHANGED, listener2);
//    mgr.registerListener(android::util::OVERLAY_STATE_CHANGED, listener3);
//    mgr.registerListener(android::util::WAKELOCK_STATE_CHANGED, listener4);
//
//    std::shared_ptr<LogEvent> event1 = buildUidProcessEvent(
//            1000,
//            android::app::ProcessStateEnum::PROCESS_STATE_TOP);  //  state value: 1002
//    std::shared_ptr<LogEvent> event2 = buildUidProcessEvent(
//            1001,
//            android::app::ProcessStateEnum::PROCESS_STATE_FOREGROUND_SERVICE);  //  state value:
//                                                                                //  1003
//    std::shared_ptr<LogEvent> event3 = buildUidProcessEvent(
//            1002,
//            android::app::ProcessStateEnum::PROCESS_STATE_PERSISTENT);  //  state value: 1000
//    std::shared_ptr<LogEvent> event4 = buildUidProcessEvent(
//            1001,
//            android::app::ProcessStateEnum::PROCESS_STATE_TOP);  //  state value: 1002
//    std::shared_ptr<LogEvent> event5 =
//            buildScreenEvent(android::view::DisplayStateEnum::DISPLAY_STATE_ON);
//    std::shared_ptr<LogEvent> event6 =
//            buildOverlayEvent(1000, "package1", OverlayStateChanged::ENTERED);
//    std::shared_ptr<LogEvent> event7 =
//            buildOverlayEvent(1000, "package2", OverlayStateChanged::EXITED);
//    std::shared_ptr<LogEvent> event8 = buildPartialWakelockEvent(1005, "tag1", true);
//    std::shared_ptr<LogEvent> event9 = buildPartialWakelockEvent(1005, "tag2", false);
//
//    mgr.onLogEvent(*event1);
//    mgr.onLogEvent(*event2);
//    mgr.onLogEvent(*event3);
//    mgr.onLogEvent(*event5);
//    mgr.onLogEvent(*event5);
//    mgr.onLogEvent(*event6);
//    mgr.onLogEvent(*event7);
//    mgr.onLogEvent(*event8);
//    mgr.onLogEvent(*event9);
//
//    // Query for UidProcessState of uid 1001
//    HashableDimensionKey queryKey1;
//    getUidProcessKey(1001, &queryKey1);
//    EXPECT_EQ(android::app::ProcessStateEnum::PROCESS_STATE_FOREGROUND_SERVICE,
//              getStateInt(mgr, android::util::UID_PROCESS_STATE_CHANGED, queryKey1));
//
//    // Query for UidProcessState of uid 1004 - not in state map
//    HashableDimensionKey queryKey2;
//    getUidProcessKey(1004, &queryKey2);
//    EXPECT_EQ(-1, getStateInt(mgr, android::util::UID_PROCESS_STATE_CHANGED,
//                              queryKey2));  // default state
//
//    // Query for UidProcessState of uid 1001 - after change in state
//    mgr.onLogEvent(*event4);
//    EXPECT_EQ(android::app::ProcessStateEnum::PROCESS_STATE_TOP,
//              getStateInt(mgr, android::util::UID_PROCESS_STATE_CHANGED, queryKey1));
//
//    // Query for ScreenState
//    EXPECT_EQ(android::view::DisplayStateEnum::DISPLAY_STATE_ON,
//              getStateInt(mgr, android::util::SCREEN_STATE_CHANGED, DEFAULT_DIMENSION_KEY));
//
//    // Query for OverlayState of uid 1000, package name "package2"
//    HashableDimensionKey queryKey3;
//    getOverlayKey(1000, "package2", &queryKey3);
//    EXPECT_EQ(OverlayStateChanged::EXITED,
//              getStateInt(mgr, android::util::OVERLAY_STATE_CHANGED, queryKey3));
//
//    // Query for WakelockState of uid 1005, tag 2
//    HashableDimensionKey queryKey4;
//    getPartialWakelockKey(1005, "tag2", &queryKey4);
//    EXPECT_EQ(WakelockStateChanged::RELEASE,
//              getStateInt(mgr, android::util::WAKELOCK_STATE_CHANGED, queryKey4));
//
//    // Query for WakelockState of uid 1005, tag 1
//    HashableDimensionKey queryKey5;
//    getPartialWakelockKey(1005, "tag1", &queryKey5);
//    EXPECT_EQ(WakelockStateChanged::ACQUIRE,
//              getStateInt(mgr, android::util::WAKELOCK_STATE_CHANGED, queryKey5));
//}

}  // namespace statsd
}  // namespace os
}  // namespace android
#else
GTEST_LOG_(INFO) << "This test does nothing.\n";
#endif