// Copyright (C) 2017 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 "src/metrics/EventMetricProducer.h"

#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <stdio.h>

#include <vector>

#include "metrics_test_helper.h"
#include "stats_event.h"
#include "tests/statsd_test_util.h"

using namespace testing;
using android::sp;
using std::set;
using std::unordered_map;
using std::vector;

#ifdef __ANDROID__

namespace android {
namespace os {
namespace statsd {

namespace {
const ConfigKey kConfigKey(0, 12345);
const uint64_t protoHash = 0x1234567890;

void makeLogEvent(LogEvent* logEvent, int32_t atomId, int64_t timestampNs, string str,
                  vector<uint8_t>* bytesField = nullptr) {
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, atomId);
    AStatsEvent_overwriteTimestamp(statsEvent, timestampNs);
    AStatsEvent_writeString(statsEvent, str.c_str());
    if (bytesField != nullptr) {
        AStatsEvent_writeByteArray(statsEvent, bytesField->data(), bytesField->size());
    }

    parseStatsEventToLogEvent(statsEvent, logEvent);
}

StatsLogReport onDumpReport(EventMetricProducer& producer, int64_t dumpTimeNs) {
    ProtoOutputStream output;
    set<int32_t> usedUids;
    producer.onDumpReport(dumpTimeNs, true /*include current partial bucket*/, true /*erase data*/,
                          FAST, nullptr, usedUids, &output);
    return outputStreamToProto(&output);
}

}  // anonymous namespace

class EventMetricProducerTest : public ::testing::Test {
    void SetUp() override {
        FlagProvider::getInstance().overrideFuncs(&isAtLeastSFuncTrue);
    }

    void TearDown() override {
        FlagProvider::getInstance().resetOverrides();
    }
};

TEST_F(EventMetricProducerTest, TestNoCondition) {
    int64_t bucketStartTimeNs = 10000000000;
    int64_t eventStartTimeNs = bucketStartTimeNs + 1;
    int64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    CreateNoValuesLogEvent(&event1, 1 /*tagId*/, bucketStartTimeNs + 1);

    LogEvent event2(/*uid=*/0, /*pid=*/0);
    CreateNoValuesLogEvent(&event2, 1 /*tagId*/, bucketStartTimeNs + 2);

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);

    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      wizard, protoHash, bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 20);
    backfillAggregatedAtoms(&report);
    EXPECT_TRUE(report.has_event_metrics());
    ASSERT_EQ(2, report.event_metrics().data_size());
    EXPECT_EQ(bucketStartTimeNs + 1, report.event_metrics().data(0).elapsed_timestamp_nanos());
    EXPECT_EQ(bucketStartTimeNs + 2, report.event_metrics().data(1).elapsed_timestamp_nanos());
}

TEST_F(EventMetricProducerTest, TestEventsWithNonSlicedCondition) {
    int64_t bucketStartTimeNs = 10000000000;
    int64_t eventStartTimeNs = bucketStartTimeNs + 1;
    int64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL;

    EventMetric metric;
    metric.set_id(1);
    metric.set_condition(StringToId("SCREEN_ON"));

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    CreateNoValuesLogEvent(&event1, 1 /*tagId*/, bucketStartTimeNs + 1);

    LogEvent event2(/*uid=*/0, /*pid=*/0);
    CreateNoValuesLogEvent(&event2, 1 /*tagId*/, bucketStartTimeNs + 10);

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);

    EventMetricProducer eventProducer(kConfigKey, metric, 0 /*condition index*/,
                                      {ConditionState::kUnknown}, wizard, protoHash,
                                      bucketStartTimeNs, provider);

    eventProducer.onConditionChanged(true /*condition*/, bucketStartTimeNs);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);

    eventProducer.onConditionChanged(false /*condition*/, bucketStartTimeNs + 2);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 20);
    EXPECT_TRUE(report.has_event_metrics());
    backfillAggregatedAtoms(&report);
    ASSERT_EQ(1, report.event_metrics().data_size());
    EXPECT_EQ(bucketStartTimeNs + 1, report.event_metrics().data(0).elapsed_timestamp_nanos());
}

