statsd/tests/LogEvent_test.cpp - platform/packages/modules/StatsD - Git at Google

 // 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/logd/LogEvent.h"

 #include <gtest/gtest.h>

 #include "frameworks/proto_logging/stats/atoms.pb.h"
 #include "frameworks/proto_logging/stats/enums/stats/launcher/launcher.pb.h"
 #include "log/log_event_list.h"
 #include "stats_event.h"

 #ifdef __ANDROID__

 namespace android {
 namespace os {
 namespace statsd {

 using std::string;
 using std::vector;
 using util::ProtoOutputStream;
 using util::ProtoReader;

 namespace {

 Field getField(int32_t tag, const vector<int32_t>& pos, int32_t depth, const vector<bool>& last) {
     Field f(tag, (int32_t*)pos.data(), depth);

     // For loop starts at 1 because the last field at depth 0 is not decorated.
     for (int i = 1; i < depth; i++) {
         if (last[i]) f.decorateLastPos(i);
     }

     return f;
 }

 void createIntWithBoolAnnotationLogEvent(LogEvent* logEvent, uint8_t annotationId,
                                          bool annotationValue) {
     AStatsEvent* statsEvent = AStatsEvent_obtain();
     AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
     AStatsEvent_writeInt32(statsEvent, 10);
     AStatsEvent_addBoolAnnotation(statsEvent, annotationId, annotationValue);
     AStatsEvent_build(statsEvent);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
     EXPECT_TRUE(logEvent->parseBuffer(buf, size));

     AStatsEvent_release(statsEvent);
 }

 void createIntWithIntAnnotationLogEvent(LogEvent* logEvent, uint8_t annotationId,
                                         int annotationValue) {
     AStatsEvent* statsEvent = AStatsEvent_obtain();
     AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
     AStatsEvent_writeInt32(statsEvent, 10);
     AStatsEvent_addInt32Annotation(statsEvent, annotationId, annotationValue);
     AStatsEvent_build(statsEvent);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
     EXPECT_TRUE(logEvent->parseBuffer(buf, size));

     AStatsEvent_release(statsEvent);
 }

 }  // anonymous namespace

 TEST(LogEventTest, TestPrimitiveParsing) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);
     AStatsEvent_writeInt32(event, 10);
     AStatsEvent_writeInt64(event, 0x123456789);
     AStatsEvent_writeFloat(event, 2.0);
     AStatsEvent_writeBool(event, true);
     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));

     EXPECT_EQ(100, logEvent.GetTagId());
     EXPECT_EQ(1000, logEvent.GetUid());
     EXPECT_EQ(1001, logEvent.GetPid());
     EXPECT_FALSE(logEvent.hasAttributionChain());

     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(4, values.size());

     const FieldValue& int32Item = values[0];
     Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
     EXPECT_EQ(expectedField, int32Item.mField);
     EXPECT_EQ(Type::INT, int32Item.mValue.getType());
     EXPECT_EQ(10, int32Item.mValue.int_value);

     const FieldValue& int64Item = values[1];
     expectedField = getField(100, {2, 1, 1}, 0, {false, false, false});
     EXPECT_EQ(expectedField, int64Item.mField);
     EXPECT_EQ(Type::LONG, int64Item.mValue.getType());
     EXPECT_EQ(0x123456789, int64Item.mValue.long_value);

     const FieldValue& floatItem = values[2];
     expectedField = getField(100, {3, 1, 1}, 0, {false, false, false});
     EXPECT_EQ(expectedField, floatItem.mField);
     EXPECT_EQ(Type::FLOAT, floatItem.mValue.getType());
     EXPECT_EQ(2.0, floatItem.mValue.float_value);

     const FieldValue& boolItem = values[3];
     expectedField = getField(100, {4, 1, 1}, 0, {true, false, false});
     EXPECT_EQ(expectedField, boolItem.mField);
     EXPECT_EQ(Type::INT, boolItem.mValue.getType());  // FieldValue does not support boolean type
     EXPECT_EQ(1, boolItem.mValue.int_value);

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestStringAndByteArrayParsing) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);
     string str = "test";
     AStatsEvent_writeString(event, str.c_str());
     AStatsEvent_writeByteArray(event, (uint8_t*)str.c_str(), str.length());
     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));

     EXPECT_EQ(100, logEvent.GetTagId());
     EXPECT_EQ(1000, logEvent.GetUid());
     EXPECT_EQ(1001, logEvent.GetPid());
     EXPECT_FALSE(logEvent.hasAttributionChain());

     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(2, values.size());

     const FieldValue& stringItem = values[0];
     Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
     EXPECT_EQ(expectedField, stringItem.mField);
     EXPECT_EQ(Type::STRING, stringItem.mValue.getType());
     EXPECT_EQ(str, stringItem.mValue.str_value);

     const FieldValue& storageItem = values[1];
     expectedField = getField(100, {2, 1, 1}, 0, {true, false, false});
     EXPECT_EQ(expectedField, storageItem.mField);
     EXPECT_EQ(Type::STORAGE, storageItem.mValue.getType());
     vector<uint8_t> expectedValue = {'t', 'e', 's', 't'};
     EXPECT_EQ(expectedValue, storageItem.mValue.storage_value);

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestEmptyString) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);
     string empty = "";
     AStatsEvent_writeString(event, empty.c_str());
     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));

     EXPECT_EQ(100, logEvent.GetTagId());
     EXPECT_EQ(1000, logEvent.GetUid());
     EXPECT_EQ(1001, logEvent.GetPid());
     EXPECT_FALSE(logEvent.hasAttributionChain());

     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(1, values.size());

     const FieldValue& item = values[0];
     Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
     EXPECT_EQ(expectedField, item.mField);
     EXPECT_EQ(Type::STRING, item.mValue.getType());
     EXPECT_EQ(empty, item.mValue.str_value);

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestByteArrayWithNullCharacter) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);
     uint8_t message[] = {'\t', 'e', '\0', 's', 't'};
     AStatsEvent_writeByteArray(event, message, 5);
     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));

     EXPECT_EQ(100, logEvent.GetTagId());
     EXPECT_EQ(1000, logEvent.GetUid());
     EXPECT_EQ(1001, logEvent.GetPid());

     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(1, values.size());

     const FieldValue& item = values[0];
     Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
     EXPECT_EQ(expectedField, item.mField);
     EXPECT_EQ(Type::STORAGE, item.mValue.getType());
     vector<uint8_t> expectedValue(message, message + 5);
     EXPECT_EQ(expectedValue, item.mValue.storage_value);

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestAttributionChain) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);

     string tag1 = "tag1";
     string tag2 = "tag2";

     uint32_t uids[] = {1001, 1002};
     const char* tags[] = {tag1.c_str(), tag2.c_str()};

     AStatsEvent_writeAttributionChain(event, uids, tags, 2);
     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));

     EXPECT_EQ(100, logEvent.GetTagId());
     EXPECT_EQ(1000, logEvent.GetUid());
     EXPECT_EQ(1001, logEvent.GetPid());

     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(4, values.size());  // 2 per attribution node

     std::pair<int, int> attrIndexRange;
     EXPECT_TRUE(logEvent.hasAttributionChain(&attrIndexRange));
     EXPECT_EQ(0, attrIndexRange.first);
     EXPECT_EQ(3, attrIndexRange.second);

     // Check first attribution node
     const FieldValue& uid1Item = values[0];
     Field expectedField = getField(100, {1, 1, 1}, 2, {true, false, false});
     EXPECT_EQ(expectedField, uid1Item.mField);
     EXPECT_EQ(Type::INT, uid1Item.mValue.getType());
     EXPECT_EQ(1001, uid1Item.mValue.int_value);

     const FieldValue& tag1Item = values[1];
     expectedField = getField(100, {1, 1, 2}, 2, {true, false, true});
     EXPECT_EQ(expectedField, tag1Item.mField);
     EXPECT_EQ(Type::STRING, tag1Item.mValue.getType());
     EXPECT_EQ(tag1, tag1Item.mValue.str_value);

     // Check second attribution nodes
     const FieldValue& uid2Item = values[2];
     expectedField = getField(100, {1, 2, 1}, 2, {true, true, false});
     EXPECT_EQ(expectedField, uid2Item.mField);
     EXPECT_EQ(Type::INT, uid2Item.mValue.getType());
     EXPECT_EQ(1002, uid2Item.mValue.int_value);

     const FieldValue& tag2Item = values[3];
     expectedField = getField(100, {1, 2, 2}, 2, {true, true, true});
     EXPECT_EQ(expectedField, tag2Item.mField);
     EXPECT_EQ(Type::STRING, tag2Item.mValue.getType());
     EXPECT_EQ(tag2, tag2Item.mValue.str_value);

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestAnnotationIdIsUid) {
     LogEvent event(/*uid=*/0, /*pid=*/0);
     createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_IS_UID, true);

     const vector<FieldValue>& values = event.getValues();
     ASSERT_EQ(values.size(), 1);
     EXPECT_EQ(event.getUidFieldIndex(), 0);
 }

 TEST(LogEventTest, TestAnnotationIdStateNested) {
     LogEvent event(/*uid=*/0, /*pid=*/0);
     createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_STATE_NESTED, true);

     const vector<FieldValue>& values = event.getValues();
     ASSERT_EQ(values.size(), 1);
     EXPECT_TRUE(values[0].mAnnotations.isNested());
 }

 TEST(LogEventTest, TestPrimaryFieldAnnotation) {
     LogEvent event(/*uid=*/0, /*pid=*/0);
     createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_PRIMARY_FIELD, true);

     const vector<FieldValue>& values = event.getValues();
     ASSERT_EQ(values.size(), 1);
     EXPECT_TRUE(values[0].mAnnotations.isPrimaryField());
 }

 TEST(LogEventTest, TestExclusiveStateAnnotation) {
     LogEvent event(/*uid=*/0, /*pid=*/0);
     createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_EXCLUSIVE_STATE, true);

     const vector<FieldValue>& values = event.getValues();
     ASSERT_EQ(values.size(), 1);
     EXPECT_TRUE(values[0].mAnnotations.isExclusiveState());
 }

 TEST(LogEventTest, TestPrimaryFieldFirstUidAnnotation) {
     // Event has 10 ints and then an attribution chain
     int numInts = 10;
     int firstUidInChainIndex = numInts;
     string tag1 = "tag1";
     string tag2 = "tag2";
     uint32_t uids[] = {1001, 1002};
     const char* tags[] = {tag1.c_str(), tag2.c_str()};

     // Construct AStatsEvent
     AStatsEvent* statsEvent = AStatsEvent_obtain();
     AStatsEvent_setAtomId(statsEvent, 100);
     for (int i = 0; i < numInts; i++) {
         AStatsEvent_writeInt32(statsEvent, 10);
     }
     AStatsEvent_writeAttributionChain(statsEvent, uids, tags, 2);
     AStatsEvent_addBoolAnnotation(statsEvent, ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, true);
     AStatsEvent_build(statsEvent);

     // Construct LogEvent
     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
     LogEvent logEvent(/*uid=*/0, /*pid=*/0);
     EXPECT_TRUE(logEvent.parseBuffer(buf, size));
     AStatsEvent_release(statsEvent);

     // Check annotation
     const vector<FieldValue>& values = logEvent.getValues();
     ASSERT_EQ(values.size(), numInts + 4);
     EXPECT_TRUE(values[firstUidInChainIndex].mAnnotations.isPrimaryField());
 }

 TEST(LogEventTest, TestResetStateAnnotation) {
     int32_t resetState = 10;
     LogEvent event(/*uid=*/0, /*pid=*/0);
     createIntWithIntAnnotationLogEvent(&event, ANNOTATION_ID_TRIGGER_STATE_RESET, resetState);

     const vector<FieldValue>& values = event.getValues();
     ASSERT_EQ(values.size(), 1);
     EXPECT_EQ(event.getResetState(), resetState);
 }

 TEST(LogEventTest, TestExclusiveStateAnnotationAfterTooManyFields) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);

     const unsigned int numAttributionNodes = 64;

     uint32_t uids[numAttributionNodes];
     const char* tags[numAttributionNodes];

     for (unsigned int i = 1; i <= numAttributionNodes; i++) {
         uids[i-1] = i;
         tags[i-1] = std::to_string(i).c_str();
     }

     AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);
     AStatsEvent_writeInt32(event, 1);
     AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_EXCLUSIVE_STATE, true);

     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_FALSE(logEvent.parseBuffer(buf, size));
     EXPECT_EQ(-1, logEvent.getExclusiveStateFieldIndex());

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestUidAnnotationAfterTooManyFields) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);

     const unsigned int numAttributionNodes = 64;

     uint32_t uids[numAttributionNodes];
     const char* tags[numAttributionNodes];

     for (unsigned int i = 1; i <= numAttributionNodes; i++) {
         uids[i-1] = i;
         tags[i-1] = std::to_string(i).c_str();
     }

     AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);
     AStatsEvent_writeInt32(event, 1);
     AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_IS_UID, true);

     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_FALSE(logEvent.parseBuffer(buf, size));
     EXPECT_EQ(-1, logEvent.getUidFieldIndex());

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestAttributionChainEndIndexAfterTooManyFields) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);

     const unsigned int numAttributionNodes = 65;

     uint32_t uids[numAttributionNodes];
     const char* tags[numAttributionNodes];

     for (unsigned int i = 1; i <= numAttributionNodes; i++) {
         uids[i-1] = i;
         tags[i-1] = std::to_string(i).c_str();
     }

     AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);

     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_FALSE(logEvent.parseBuffer(buf, size));
     EXPECT_FALSE(logEvent.hasAttributionChain());

     AStatsEvent_release(event);
 }

 TEST(LogEventTest, TestEmptyAttributionChainWithPrimaryFieldFirstUidAnnotation) {
     AStatsEvent* event = AStatsEvent_obtain();
     AStatsEvent_setAtomId(event, 100);

     uint32_t uids[] = {};
     const char* tags[] = {};

     AStatsEvent_writeInt32(event, 10);
     AStatsEvent_writeAttributionChain(event, uids, tags, 0);
     AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, true);

     AStatsEvent_build(event);

     size_t size;
     uint8_t* buf = AStatsEvent_getBuffer(event, &size);

     LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
     EXPECT_FALSE(logEvent.parseBuffer(buf, size));

     AStatsEvent_release(event);
 }

 }  // namespace statsd
 }  // namespace os
 }  // namespace android
 #else
 GTEST_LOG_(INFO) << "This test does nothing.\n";
 #endif
	// 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/logd/LogEvent.h"

	#include <gtest/gtest.h>

	#include "frameworks/proto_logging/stats/atoms.pb.h"
	#include "frameworks/proto_logging/stats/enums/stats/launcher/launcher.pb.h"
	#include "log/log_event_list.h"
	#include "stats_event.h"

	#ifdef __ANDROID__

	namespace android {
	namespace os {
	namespace statsd {

	using std::string;
	using std::vector;
	using util::ProtoOutputStream;
	using util::ProtoReader;

	namespace {

	Field getField(int32_t tag, const vector<int32_t>& pos, int32_t depth, const vector<bool>& last) {
	Field f(tag, (int32_t*)pos.data(), depth);

	// For loop starts at 1 because the last field at depth 0 is not decorated.
	for (int i = 1; i < depth; i++) {
	if (last[i]) f.decorateLastPos(i);
	}

	return f;
	}

	void createIntWithBoolAnnotationLogEvent(LogEvent* logEvent, uint8_t annotationId,
	bool annotationValue) {
	AStatsEvent* statsEvent = AStatsEvent_obtain();
	AStatsEvent_setAtomId(statsEvent, /atomId=/100);
	AStatsEvent_writeInt32(statsEvent, 10);
	AStatsEvent_addBoolAnnotation(statsEvent, annotationId, annotationValue);
	AStatsEvent_build(statsEvent);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
	EXPECT_TRUE(logEvent->parseBuffer(buf, size));

	AStatsEvent_release(statsEvent);
	}

	void createIntWithIntAnnotationLogEvent(LogEvent* logEvent, uint8_t annotationId,
	int annotationValue) {
	AStatsEvent* statsEvent = AStatsEvent_obtain();
	AStatsEvent_setAtomId(statsEvent, /atomId=/100);
	AStatsEvent_writeInt32(statsEvent, 10);
	AStatsEvent_addInt32Annotation(statsEvent, annotationId, annotationValue);
	AStatsEvent_build(statsEvent);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
	EXPECT_TRUE(logEvent->parseBuffer(buf, size));

	AStatsEvent_release(statsEvent);
	}

	} // anonymous namespace

	TEST(LogEventTest, TestPrimitiveParsing) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);
	AStatsEvent_writeInt32(event, 10);
	AStatsEvent_writeInt64(event, 0x123456789);
	AStatsEvent_writeFloat(event, 2.0);
	AStatsEvent_writeBool(event, true);
	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));

	EXPECT_EQ(100, logEvent.GetTagId());
	EXPECT_EQ(1000, logEvent.GetUid());
	EXPECT_EQ(1001, logEvent.GetPid());
	EXPECT_FALSE(logEvent.hasAttributionChain());

	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(4, values.size());

	const FieldValue& int32Item = values[0];
	Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
	EXPECT_EQ(expectedField, int32Item.mField);
	EXPECT_EQ(Type::INT, int32Item.mValue.getType());
	EXPECT_EQ(10, int32Item.mValue.int_value);

	const FieldValue& int64Item = values[1];
	expectedField = getField(100, {2, 1, 1}, 0, {false, false, false});
	EXPECT_EQ(expectedField, int64Item.mField);
	EXPECT_EQ(Type::LONG, int64Item.mValue.getType());
	EXPECT_EQ(0x123456789, int64Item.mValue.long_value);

	const FieldValue& floatItem = values[2];
	expectedField = getField(100, {3, 1, 1}, 0, {false, false, false});
	EXPECT_EQ(expectedField, floatItem.mField);
	EXPECT_EQ(Type::FLOAT, floatItem.mValue.getType());
	EXPECT_EQ(2.0, floatItem.mValue.float_value);

	const FieldValue& boolItem = values[3];
	expectedField = getField(100, {4, 1, 1}, 0, {true, false, false});
	EXPECT_EQ(expectedField, boolItem.mField);
	EXPECT_EQ(Type::INT, boolItem.mValue.getType()); // FieldValue does not support boolean type
	EXPECT_EQ(1, boolItem.mValue.int_value);

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestStringAndByteArrayParsing) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);
	string str = "test";
	AStatsEvent_writeString(event, str.c_str());
	AStatsEvent_writeByteArray(event, (uint8_t*)str.c_str(), str.length());
	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));

	EXPECT_EQ(100, logEvent.GetTagId());
	EXPECT_EQ(1000, logEvent.GetUid());
	EXPECT_EQ(1001, logEvent.GetPid());
	EXPECT_FALSE(logEvent.hasAttributionChain());

	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(2, values.size());

	const FieldValue& stringItem = values[0];
	Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
	EXPECT_EQ(expectedField, stringItem.mField);
	EXPECT_EQ(Type::STRING, stringItem.mValue.getType());
	EXPECT_EQ(str, stringItem.mValue.str_value);

	const FieldValue& storageItem = values[1];
	expectedField = getField(100, {2, 1, 1}, 0, {true, false, false});
	EXPECT_EQ(expectedField, storageItem.mField);
	EXPECT_EQ(Type::STORAGE, storageItem.mValue.getType());
	vector<uint8_t> expectedValue = {'t', 'e', 's', 't'};
	EXPECT_EQ(expectedValue, storageItem.mValue.storage_value);

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestEmptyString) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);
	string empty = "";
	AStatsEvent_writeString(event, empty.c_str());
	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));

	EXPECT_EQ(100, logEvent.GetTagId());
	EXPECT_EQ(1000, logEvent.GetUid());
	EXPECT_EQ(1001, logEvent.GetPid());
	EXPECT_FALSE(logEvent.hasAttributionChain());

	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(1, values.size());

	const FieldValue& item = values[0];
	Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
	EXPECT_EQ(expectedField, item.mField);
	EXPECT_EQ(Type::STRING, item.mValue.getType());
	EXPECT_EQ(empty, item.mValue.str_value);

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestByteArrayWithNullCharacter) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);
	uint8_t message[] = {'\t', 'e', '\0', 's', 't'};
	AStatsEvent_writeByteArray(event, message, 5);
	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));

	EXPECT_EQ(100, logEvent.GetTagId());
	EXPECT_EQ(1000, logEvent.GetUid());
	EXPECT_EQ(1001, logEvent.GetPid());

	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(1, values.size());

	const FieldValue& item = values[0];
	Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
	EXPECT_EQ(expectedField, item.mField);
	EXPECT_EQ(Type::STORAGE, item.mValue.getType());
	vector<uint8_t> expectedValue(message, message + 5);
	EXPECT_EQ(expectedValue, item.mValue.storage_value);

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestAttributionChain) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);

	string tag1 = "tag1";
	string tag2 = "tag2";

	uint32_t uids[] = {1001, 1002};
	const char* tags[] = {tag1.c_str(), tag2.c_str()};

	AStatsEvent_writeAttributionChain(event, uids, tags, 2);
	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));

	EXPECT_EQ(100, logEvent.GetTagId());
	EXPECT_EQ(1000, logEvent.GetUid());
	EXPECT_EQ(1001, logEvent.GetPid());

	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(4, values.size()); // 2 per attribution node

	std::pair<int, int> attrIndexRange;
	EXPECT_TRUE(logEvent.hasAttributionChain(&attrIndexRange));
	EXPECT_EQ(0, attrIndexRange.first);
	EXPECT_EQ(3, attrIndexRange.second);

	// Check first attribution node
	const FieldValue& uid1Item = values[0];
	Field expectedField = getField(100, {1, 1, 1}, 2, {true, false, false});
	EXPECT_EQ(expectedField, uid1Item.mField);
	EXPECT_EQ(Type::INT, uid1Item.mValue.getType());
	EXPECT_EQ(1001, uid1Item.mValue.int_value);

	const FieldValue& tag1Item = values[1];
	expectedField = getField(100, {1, 1, 2}, 2, {true, false, true});
	EXPECT_EQ(expectedField, tag1Item.mField);
	EXPECT_EQ(Type::STRING, tag1Item.mValue.getType());
	EXPECT_EQ(tag1, tag1Item.mValue.str_value);

	// Check second attribution nodes
	const FieldValue& uid2Item = values[2];
	expectedField = getField(100, {1, 2, 1}, 2, {true, true, false});
	EXPECT_EQ(expectedField, uid2Item.mField);
	EXPECT_EQ(Type::INT, uid2Item.mValue.getType());
	EXPECT_EQ(1002, uid2Item.mValue.int_value);

	const FieldValue& tag2Item = values[3];
	expectedField = getField(100, {1, 2, 2}, 2, {true, true, true});
	EXPECT_EQ(expectedField, tag2Item.mField);
	EXPECT_EQ(Type::STRING, tag2Item.mValue.getType());
	EXPECT_EQ(tag2, tag2Item.mValue.str_value);

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestAnnotationIdIsUid) {
	LogEvent event(/uid=/0, /pid=/0);
	createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_IS_UID, true);

	const vector<FieldValue>& values = event.getValues();
	ASSERT_EQ(values.size(), 1);
	EXPECT_EQ(event.getUidFieldIndex(), 0);
	}

	TEST(LogEventTest, TestAnnotationIdStateNested) {
	LogEvent event(/uid=/0, /pid=/0);
	createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_STATE_NESTED, true);

	const vector<FieldValue>& values = event.getValues();
	ASSERT_EQ(values.size(), 1);
	EXPECT_TRUE(values[0].mAnnotations.isNested());
	}

	TEST(LogEventTest, TestPrimaryFieldAnnotation) {
	LogEvent event(/uid=/0, /pid=/0);
	createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_PRIMARY_FIELD, true);

	const vector<FieldValue>& values = event.getValues();
	ASSERT_EQ(values.size(), 1);
	EXPECT_TRUE(values[0].mAnnotations.isPrimaryField());
	}

	TEST(LogEventTest, TestExclusiveStateAnnotation) {
	LogEvent event(/uid=/0, /pid=/0);
	createIntWithBoolAnnotationLogEvent(&event, ANNOTATION_ID_EXCLUSIVE_STATE, true);

	const vector<FieldValue>& values = event.getValues();
	ASSERT_EQ(values.size(), 1);
	EXPECT_TRUE(values[0].mAnnotations.isExclusiveState());
	}

	TEST(LogEventTest, TestPrimaryFieldFirstUidAnnotation) {
	// Event has 10 ints and then an attribution chain
	int numInts = 10;
	int firstUidInChainIndex = numInts;
	string tag1 = "tag1";
	string tag2 = "tag2";
	uint32_t uids[] = {1001, 1002};
	const char* tags[] = {tag1.c_str(), tag2.c_str()};

	// Construct AStatsEvent
	AStatsEvent* statsEvent = AStatsEvent_obtain();
	AStatsEvent_setAtomId(statsEvent, 100);
	for (int i = 0; i < numInts; i++) {
	AStatsEvent_writeInt32(statsEvent, 10);
	}
	AStatsEvent_writeAttributionChain(statsEvent, uids, tags, 2);
	AStatsEvent_addBoolAnnotation(statsEvent, ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, true);
	AStatsEvent_build(statsEvent);

	// Construct LogEvent
	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
	LogEvent logEvent(/uid=/0, /pid=/0);
	EXPECT_TRUE(logEvent.parseBuffer(buf, size));
	AStatsEvent_release(statsEvent);

	// Check annotation
	const vector<FieldValue>& values = logEvent.getValues();
	ASSERT_EQ(values.size(), numInts + 4);
	EXPECT_TRUE(values[firstUidInChainIndex].mAnnotations.isPrimaryField());
	}

	TEST(LogEventTest, TestResetStateAnnotation) {
	int32_t resetState = 10;
	LogEvent event(/uid=/0, /pid=/0);
	createIntWithIntAnnotationLogEvent(&event, ANNOTATION_ID_TRIGGER_STATE_RESET, resetState);

	const vector<FieldValue>& values = event.getValues();
	ASSERT_EQ(values.size(), 1);
	EXPECT_EQ(event.getResetState(), resetState);
	}

	TEST(LogEventTest, TestExclusiveStateAnnotationAfterTooManyFields) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);

	const unsigned int numAttributionNodes = 64;

	uint32_t uids[numAttributionNodes];
	const char* tags[numAttributionNodes];

	for (unsigned int i = 1; i <= numAttributionNodes; i++) {
	uids[i-1] = i;
	tags[i-1] = std::to_string(i).c_str();
	}

	AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);
	AStatsEvent_writeInt32(event, 1);
	AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_EXCLUSIVE_STATE, true);

	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_FALSE(logEvent.parseBuffer(buf, size));
	EXPECT_EQ(-1, logEvent.getExclusiveStateFieldIndex());

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestUidAnnotationAfterTooManyFields) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);

	const unsigned int numAttributionNodes = 64;

	uint32_t uids[numAttributionNodes];
	const char* tags[numAttributionNodes];

	for (unsigned int i = 1; i <= numAttributionNodes; i++) {
	uids[i-1] = i;
	tags[i-1] = std::to_string(i).c_str();
	}

	AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);
	AStatsEvent_writeInt32(event, 1);
	AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_IS_UID, true);

	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_FALSE(logEvent.parseBuffer(buf, size));
	EXPECT_EQ(-1, logEvent.getUidFieldIndex());

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestAttributionChainEndIndexAfterTooManyFields) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);

	const unsigned int numAttributionNodes = 65;

	uint32_t uids[numAttributionNodes];
	const char* tags[numAttributionNodes];

	for (unsigned int i = 1; i <= numAttributionNodes; i++) {
	uids[i-1] = i;
	tags[i-1] = std::to_string(i).c_str();
	}

	AStatsEvent_writeAttributionChain(event, uids, tags, numAttributionNodes);

	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_FALSE(logEvent.parseBuffer(buf, size));
	EXPECT_FALSE(logEvent.hasAttributionChain());

	AStatsEvent_release(event);
	}

	TEST(LogEventTest, TestEmptyAttributionChainWithPrimaryFieldFirstUidAnnotation) {
	AStatsEvent* event = AStatsEvent_obtain();
	AStatsEvent_setAtomId(event, 100);

	uint32_t uids[] = {};
	const char* tags[] = {};

	AStatsEvent_writeInt32(event, 10);
	AStatsEvent_writeAttributionChain(event, uids, tags, 0);
	AStatsEvent_addBoolAnnotation(event, ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, true);

	AStatsEvent_build(event);

	size_t size;
	uint8_t* buf = AStatsEvent_getBuffer(event, &size);

	LogEvent logEvent(/uid=/1000, /pid=/1001);
	EXPECT_FALSE(logEvent.parseBuffer(buf, size));

	AStatsEvent_release(event);
	}

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