From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 02:47:55 +0200 Subject: Adding upstream version 124.0.1. Signed-off-by: Daniel Baumann --- dom/quota/test/gtest/TestTelemetry.cpp | 185 +++++++++++++++++++++++++++++++++ 1 file changed, 185 insertions(+) create mode 100644 dom/quota/test/gtest/TestTelemetry.cpp (limited to 'dom/quota/test/gtest/TestTelemetry.cpp') diff --git a/dom/quota/test/gtest/TestTelemetry.cpp b/dom/quota/test/gtest/TestTelemetry.cpp new file mode 100644 index 0000000000..7936a5a253 --- /dev/null +++ b/dom/quota/test/gtest/TestTelemetry.cpp @@ -0,0 +1,185 @@ +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set ts=8 sts=2 et sw=2 tw=80: */ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this file, + * You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#include "mozilla/dom/ScriptSettings.h" +#include "mozilla/dom/SimpleGlobalObject.h" +#include "mozilla/gtest/MozAssertions.h" +#include "nsITelemetry.h" +#include "QuotaManagerDependencyFixture.h" + +namespace mozilla::dom::quota::test { + +namespace { + +namespace js_helpers { + +void CallFunction(JSContext* aCx, JS::Handle aValue, + const char* aName, JS::MutableHandle aResult) { + JS::Rooted obj(aCx, &aValue.toObject()); + + JS::Rooted result(aCx); + ASSERT_TRUE(JS_CallFunctionName(aCx, obj, aName, + JS::HandleValueArray::empty(), &result)); + + aResult.set(result); +} + +void HasProperty(JSContext* aCx, JS::Handle aValue, + const char* aName, bool* aResult) { + JS::Rooted obj(aCx, &aValue.toObject()); + + bool result; + ASSERT_TRUE(JS_HasProperty(aCx, obj, aName, &result)); + + *aResult = result; +} + +void GetProperty(JSContext* aCx, JS::Handle aValue, + const char* aName, JS::MutableHandle aResult) { + JS::Rooted obj(aCx, &aValue.toObject()); + + JS::Rooted result(aCx); + ASSERT_TRUE(JS_GetProperty(aCx, obj, aName, &result)); + + aResult.set(result); +} + +void Enumerate(JSContext* aCx, JS::Handle aValue, + JS::MutableHandle aResult) { + JS::Rooted obj(aCx, &aValue.toObject()); + ASSERT_TRUE(JS_Enumerate(aCx, obj, aResult)); +} + +void GetElement(JSContext* aCx, JS::Handle aValue, uint32_t aIndex, + JS::MutableHandle aResult) { + JS::Rooted obj(aCx, &aValue.toObject()); + + JS::Rooted result(aCx); + ASSERT_TRUE(JS_GetElement(aCx, obj, aIndex, &result)); + + aResult.set(result); +} + +void GetIndexValue(JSContext* aCx, JS::Handle aValues, + JS::PropertyKey aIndexId, uint32_t* aResult) { + nsAutoJSString indexId; + ASSERT_TRUE(indexId.init(aCx, aIndexId)); + + nsresult rv; + const auto index = indexId.ToInteger64(&rv); + ASSERT_NS_SUCCEEDED(rv); + + JS::Rooted element(aCx); + GetElement(aCx, aValues, index, &element); + + uint32_t value = 0; + ASSERT_TRUE(JS::ToUint32(aCx, element, &value)); + + *aResult = value; +} + +} // namespace js_helpers + +} // namespace + +TEST(DOM_Quota_Telemetry, ShutdownTime) +{ + nsCOMPtr telemetry = + do_GetService("@mozilla.org/base/telemetry;1"); + + JSObject* simpleGlobal = dom::SimpleGlobalObject::Create( + dom::SimpleGlobalObject::GlobalType::BindingDetail); + + JS::Rooted global(dom::RootingCx(), simpleGlobal); + + AutoJSAPI jsapi; + ASSERT_TRUE(jsapi.Init(global)); + + JSContext* cx = jsapi.cx(); + + JS::Rooted histogram(cx); + ASSERT_NS_SUCCEEDED(telemetry->GetKeyedHistogramById("QM_SHUTDOWN_TIME_V0"_ns, + cx, &histogram)); + + JS::Rooted dummy(cx); + ASSERT_NO_FATAL_FAILURE( + js_helpers::CallFunction(cx, histogram, "clear", &dummy)); + + ASSERT_NO_FATAL_FAILURE(QuotaManagerDependencyFixture::InitializeFixture()); + ASSERT_NO_FATAL_FAILURE(QuotaManagerDependencyFixture::ShutdownFixture()); + + JS::Rooted snapshot(cx); + ASSERT_NO_FATAL_FAILURE( + js_helpers::CallFunction(cx, histogram, "snapshot", &snapshot)); + + { + bool hasKey = false; + ASSERT_NO_FATAL_FAILURE( + js_helpers::HasProperty(cx, snapshot, "Normal", &hasKey)); + ASSERT_TRUE(hasKey); + + JS::Rooted key(cx); + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetProperty(cx, snapshot, "Normal", &key)); + + JS::Rooted values(cx); + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetProperty(cx, key, "values", &values)); + + JS::Rooted indexIds(cx, JS::IdVector(cx)); + ASSERT_NO_FATAL_FAILURE(js_helpers::Enumerate(cx, values, &indexIds)); + ASSERT_TRUE(indexIds.length() == 2u || indexIds.length() == 3u); + + if (indexIds.length() == 2) { + uint32_t value = 0u; + + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetIndexValue(cx, values, indexIds[0], &value)); + ASSERT_EQ(value, 1u); + + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetIndexValue(cx, values, indexIds[1], &value)); + ASSERT_EQ(value, 0u); + } else { + uint32_t value = 1u; + + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetIndexValue(cx, values, indexIds[0], &value)); + ASSERT_EQ(value, 0u); + + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetIndexValue(cx, values, indexIds[1], &value)); + ASSERT_EQ(value, 1u); + + ASSERT_NO_FATAL_FAILURE( + js_helpers::GetIndexValue(cx, values, indexIds[2], &value)); + ASSERT_EQ(value, 0u); + } + } + + { + bool hasKey = true; + ASSERT_NO_FATAL_FAILURE( + js_helpers::HasProperty(cx, snapshot, "WasSuspended", &hasKey)); + ASSERT_FALSE(hasKey); + } + + { + bool hasKey = true; + ASSERT_NO_FATAL_FAILURE( + js_helpers::HasProperty(cx, snapshot, "TimeStampErr1", &hasKey)); + ASSERT_FALSE(hasKey); + } + + { + bool hasKey = true; + ASSERT_NO_FATAL_FAILURE( + js_helpers::HasProperty(cx, snapshot, "TimeStampErr2", &hasKey)); + ASSERT_FALSE(hasKey); + } +} + +} // namespace mozilla::dom::quota::test -- cgit v1.2.3