TEST_F(EventMetricProducerTest, TestEventsWithSlicedCondition) {
    int64_t bucketStartTimeNs = 10000000000;
    int64_t bucketSizeNs = 30 * 1000 * 1000 * 1000LL;

    int tagId = 1;
    int conditionTagId = 2;

    EventMetric metric;
    metric.set_id(1);
    metric.set_condition(StringToId("APP_IN_BACKGROUND_PER_UID_AND_SCREEN_ON"));
    MetricConditionLink* link = metric.add_links();
    link->set_condition(StringToId("APP_IN_BACKGROUND_PER_UID"));
    buildSimpleAtomFieldMatcher(tagId, 1, link->mutable_fields_in_what());
    buildSimpleAtomFieldMatcher(conditionTagId, 2, link->mutable_fields_in_condition());

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, 1 /*tagId*/, bucketStartTimeNs + 1, "111");
    ConditionKey key1;
    key1[StringToId("APP_IN_BACKGROUND_PER_UID")] = {
            getMockedDimensionKey(conditionTagId, 2, "111")};

    LogEvent event2(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event2, 1 /*tagId*/, bucketStartTimeNs + 10, "222");
    ConditionKey key2;
    key2[StringToId("APP_IN_BACKGROUND_PER_UID")] = {
            getMockedDimensionKey(conditionTagId, 2, "222")};

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    // Condition is false for first event.
    EXPECT_CALL(*wizard, query(_, key1, _)).WillOnce(Return(ConditionState::kFalse));
    // Condition is true for second event.
    EXPECT_CALL(*wizard, query(_, key2, _)).WillOnce(Return(ConditionState::kTrue));
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);

    EventMetricProducer eventProducer(kConfigKey, metric, 0 /*condition index*/,
                                      {ConditionState::kUnknown}, wizard, protoHash,
                                      bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 20);
    backfillAggregatedAtoms(&report);
    EXPECT_TRUE(report.has_event_metrics());
    ASSERT_EQ(1, report.event_metrics().data_size());
    EXPECT_EQ(bucketStartTimeNs + 10, report.event_metrics().data(0).elapsed_timestamp_nanos());
}

TEST_F(EventMetricProducerTest, TestOneAtomTagAggregatedEvents) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");
    LogEvent event2(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event2, tagId, bucketStartTimeNs + 20, "111");
    LogEvent event3(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event3, tagId, bucketStartTimeNs + 30, "111");

    LogEvent event4(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event4, tagId, bucketStartTimeNs + 40, "222");

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      wizard, protoHash, bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event3);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event4);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 50);
    EXPECT_TRUE(report.has_event_metrics());
    ASSERT_EQ(2, report.event_metrics().data_size());

    for (EventMetricData metricData : report.event_metrics().data()) {
        AggregatedAtomInfo atomInfo = metricData.aggregated_atom_info();
        if (atomInfo.elapsed_timestamp_nanos_size() == 1) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 40);
        } else if (atomInfo.elapsed_timestamp_nanos_size() == 3) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 10);
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(1), bucketStartTimeNs + 20);
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(2), bucketStartTimeNs + 30);
        } else {
            FAIL();
        }
    }
}

TEST_F(EventMetricProducerTest, TestBytesFieldAggregatedEvents) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    vector<uint8_t> bytesField1{10, 20, 30};
    vector<uint8_t> bytesField2{10, 20, 30, 40};
    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111", &bytesField1);
    LogEvent event2(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event2, tagId, bucketStartTimeNs + 20, "111", &bytesField1);
    LogEvent event3(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event3, tagId, bucketStartTimeNs + 30, "111", &bytesField1);

    LogEvent event4(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event4, tagId, bucketStartTimeNs + 40, "111", &bytesField2);

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      wizard, protoHash, bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event3);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event4);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 50);
    EXPECT_TRUE(report.has_event_metrics());
    ASSERT_EQ(2, report.event_metrics().data_size());

    for (EventMetricData metricData : report.event_metrics().data()) {
        AggregatedAtomInfo atomInfo = metricData.aggregated_atom_info();
        if (atomInfo.elapsed_timestamp_nanos_size() == 1) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 40);
        } else if (atomInfo.elapsed_timestamp_nanos_size() == 3) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 10);
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(1), bucketStartTimeNs + 20);
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(2), bucketStartTimeNs + 30);
        } else {
            FAIL();
        }
    }
}

TEST_F(EventMetricProducerTest, TestTwoAtomTagAggregatedEvents) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;
    int tagId2 = 0;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");
    LogEvent event2(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event2, tagId, bucketStartTimeNs + 20, "111");

    LogEvent event3(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event3, tagId2, bucketStartTimeNs + 40, "222");

    sp<MockConditionWizard> wizard = new NaggyMock<MockConditionWizard>();
    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      wizard, protoHash, bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event2);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event3);

    // Check dump report content.
    StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 50);

    EXPECT_TRUE(report.has_event_metrics());
    ASSERT_EQ(2, report.event_metrics().data_size());

    for (EventMetricData metricData : report.event_metrics().data()) {
        AggregatedAtomInfo atomInfo = metricData.aggregated_atom_info();
        if (atomInfo.elapsed_timestamp_nanos_size() == 1) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 40);
        } else if (atomInfo.elapsed_timestamp_nanos_size() == 2) {
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(0), bucketStartTimeNs + 10);
            EXPECT_EQ(atomInfo.elapsed_timestamp_nanos(1), bucketStartTimeNs + 20);
        } else {
            FAIL();
        }
    }
}

TEST_F(EventMetricProducerTest, TestCorruptedDataReason_OnDumpReport) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");

    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      nullptr, protoHash, bucketStartTimeNs, provider);

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 50);

        EXPECT_TRUE(report.has_event_metrics());
        EXPECT_EQ(1, report.event_metrics().data_size());
        ASSERT_EQ(1, report.data_corrupted_reason_size());
        ASSERT_EQ(DATA_CORRUPTED_SOCKET_LOSS, report.data_corrupted_reason()[0]);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kNone);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kNone);
    }

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 150);

        EXPECT_TRUE(report.has_event_metrics());
        EXPECT_EQ(1, report.event_metrics().data_size());
        ASSERT_EQ(1, report.data_corrupted_reason_size());
        ASSERT_EQ(DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW, report.data_corrupted_reason()[0]);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kNone);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kNone);
    }

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 250);

        EXPECT_TRUE(report.has_event_metrics());
        EXPECT_EQ(1, report.event_metrics().data_size());
        EXPECT_EQ(2, report.data_corrupted_reason_size());
        EXPECT_THAT(report.data_corrupted_reason(),
                    ElementsAre(DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW, DATA_CORRUPTED_SOCKET_LOSS));
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kNone);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kNone);
    }
}

TEST_F(EventMetricProducerTest, TestCorruptedDataReason_OnDropData) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");

    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      nullptr, protoHash, bucketStartTimeNs, provider);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.dropData(bucketStartTimeNs + 100);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    eventProducer.dropData(bucketStartTimeNs + 200);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    eventProducer.dropData(bucketStartTimeNs + 300);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kCondition);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.dropData(bucketStartTimeNs + 400);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kCondition);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);

    eventProducer.dropData(bucketStartTimeNs + 500);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
}

TEST_F(EventMetricProducerTest, TestCorruptedDataReason_OnClearPastBuckets) {
    int64_t bucketStartTimeNs = 10000000000;
    int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");

    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      nullptr, protoHash, bucketStartTimeNs, provider);
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.clearPastBuckets(bucketStartTimeNs + 100);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    eventProducer.clearPastBuckets(bucketStartTimeNs + 200);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kWhat);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    eventProducer.clearPastBuckets(bucketStartTimeNs + 300);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kCondition);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.clearPastBuckets(bucketStartTimeNs + 400);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kCondition);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);

    eventProducer.clearPastBuckets(bucketStartTimeNs + 500);

    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
}

TEST_F(EventMetricProducerTest, TestCorruptedDataReason_UnrecoverableLossOfCondition) {
    const int64_t bucketStartTimeNs = 10000000000;
    const int tagId = 1;

    EventMetric metric;
    metric.set_id(1);

    sp<MockConfigMetadataProvider> provider = makeMockConfigMetadataProvider(/*enabled=*/false);
    EventMetricProducer eventProducer(kConfigKey, metric, -1 /*-1 meaning no condition*/, {},
                                      nullptr, protoHash, bucketStartTimeNs, provider);

    LogEvent event1(/*uid=*/0, /*pid=*/0);
    makeLogEvent(&event1, tagId, bucketStartTimeNs + 10, "111");
    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kNone);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_SOCKET_LOSS,
                                        MetricProducer::LostAtomType::kCondition);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 50);

        EXPECT_TRUE(report.has_event_metrics());
        ASSERT_EQ(1, report.event_metrics().data_size());
        EXPECT_EQ(1, report.data_corrupted_reason_size());
        EXPECT_THAT(report.data_corrupted_reason(), ElementsAre(DATA_CORRUPTED_SOCKET_LOSS));
        // confirm that unrecoverable loss status persist after dumpReport
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kUnrecoverable);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kNone);
    }

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kWhat);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kResetOnDump);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 150);
        EXPECT_TRUE(report.has_event_metrics());
        EXPECT_EQ(1, report.event_metrics().data_size());
        EXPECT_EQ(2, report.data_corrupted_reason_size());
        EXPECT_THAT(report.data_corrupted_reason(),
                    ElementsAre(DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW, DATA_CORRUPTED_SOCKET_LOSS));
        // confirm that unrecoverable loss status persist after dumpReport
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kUnrecoverable);
        // confirm that kResetOnDump loss status is reset after dumpReport
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kNone);
    }

    eventProducer.onMatchedLogEvent(1 /*matcher index*/, event1);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kNone);
    eventProducer.onMatchedLogEventLost(tagId, DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW,
                                        MetricProducer::LostAtomType::kCondition);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
              MetricProducer::DataCorruptionSeverity::kUnrecoverable);

    {
        // Check dump report content.
        StatsLogReport report = onDumpReport(eventProducer, bucketStartTimeNs + 150);

        EXPECT_TRUE(report.has_event_metrics());
        EXPECT_EQ(1, report.event_metrics().data_size());
        EXPECT_EQ(2, report.data_corrupted_reason_size());
        EXPECT_THAT(report.data_corrupted_reason(),
                    ElementsAre(DATA_CORRUPTED_EVENT_QUEUE_OVERFLOW, DATA_CORRUPTED_SOCKET_LOSS));
        // confirm that unrecoverable loss status persist after dumpReport
        EXPECT_EQ(eventProducer.mDataCorruptedDueToSocketLoss,
                  MetricProducer::DataCorruptionSeverity::kUnrecoverable);
        EXPECT_EQ(eventProducer.mDataCorruptedDueToQueueOverflow,
                  MetricProducer::DataCorruptionSeverity::kUnrecoverable);
    }
}

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